Make initial_data.py more fake

mykhaly
1 parent 923c6429
Showing 2 changed files with 94 additions and 303 deletions
op_robot_tests/tests_files/initial_data.py
op_robot_tests/tests_files/service_keywords.py
--- a/op_robot_tests/tests_files/initial_data.py
View file @6644874
+++ b/op_robot_tests/tests_files/initial_data.py
View file @6644874
@@ -4,11 +4,14 @@ from faker import Factory
 from munch import munchify
 from tempfile import NamedTemporaryFile
 from .local_time import get_now
+from .op_faker import OP_Provider
 import random
+
 fake = Factory.create('uk_UA')
 fake_ru = Factory.create('ru')
 fake_en = Factory.create()
+fake.add_provider(OP_Provider)
 def create_fake_sentence():
@@ -16,7 +19,21 @@ def create_fake_sentence():
 def field_with_id(prefix, sentence):
-    return "{}-{}: {}".format(prefix, fake.uuid4()[:8], sentence)
+    return u"{}-{}: {}".format(prefix, fake.uuid4()[:8], sentence)
+
+
+def translate_country_en(country):
+    if country == u"Україна":
+        return "Ukraine"
+    else:
+        raise Exception(u"Cannot translate country to english: {}".format(country))
+
+
+def translate_country_ru(country):
+    if country == u"Україна":
+        return u"Украина"
+    else:
+        raise Exception(u"Cannot translate country to russian: {}".format(country))
 def create_fake_doc():
@@ -30,38 +47,17 @@ def create_fake_doc():
 def test_tender_data(intervals, periods=("enquiry", "tender")):
     now = get_now()
-    value_amount = 50000.99
-    t_data = {
-        "title": u"[ТЕСТУВАННЯ] " + fake.catch_phrase(),
+    value_amount = round(random.uniform(3000, 250000000000), 2) #max value equals to budget of Ukraine in hryvnias
+    data = {
         "mode": "test",
         "submissionMethodDetails": "quick",
-        "description": u"Тестовий тендер",
-        "description_ru": u"Тестовый тендер",
-        "description_en": "Test tender",
-        "procuringEntity": {
-            "name": fake.company(),
-            "name_ru": fake_ru.company(),
-            "name_en": fake_en.company(),
-            "identifier": {
-                "scheme": u"UA-EDR",
-                "id": u"{:08d}".format(fake.pyint()),
-                "uri": fake.image_url(width=None, height=None)
-            },
-            "address": {
-                "countryName": u"Україна",
-                "countryName_ru": u"Украина",
-                "countryName_en": "Ukraine",
-                "postalCode": fake.postalcode(),
-                "region": u"м. Київ",
-                "locality": u"м. Київ",
-                "streetAddress": fake.street_address()
-            },
-            "contactPoint": {
-                "name": fake.name(),
-                "telephone": fake.phone_number()
-            },
-            "kind": "other"
-        },
+        "description": fake.description(),
+        "description_en": fake_en.sentence(nb_words=10, variable_nb_words=True),
+        "description_ru": fake_ru.sentence(nb_words=10, variable_nb_words=True),
+        "title": fake.title(),
+        "title_en": fake_en.catch_phrase(),
+        "title_ru": fake_ru.catch_phrase(),
+        "procuringEntity": fake.procuringEntity(),
         "value": {
             "amount": value_amount,
             "currency": u"UAH",
@@ -73,8 +69,13 @@ def test_tender_data(intervals, periods=("enquiry", "tender")):
         },
         "items": []
     }
+    data["procuringEntity"]["kind"] = "other"
     new_item = test_item_data()
-    t_data['items'].append(new_item)
+    data["items"].append(new_item)
+    if data.get("mode") == "test":
+        data["title"] = u"[ТЕСТУВАННЯ] {}".format(data["title"])
+        data["title_en"] = u"[TESTING] {}".format(data["title_en"])
+        data["title_ru"] = u"[ТЕСТИРОВАНИЕ] {}".format(data["title_ru"])
     period_dict = {}
     inc_dt = now
     for period_name in periods:
@@ -82,90 +83,18 @@ def test_tender_data(intervals, periods=("enquiry", "tender")):
         for i, j in zip(range(2), ("start", "end")):
             inc_dt += timedelta(minutes=intervals[period_name][i])
             period_dict[period_name + "Period"][j + "Date"] = inc_dt.isoformat()
-    t_data.update(period_dict)
-    return t_data
+    data.update(period_dict)
+    return munchify(data)
 def test_tender_data_limited(intervals, procurement_method_type):
-    now = get_now()
-    data = {
-        "items":
-        [
-            {
-                "additionalClassifications":
-                [
-                    {
-                        "description": u"Послуги щодо забезпечення харчуванням, інші",
-                        "id": "56.29",
-                        "scheme": u"ДКПП"
-                    }
-                ],
-                "classification":
-                {
-                    "description": u"Послуги з організації шкільного харчування",
-                    "id": "55523100-3",
-                    "scheme": "CPV"
-                },
-                "description": field_with_id('i', fake.sentence(nb_words=10, variable_nb_words=True)),
-                "id": "2dc54675d6364e2baffbc0f8e74432ac",
-                "deliveryDate": {
-                    "endDate": (now + timedelta(days=5)).isoformat()
-                },
-                "deliveryLocation": {
-                    "latitude": 49.8500,
-                    "longitude": 24.0167
-                },
-                "deliveryAddress": {
-                    "countryName": u"Україна",
-                    "countryName_ru": u"Украина",
-                    "countryName_en": "Ukraine",
-                    "postalCode": fake.postalcode(),
-                    "region": u"м. Київ",
-                    "locality": u"м. Київ",
-                    "streetAddress": fake.street_address()
-                }
-            }
-        ],
-        "mode": "test",
-        "procurementMethod": "limited",
-        "procurementMethodType": procurement_method_type,
-        "procuringEntity":
-        {
-            "address":
-            {
-                "countryName": u"Україна",
-                "locality": u"м. Вінниця",
-                "postalCode": "21027",
-                "region": u"м. Вінниця",
-                "streetAddress": u"вул. Стахурського. 22"
-            },
-            "contactPoint":
-            {
-                "name": u"Куца Світлана Валентинівна",
-                "telephone": "+380 (432) 46-53-02",
-                "url": "http://sch10.edu.vn.ua/"
-            },
-            "identifier":
-            {
-                "id": "21725150",
-                "legalName": u"Заклад \"Загальноосвітня школа І-ІІІ ступенів № 10 Вінницької міської ради\"",
-                "scheme": u"UA-EDR"
-            },
-            "name": u"ЗОСШ #10 м.Вінниці",
-            "kind": "general"
-        },
-        "value": {
-            "amount": 500000,
-            "currency": "UAH",
-            "valueAddedTaxIncluded": True
-        },
-        "description": fake.sentence(nb_words=10, variable_nb_words=True),
-        "description_en": fake.sentence(nb_words=10, variable_nb_words=True),
-        "description_ru": fake.sentence(nb_words=10, variable_nb_words=True),
-        "title": fake.catch_phrase(),
-        "title_en": fake.catch_phrase(),
-        "title_ru": fake.catch_phrase()
-    }
+    data = test_tender_data(intervals)
+    del data["submissionMethodDetails"]
+    del data["minimalStep"]
+    del data["enquiryPeriod"]
+    del data["tenderPeriod"]
+    data["procuringEntity"]["kind"] = "general"
+    data.update({"procurementMethodType": procurement_method_type, "procurementMethod": "limited"})
     if procurement_method_type == "negotiation":
         cause_variants = (
             "artContestIP",
@@ -181,13 +110,12 @@ def test_tender_data_limited(intervals, procurement_method_type):
         cause_variants = ('quick',)
         cause = fake.random_element(cause_variants)
         data.update({"cause": cause})
-    if procurement_method_type == "negotiation" \
-            or procurement_method_type == "negotiation.quick":
+    if procurement_method_type in ("negotiation", "negotiation.quick"):
         data.update({
             "procurementMethodDetails": "quick, accelerator=1440",
-            "causeDescription": fake.sentence(nb_words=10, variable_nb_words=True)
+            "causeDescription": fake.description()
         })
-    return data
+    return munchify(data)
 def test_tender_data_multiple_items(intervals):
@@ -195,7 +123,7 @@ def test_tender_data_multiple_items(intervals):
     for _ in range(4):
         new_item = test_item_data()
         t_data['items'].append(new_item)
-    return t_data
+    return munchify(t_data)
 def test_tender_data_multiple_lots(intervals):
@@ -204,7 +132,7 @@ def test_tender_data_multiple_lots(intervals):
     tender['items'][0]['relatedLot'] = first_lot_id
     tender['lots'] = [test_lot_data()]
     tender['lots'][0]['id'] = first_lot_id
-    return tender
+    return munchify(tender)
 def test_tender_data_meat(intervals):
@@ -215,24 +143,24 @@ def test_tender_data_meat(intervals):
         {
             "code": "ee3e24bc17234a41bd3e3a04cc28e9c6",
             "featureOf": "tenderer",
-            "title":  field_with_id('f', "Термін оплати"),
-            "description": "Умови відстрочки платежу після поставки товару",
+            "title": field_with_id("f", fake.title()),
+            "description": fake.description(),
             "enum": [
                 {
                     "value": 0.15,
-                    "title": "180 днів та більше"
+                    "title": fake.word()
                 },
                 {
                     "value": 0.1,
-                    "title": "90-179 днів",
+                    "title": fake.word()
                 },
                 {
                     "value": 0.05,
-                    "title": "30-89 днів"
+                    "title": fake.word()
                 },
                 {
                     "value": 0,
-                    "title": "Менше 30 днів"
+                    "title": fake.word()
                 }
             ]
         },
@@ -240,56 +168,35 @@ def test_tender_data_meat(intervals):
             "code": "48cfd91612c04125ab406374d7cc8d93",
             "featureOf": "item",
             "relatedItem": item_id,
-            "title":  field_with_id('f', "Сорт"),
-            "description": "Сорт продукції",
+            "title": field_with_id("f", fake.title()),
+            "description": fake.description(),
             "enum": [
                 {
                     "value": 0.05,
-                    "title": "Вищий"
+                    "title": fake.word()
                 },
                 {
                     "value": 0.01,
-                    "title": "Перший",
+                    "title": fake.word()
                 },
                 {
                     "value": 0,
-                    "title": "Другий"
+                    "title": fake.word()
                 }
             ]
         }
     ]
-    return tender
+    return munchify(tender)
 def test_question_data():
-    data = munchify({
+    return munchify({
         "data": {
-            "author": {
-                "address": {
-                    "countryName": u"Україна",
-                    "countryName_ru": u"Украина",
-                    "countryName_en": "Ukraine",
-                    "locality": u"м. Вінниця",
-                    "postalCode": "21100",
-                    "region": u"Вінницька область",
-                    "streetAddress": fake.street_address()
-                },
-                "contactPoint": {
-                    "name": fake.name(),
-                    "telephone": fake.phone_number()
-                },
-                "identifier": {
-                    "scheme": u"UA-EDR",
-                    "id": u"{:08d}".format(fake.pyint()),
-                    "uri": fake.image_url(width=None, height=None)
-                },
-                "name": fake.company()
-            },
-            "description": fake.sentence(nb_words=10, variable_nb_words=True),
-            "title": field_with_id('q', fake.sentence(nb_words=6, variable_nb_words=True))
+            "author": fake.procuringEntity(),
+            "description": fake.description(),
+            "title": field_with_id("q", fake.title())
         }
     })
-    return data
 def test_question_answer_data():
@@ -302,30 +209,10 @@ def test_question_answer_data():
 def test_complaint_data(lot=False):
     data = munchify({
-        "data": {
-            "author": {
-                "address": {
-                    "countryName": u"Україна",
-                    "countryName_ru": u"Украина",
-                    "countryName_en": "Ukraine",
-                    "locality": u"м. Вінниця",
-                    "postalCode": "21100",
-                    "region": u"Вінницька область",
-                    "streetAddress": fake.street_address()
-                },
-                "contactPoint": {
-                    "name": fake.name(),
-                    "telephone": fake.phone_number()
-                },
-                "identifier": {
-                    "scheme": u"UA-EDR",
-                    "id": u"{:08d}".format(fake.pyint()),
-                    "uri": fake.image_url(width=None, height=None)
-                },
-                "name": fake.company()
-            },
-            "description": fake.sentence(nb_words=10, variable_nb_words=True),
-            "title": fake.sentence(nb_words=6, variable_nb_words=True)
+        "data" : {
+            "author": fake.procuringEntity(),
+            "description": fake.description(),
+            "title": fake.title()
         }
     })
     if lot:
@@ -336,17 +223,6 @@ def test_complaint_data(lot=False):
 test_claim_data = test_complaint_data
-def test_complaint_answer_data(complaint_id):
-    return munchify({
-        "data": {
-            "id": complaint_id,
-            "status": "answered",
-            "resolutionType": "resolved",
-            "resolution": fake.sentence(nb_words=40, variable_nb_words=True)
-        }
-    })
-
-
 def test_claim_answer_satisfying_data(claim_id):
     return munchify({
         "data": {
@@ -397,14 +273,6 @@ def test_cancel_claim_data(claim_id, cancellation_reason):
     })
-def test_change_cancellation_document_field_data(key, value):
-    return munchify({
-        "data": {
-            key: value
-        }
-    })
-
-
 def test_confirm_data(id):
     return munchify({
         "data": {
@@ -435,29 +303,12 @@ def test_bid_data(mode):
     bid = munchify({
         "data": {
             "tenderers": [
-                {
-                    "address": {
-                        "countryName": u"Україна",
-                        "countryName_ru": u"Украина",
-                        "countryName_en": "Ukraine",
-                        "locality": u"м. Вінниця",
-                        "postalCode": "21100",
-                        "region": u"Вінницька область",
-                        "streetAddress": fake.street_address()
-                    },
-                    "contactPoint": {
-                        "name": fake.name(),
-                        "telephone": fake.phone_number()
-                    },
-                    "identifier": {
-                        "scheme": u"UA-EDR",
-                        "id": u"{:08d}".format(fake.pyint()),
-                    },
-                    "name": fake.company()
-                }
+                fake.procuringEntity()
             ]
         }
-    })
+    }
+    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"])
     if 'open' in mode:
         bid.data['selfEligible'] = True
         bid.data['selfQualified'] = True
@@ -500,30 +351,10 @@ def test_supplier_data():
     return munchify({
         "data": {
             "suppliers": [
-                {
-                    "address": {
-                        "countryName": u"Україна",
-                        "locality": u"м. Вінниця",
-                        "postalCode": "21100",
-                        "region": u"м. Вінниця",
-                        "streetAddress": u"вул. Островського, 33"
-                    },
-                    "contactPoint": {
-                        "email": "soleksuk@gmail.com",
-                        "name": u"Сергій Олексюк",
-                        "telephone": "+380 (432) 21-69-30"
-                    },
-                    "identifier": {
-                        "id": "13313462",
-                        "legalName": u"Державне комунальне підприємство громадського харчування «Школяр»",
-                        "scheme": "UA-EDR",
-                        "uri": "http://sch10.edu.vn.ua/"
-                    },
-                    "name": u"ДКП «Школяр»"
-                }
+                fake.procuringEntity()
             ],
             "value": {
-                "amount": 475000,
+                "amount": fake.random_int(min=1),
                 "currency": "UAH",
                 "valueAddedTaxIncluded": True
             }
@@ -531,52 +362,14 @@ def test_supplier_data():
     })
-def test_award_data():
-    return munchify({'data': {}})
-
-
-def test_item_data():
-    now = get_now()
-    return munchify({
-        "description": field_with_id('i', fake.catch_phrase()),
-        "deliveryDate": {
-            "endDate": (now + timedelta(days=5)).isoformat()
-        },
-        "deliveryLocation": {
-            "latitude": 49.8500,
-            "longitude": 24.0167
-        },
-        "deliveryAddress": {
-            "countryName": u"Україна",
-            "countryName_ru": u"Украина",
-            "countryName_en": "Ukraine",
-            "postalCode": fake.postalcode(),
-            "region": u"м. Київ",
-            "locality": u"м. Київ",
-            "streetAddress": fake.street_address()
-        },
-        "classification": {
-            "scheme": u"CPV",
-            "id": u"44617100-9",
-            "description": u"Картонні коробки",
-            "description_ru": u"Большие картонные коробки",
-            "description_en": u"Cartons"
-        },
-        "additionalClassifications": [
-            {
-                "scheme": u"ДКПП",
-                "id": u"17.21.1",
-                "description": u"Папір і картон гофровані, паперова й картонна тара"
-            }
-        ],
-        "unit": {
-            "name": u"кілограм",
-            "name_ru": u"килограмм",
-            "name_en": "kilogram",
-            "code": u"KGM"
-        },
-        "quantity": fake.pyint()
-    })
+def test_item_data(cpv=None):
+    data = fake.fake_item(cpv)
+    data["description"] = field_with_id("i", data["description"])
+    days = fake.random_int(min=1, max=30)
+    data["deliveryDate"] = {"endDate": (get_now() + timedelta(days=days)).isoformat()}
+    data["deliveryAddress"]["countryName_en"] = translate_country_en(data["deliveryAddress"]["countryName"])
+    data["deliveryAddress"]["countryName_ru"] = translate_country_ru(data["deliveryAddress"]["countryName"])
+    return munchify(data)
 def test_invalid_features_data():
@@ -584,16 +377,16 @@ def test_invalid_features_data():
         {
             "code": "ee3e24bc17234a41bd3e3a04cc28e9c6",
             "featureOf": "tenderer",
-            "title": "Термін оплати",
-            "description": "Умови відстрочки платежу після поставки товару",
+            "title": fake.title(),
+            "description": fake.description(),
             "enum": [
                 {
                     "value": 0.35,
-                    "title": "180 днів та більше"
+                    "title": fake.word()
                 },
                 {
                     "value": 0,
-                    "title": "Менше 30 днів"
+                    "title": fake.word()
                 }
             ]
         },
@@ -601,16 +394,16 @@ def test_invalid_features_data():
             "code": "48cfd91612c04125ab406374d7cc8d93",
             "featureOf": "item",
             "relatedItem": "edd0032574bf4402877ad5f362df225a",
-            "title": "Сорт",
-            "description": "Сорт продукції",
+            "title": fake.title(),
+            "description": fake.description(),
             "enum": [
                 {
                     "value": 0.35,
-                    "title": "Вищий"
+                    "title": fake.word()
                 },
                 {
                     "value": 0,
-                    "title": "Другий"
+                    "title": fake.word()
                 }
             ]
         }
@@ -618,18 +411,19 @@ def test_invalid_features_data():
 def test_lot_data():
+    value_amount = round(random.uniform(3000, 250000000000), 2) #max value equals to budget of Ukraine in hryvnias
     return munchify(
         {
-            "description": fake.sentence(nb_words=10, variable_nb_words=True),
-            "title": field_with_id('l', fake.sentence(nb_words=6, variable_nb_words=True)),
+            "description": fake.description(),
+            "title": field_with_id('l', fake.title()),
             "value": {
                 "currency": "UAH",
-                "amount": 2000 + fake.pyfloat(left_digits=4, right_digits=1, positive=True),
+                "amount": value_amount,
                 "valueAddedTaxIncluded": True
             },
             "minimalStep": {
                 "currency": "UAH",
-                "amount": 30.0,
+                "amount": round(random.uniform(0.005, 0.03) * value_amount, 2),
                 "valueAddedTaxIncluded": True
             },
             "status": "active"
--- a/op_robot_tests/tests_files/service_keywords.py
View file @6644874
+++ b/op_robot_tests/tests_files/service_keywords.py
View file @6644874
@@ -20,15 +20,12 @@ from math import radians, cos, sin, asin, sqrt
 from .initial_data import (
     create_fake_doc,
     create_fake_sentence,
-    test_award_data,
     test_bid_data,
     test_cancel_claim_data,
     test_cancel_tender_data,
-    test_change_cancellation_document_field_data,
     test_claim_answer_data,
     test_claim_answer_satisfying_data,
     test_claim_data,
-    test_complaint_answer_data,
     test_complaint_data,
     test_complaint_reply_data,
     test_confirm_data,