stage 2 competitive dialogue bid data generation

@@ -624,6 +624,21 @@ def test_bid_competitive_data():
     return bid
     return bid
 
 
 
 
+def test_bid_competitive_data_stage_2(id):
+    bid = munchify({
+        "data": {
+            "tenderers": [
+                fake.procuringTenderer()
+            ]
+        }
+    })
+    bid.data.tenderers[0].identifier.id = id
+    bid.data.tenderers[0].address.countryName_en = translate_country_en(bid.data.tenderers[0].address.countryName)
+    bid.data.tenderers[0].address.countryName_ru = translate_country_ru(bid.data.tenderers[0].address.countryName)
+    bid.data['status'] = 'draft'
+    return bid
+
+
 def test_bid_data():
 def test_bid_data():
     bid = munchify({
     bid = munchify({
         "data": {
         "data": {

@@ -86,7 +86,8 @@ from .initial_data import (
     test_cancellation_data,
     test_cancellation_data,
     test_cancel_pending_data,
     test_cancel_pending_data,
     test_payment_data,
     test_payment_data,
-    test_24_hours_data
+    test_24_hours_data,
+    test_bid_competitive_data_stage_2
 )
 )
 from barbecue import chef
 from barbecue import chef
 from restkit import request
 from restkit import request
@@ -621,7 +622,7 @@ def get_object_by_id(data, given_object_id, slice_element, object_id):
     return sliced_object[0]
     return sliced_object[0]
 
 
 
 
-def generate_test_bid_data(tender_data):
+def generate_test_bid_data(tender_data, edrpou=None):
     if tender_data.get('procurementMethodType', '') in (
     if tender_data.get('procurementMethodType', '') in (
             'aboveThresholdUA',
             'aboveThresholdUA',
             'aboveThresholdUA.defense',
             'aboveThresholdUA.defense',
@@ -634,6 +635,12 @@ def generate_test_bid_data(tender_data):
         bid = test_bid_competitive_data()
         bid = test_bid_competitive_data()
         bid.data.selfEligible = True
         bid.data.selfEligible = True
         bid.data.selfQualified = True
         bid.data.selfQualified = True
+    elif tender_data.get('procurementMethodType', '') in (
+            'competitiveDialogueUA.stage2',
+            'competitiveDialogueEU.stage2'):
+        bid = test_bid_competitive_data_stage_2(edrpou)
+        bid.data.selfEligible = True
+        bid.data.selfQualified = True
     else:
     else:
         bid = test_bid_data()
         bid = test_bid_data()
     if 'lots' in tender_data:
     if 'lots' in tender_data: