Generate test data for openua_defense procedure

@@ -466,6 +466,16 @@ def test_tender_data_openua(params, submissionMethodDetails):
     return data
     return data
 
 
 
 
+def test_tender_data_openua_defense(params, submissionMethodDetails):
+    """We should not provide any values for `enquiryPeriod` when creating
+    an openUA, openEU or openUA_defense procedure. That field should not be present at all.
+    Therefore, we pass a nondefault list of periods to `test_tender_data()`."""
+    data = test_tender_data(params, ('tender',), submissionMethodDetails)
+    data['procurementMethodType'] = 'aboveThresholdUA.defense'
+    data['procuringEntity']['kind'] = 'defense'
+    return data
+
+
 def test_tender_data_openeu(params, submissionMethodDetails):
 def test_tender_data_openeu(params, submissionMethodDetails):
     # We should not provide any values for `enquiryPeriod` when creating
     # We should not provide any values for `enquiryPeriod` when creating
     # an openUA or openEU procedure. That field should not be present at all.
     # an openUA or openEU procedure. That field should not be present at all.

@@ -47,6 +47,7 @@ from .initial_data import (
     test_tender_data_openeu,
     test_tender_data_openeu,
     test_tender_data_openua,
     test_tender_data_openua,
     test_tender_data_planning,
     test_tender_data_planning,
+    test_tender_data_openua_defense,
     create_fake_title,
     create_fake_title,
     create_fake_value_amount,
     create_fake_value_amount,
     test_change_document_data,
     test_change_document_data,
@@ -315,6 +316,9 @@ def prepare_test_tender_data(procedure_intervals,
     elif mode == 'openua':
     elif mode == 'openua':
         return munchify({'data': test_tender_data_openua(
         return munchify({'data': test_tender_data_openua(
             tender_parameters, submissionMethodDetails)})
             tender_parameters, submissionMethodDetails)})
+    elif mode == 'openua_defense':
+        return munchify({'data': test_tender_data_openua_defense(
+            tender_parameters, submissionMethodDetails)})
     elif mode == 'open_competitive_dialogue':
     elif mode == 'open_competitive_dialogue':
         return munchify({'data': test_tender_data_competitive_dialogue(
         return munchify({'data': test_tender_data_competitive_dialogue(
             tender_parameters, submissionMethodDetails)})
             tender_parameters, submissionMethodDetails)})
@@ -511,7 +515,13 @@ def get_object_by_id(data, given_object_id, slice_element, object_id):
 
 
 def generate_test_bid_data(tender_data):
 def generate_test_bid_data(tender_data):
     bid = test_bid_data()
     bid = test_bid_data()
-    if tender_data.get('procurementMethodType', '')[:-2] in ('aboveThreshold', 'competitiveDialogue'):
+    if tender_data.get('procurementMethodType', '') in (
+            'aboveThresholdUA',
+            'aboveThresholdUA.defense',
+            'aboveThresholdEU',
+            'competitiveDialogueUA'
+            'competitiveDialogueEU'
+        ):
         bid.data.selfEligible = True
         bid.data.selfEligible = True
         bid.data.selfQualified = True
         bid.data.selfQualified = True
     if 'lots' in tender_data:
     if 'lots' in tender_data: