Merge pull request #274 from ProzorroUKR/PriceQuotation

fix kind for PQ planning

@@ -303,7 +303,6 @@ def test_tender_data_planning(params):
             }
             }
         },
         },
         "procuringEntity": {
         "procuringEntity": {
-            #"kind": "general",
             "identifier": {
             "identifier": {
                 "scheme": "UA-EDR",
                 "scheme": "UA-EDR",
                 "id": random.choice(["13313462", "00037256"]),
                 "id": random.choice(["13313462", "00037256"]),
@@ -333,6 +332,8 @@ def test_tender_data_planning(params):
         data["procuringEntity"]["kind"] = "defense"
         data["procuringEntity"]["kind"] = "defense"
     elif params.get("mode") in ["belowThreshold", "reporting"]:
     elif params.get("mode") in ["belowThreshold", "reporting"]:
         data["procuringEntity"]["kind"] = "other"
         data["procuringEntity"]["kind"] = "other"
+    elif params.get("mode") in ["priceQuotation"]:
+        data["procuringEntity"]["kind"] = random.choice(['authority', 'defense', 'general', 'social', 'special'])
     else:
     else:
         data["procuringEntity"]["kind"] = random.choice(["general", "special", "central", "authority", "social"])
         data["procuringEntity"]["kind"] = random.choice(["general", "special", "central", "authority", "social"])
     buyers = test_buyers_data()
     buyers = test_buyers_data()
@@ -376,12 +377,6 @@ def test_tender_data_planning(params):
             breakdown_element = test_breakdown_data()
             breakdown_element = test_breakdown_data()
             breakdown_element['value']['amount'] = value
             breakdown_element['value']['amount'] = value
             data['budget']['breakdown'].append(breakdown_element)
             data['budget']['breakdown'].append(breakdown_element)
-    if params['mode'] == "priceQuotation":
-        for buyer in data['buyers']:
-            del buyer['kind']
-            del buyer['address']
-        del data['procuringEntity']['kind']
-        del data['procuringEntity']['address']
     return munchify(data)
     return munchify(data)