Update data preparation code to support openUA

@@ -19,7 +19,7 @@ def create_fake_doc():
     return tf.name
     return tf.name
 
 
 
 
-def test_tender_data(intervals):
+def test_tender_data(intervals, periods=("enquiry", "tender")):
     now = get_now()
     now = get_now()
     t_data = {
     t_data = {
         "title": u"[ТЕСТУВАННЯ] " + fake.catch_phrase(),
         "title": u"[ТЕСТУВАННЯ] " + fake.catch_phrase(),
@@ -104,7 +104,7 @@ def test_tender_data(intervals):
     }
     }
     period_dict = {}
     period_dict = {}
     inc_dt = now
     inc_dt = now
-    for period_name in ("enquiry", "tender"):
+    for period_name in periods:
         period_dict[period_name + "Period"] = {}
         period_dict[period_name + "Period"] = {}
         for i, j in zip(range(2), ("start", "end")):
         for i, j in zip(range(2), ("start", "end")):
             inc_dt += timedelta(minutes=intervals[period_name][i])
             inc_dt += timedelta(minutes=intervals[period_name][i])
@@ -166,7 +166,6 @@ def test_tender_data_limited(intervals):
 
 
 
 
 def test_tender_data_multiple_items(intervals):
 def test_tender_data_multiple_items(intervals):
-    now = get_now()
     t_data = test_tender_data(intervals)
     t_data = test_tender_data(intervals)
     for _ in range(4):
     for _ in range(4):
         new_item = test_item_data()
         new_item = test_item_data()
@@ -581,14 +580,16 @@ def test_lot_complaint_data(complaint, lot_id="3c8f387879de4c38957402dbdb8b31af"
     return munchify({"data": lot_complaint})
     return munchify({"data": lot_complaint})
 
 
 
 
-def test_tender_data_openua(intervals, accelerator=0):
-    t_data = test_tender_data(intervals)
-    now = get_now()
+def test_tender_data_openua(intervals):
+    accelerator = intervals['accelerator']
+    # Since `accelerator` field is not really a list containing timings
+    # for a period called `acceleratorPeriod`, let's remove it :)
+    del intervals['accelerator']
+    # We should not provide any values for `enquiryPeriod` when creating
+    # an openUA or openEU procedure. That field should not be present at all.
+    # Therefore, we pass a nondefault list of periods to `test_tender_data()`.
+    t_data = test_tender_data(intervals, periods=('tender',))
     t_data['procurementMethodType'] = 'aboveThresholdUA'
     t_data['procurementMethodType'] = 'aboveThresholdUA'
-    t_data['procurementMethodDetails'] = 'quick, accelerator={}'.format(accelerator)
-    t_data["tenderPeriod"] = {
-        "startDate": (now).isoformat(),
-        "endDate": (now + timedelta(minutes=15)).isoformat()
-    }
-    del t_data["enquiryPeriod"]
+    t_data['procurementMethodDetails'] = 'quick, ' \
+        'accelerator={}'.format(accelerator)
     return t_data
     return t_data

@@ -107,14 +107,34 @@ def load_initial_data_from(file_name):
             return fromYAML(file_obj)
             return fromYAML(file_obj)
 
 
 
 
-def prepare_test_tender_data(period_intervals, mode):
+def prepare_test_tender_data(procedure_intervals, mode):
+    # Get actual intervals by mode name
+    if mode in procedure_intervals:
+        intervals = procedure_intervals[mode]
+    else:
+        intervals = procedure_intervals['default']
+    LOGGER.log_message(Message(intervals))
+
+    # Set acceleration value for certain modes
+    if mode in ['openua', 'openeu']:
+        assert isinstance(intervals['accelerator'], int), \
+            "Accelerator should be an 'int', " \
+            "not '{}'".format(type(intervals['accelerator']).__name__)
+        assert intervals['accelerator'] >= 0, \
+            "Accelerator should not be less than 0"
+    else:
+        assert 'accelerator' not in intervals.keys(), \
+               "Accelerator is not available for mode '{0}'".format(mode)
+
     if mode == 'single':
     if mode == 'single':
-        return munchify({'data': test_tender_data(period_intervals)})
+        return munchify({'data': test_tender_data(intervals)})
     elif mode == 'multi':
     elif mode == 'multi':
-        return munchify({'data': test_tender_data_multiple_items(period_intervals)})
+        return munchify({'data': test_tender_data_multiple_items(intervals)})
     elif mode == 'limited':
     elif mode == 'limited':
-        return munchify({'data': test_tender_data_limited(period_intervals)})
-    raise ValueError('Invalid mode for test_tender_data')
+        return munchify({'data': test_tender_data_limited(intervals)})
+    elif mode == 'openua':
+        return munchify({'data': test_tender_data_openua(intervals)})
+    raise ValueError("Invalid mode for prepare_test_tender_data")
 
 
 
 
 def run_keyword_and_ignore_keyword_definitions(name, *args):
 def run_keyword_and_ignore_keyword_definitions(name, *args):