generate test data for guarantee criteria/response

@@ -189,10 +189,6 @@ def test_tender_data(params,
             "currency": u"UAH",
             "currency": u"UAH",
             "valueAddedTaxIncluded": vat_included
             "valueAddedTaxIncluded": vat_included
         },
         },
-        "guarantee": {
-            "amount": value_amount * 0.75,
-            "currency": u"UAH",
-        },
         "minimalStep": {
         "minimalStep": {
             "amount": round(random.uniform(0.005, 0.03) * value_amount, 2),
             "amount": round(random.uniform(0.005, 0.03) * value_amount, 2),
             "currency": u"UAH",
             "currency": u"UAH",
@@ -201,6 +197,9 @@ def test_tender_data(params,
         "items": [],
         "items": [],
         "features": []
         "features": []
     }
     }
+    if params.get("mode") in ["belowThreshold", "aboveThresholdUA", "aboveThresholdEU", "esco"]:
+        data["guarantee"]["amount"] = value_amount * 0.75
+        data["guarantee"]["currency"] = u"UAH"
     if params.get("mode") == "open_framework":
     if params.get("mode") == "open_framework":
         data["mainProcurementCategory"] = random.choice(['goods', 'services'])
         data["mainProcurementCategory"] = random.choice(['goods', 'services'])
     elif params.get("mode") == "open_competitive_dialogue":
     elif params.get("mode") == "open_competitive_dialogue":
@@ -1270,6 +1269,13 @@ def test_article_17_data():
     })
     })
 
 
 
 
+def test_criteria_guarantee_data():
+    criteria = fake.criteria_bid_contract_guarantee()
+    return munchify({
+        "data": criteria
+    })
+
+
 def test_data_bid_criteria():
 def test_data_bid_criteria():
     bid = munchify({
     bid = munchify({
         "data": []
         "data": []
@@ -1302,20 +1308,20 @@ def test_bid_criteria(tender_data, criteria_len, bid_data, bid_document):
     for criteria in tender_data["data"]['criteria']:
     for criteria in tender_data["data"]['criteria']:
         if criteria.get('source') == 'tenderer':
         if criteria.get('source') == 'tenderer':
             for requirement in criteria['requirementGroups'][0]['requirements']:
             for requirement in criteria['requirementGroups'][0]['requirements']:
-                mock = deepcopy(mock)
-                mock["requirement"]["id"] = requirement["id"]
-                mock["requirement"]["title"] = requirement["title"]
+                mock_tenderer = deepcopy(mock)
+                mock_tenderer["requirement"]["id"] = requirement["id"]
+                mock_tenderer["requirement"]["title"] = requirement["title"]
+                mock_tenderer["evidences"][0]["relatedDocument"]["id"] = bid_document["data"]["id"]
+                mock_tenderer["evidences"][0]["relatedDocument"]["title"] = bid_document["data"]["title"]
                 if criteria.get('title') == u'Мова (мови), якою (якими) повинні готуватися тендерні пропозиції':
                 if criteria.get('title') == u'Мова (мови), якою (якими) повинні готуватися тендерні пропозиції':
-                    del mock["evidences"][0]
-                else:
-                    mock["evidences"][0]["relatedDocument"]["id"] = bid_document["data"]["id"]
-                    mock["evidences"][0]["relatedDocument"]["title"] = bid_document["data"]["title"]
-                bid.data.append(mock)
+                    del mock_tenderer["evidences"][0]
+                bid.data.append(mock_tenderer)
         elif criteria.get('source') == 'winner':
         elif criteria.get('source') == 'winner':
             for requirement in criteria['requirementGroups'][0]['requirements']:
             for requirement in criteria['requirementGroups'][0]['requirements']:
                 mock = deepcopy(mock)
                 mock = deepcopy(mock)
                 mock["requirement"]["id"] = requirement["id"]
                 mock["requirement"]["id"] = requirement["id"]
                 mock["requirement"]["title"] = requirement["title"]
                 mock["requirement"]["title"] = requirement["title"]
+                del mock["evidences"][0]
             bid.data.append(mock)
             bid.data.append(mock)
         else:
         else:
             pass
             pass

@@ -40,6 +40,7 @@ class OP_Provider(BaseProvider):
     profiles_shortlistedfirms_empty = _fake_data.profiles_shortlistedfirms_empty
     profiles_shortlistedfirms_empty = _fake_data.profiles_shortlistedfirms_empty
     unknown_profile = _fake_data.unknown_profiles
     unknown_profile = _fake_data.unknown_profiles
     criteria = _fake_data.criteria
     criteria = _fake_data.criteria
+    criteria_guarant = _fake_data.criteria_guarantee
 
 
     @classmethod
     @classmethod
     def randomize_nb_elements(self, number=10, le=60, ge=140):
     def randomize_nb_elements(self, number=10, le=60, ge=140):
@@ -259,3 +260,7 @@ class OP_Provider(BaseProvider):
     @classmethod
     @classmethod
     def criteria_article_17(self):
     def criteria_article_17(self):
         return deepcopy(self.criteria)
         return deepcopy(self.criteria)
+
+    @classmethod
+    def criteria_bid_contract_guarantee(self):
+        return deepcopy(self.criteria_guarant)

@@ -96,7 +96,8 @@ from .initial_data import (
     test_qualification_criteria,
     test_qualification_criteria,
     test_awards_criteria,
     test_awards_criteria,
     test_tender_data_simple_defense,
     test_tender_data_simple_defense,
-    test_contract_criteria_response_data
+    test_contract_criteria_response_data,
+    test_criteria_guarantee_data
 )
 )
 from barbecue import chef
 from barbecue import chef
 from restkit import request
 from restkit import request