Commit 9c5959ad30fb6b2c50290ab955009fd369325373
1 parent
c7c71145
add function to generate data for dasu
Showing
2 changed files
with
89 additions
and
4 deletions
| ... | ... | @@ -19,9 +19,12 @@ fake_uk = Factory.create(locale='uk_UA') |
| 19 | 19 | fake_uk.add_provider(OP_Provider) |
| 20 | 20 | fake = fake_uk |
| 21 | 21 | used_identifier_id = [] |
| 22 | -mode_open = ["belowThreshold", "aboveThresholdUA", "aboveThresholdEU", | |
| 23 | - "aboveThresholdUA.defense", "competitiveDialogueUA", "competitiveDialogueEU", "esco"] | |
| 24 | -mode_limited = ["reporting", "negotiation.quick", "negotiation"] | |
| 22 | +mode_open = ["belowThreshold", "aboveThresholdUA", "aboveThresholdEU", | |
| 23 | + "aboveThresholdUA.defense", "competitiveDialogueUA", "competitiveDialogueEU", "esco"] | |
| 24 | +mode_limited = ["reporting", "negotiation.quick", "negotiation"] | |
| 25 | +violationType = ["corruptionDescription", "corruptionProcurementMethodType", "corruptionChanges", | |
| 26 | + "corruptionPublicDisclosure", "corruptionBiddingDocuments", "documentsForm", | |
| 27 | + "corruptionAwarded", "corruptionCancelled", "corruptionContracting"] | |
| 25 | 28 | |
| 26 | 29 | # This workaround fixes an error caused by missing "catch_phrase" class method |
| 27 | 30 | # for the "ru_RU" locale in Faker >= 0.7.4 |
| ... | ... | @@ -511,6 +514,7 @@ def test_lot_document_data(document, lot_id): |
| 511 | 514 | document.data.update({"documentOf": "lot", "relatedItem": lot_id}) |
| 512 | 515 | return munchify(document) |
| 513 | 516 | |
| 517 | + | |
| 514 | 518 | def test_change_document_data(document, change_id): |
| 515 | 519 | document.data.update({"documentOf": "change", "relatedItem": change_id}) |
| 516 | 520 | return munchify(document) |
| ... | ... | @@ -588,4 +592,79 @@ def test_change_data(): |
| 588 | 592 | |
| 589 | 593 | |
| 590 | 594 | def get_hash(file_contents): |
| 591 | - return ("md5:"+hashlib.md5(file_contents).hexdigest()) | |
| \ No newline at end of file | ||
| 595 | + return ("md5:"+hashlib.md5(file_contents).hexdigest()) | |
| 596 | + | |
| 597 | + | |
| 598 | +def tets_monitoring_data( tender_id, accelerator=None): | |
| 599 | + data = { | |
| 600 | + "reasons": [random.choice(["public", "fiscal", "indicator", "authorities", "media"])], | |
| 601 | + "tender_id": tender_id, | |
| 602 | + "procuringStages": [random.choice(["awarding", "contracting", "planning"])], | |
| 603 | + "parties": [fake.procuringEntity()], | |
| 604 | + "decision": { | |
| 605 | + "date": get_now().isoformat(), | |
| 606 | + "description": fake_en.sentence(nb_words=10, variable_nb_words=True) | |
| 607 | + }, | |
| 608 | + "mode": "test" | |
| 609 | + } | |
| 610 | + data["parties"][0]["roles"] = [random.choice(["create", "decision", "conclusion"])] | |
| 611 | + data["parties"][0]["name"] = "The State Audit Service of Ukraine" | |
| 612 | + data['monitoringDetails'] = 'quick, ' \ | |
| 613 | + 'accelerator={}'.format(accelerator) | |
| 614 | + return munchify({'data':data}) | |
| 615 | + | |
| 616 | + | |
| 617 | +def test_party(party): | |
| 618 | + party["roles"] = "dialogue" | |
| 619 | + del party["kind"] | |
| 620 | + return munchify({"data":party}) | |
| 621 | + | |
| 622 | + | |
| 623 | +def test_dialogue(relatedParty_id): | |
| 624 | + return munchify( | |
| 625 | + { | |
| 626 | + "data": | |
| 627 | + { | |
| 628 | + "title": fake_en.sentence(nb_words=10, variable_nb_words=True), | |
| 629 | + "relatedParty": relatedParty_id, | |
| 630 | + "description": fake_en.sentence(nb_words=10, variable_nb_words=True) | |
| 631 | + } | |
| 632 | + }) | |
| 633 | + | |
| 634 | + | |
| 635 | +def test_conclusion(violationOccurred=False): | |
| 636 | + return munchify( | |
| 637 | + { | |
| 638 | + "data": { | |
| 639 | + "conclusion": { | |
| 640 | + "violationOccurred": violationOccurred, | |
| 641 | + "violationType": random.choice(violationType) | |
| 642 | + } | |
| 643 | + } | |
| 644 | + }) | |
| 645 | + | |
| 646 | + | |
| 647 | +def test_status_data(status): | |
| 648 | + data = { | |
| 649 | + "data": { | |
| 650 | + "status": status | |
| 651 | + } | |
| 652 | + } | |
| 653 | + if status in ('stopped', 'cancelled'): | |
| 654 | + data["data"]["cancellation"] = {} | |
| 655 | + data["data"]["cancellation"]["description"] = fake_en.sentence(nb_words=10, variable_nb_words=True) | |
| 656 | + return munchify(data) | |
| 657 | + | |
| 658 | + | |
| 659 | +def test_elimination_report(corruption): | |
| 660 | + return munchify({ | |
| 661 | + "data": { | |
| 662 | + "eliminationResolution": { | |
| 663 | + "resultByType": { | |
| 664 | + corruption: random.choice(["eliminated", "not_eliminated", "no_mechanism"]) | |
| 665 | + }, | |
| 666 | + "result": random.choice(["completely", "partly", "none"]), | |
| 667 | + "description": fake_en.sentence(nb_words=10, variable_nb_words=True) | |
| 668 | + } | |
| 669 | + } | |
| 670 | + }) | |
| \ No newline at end of file | ... | ... |
| ... | ... | @@ -52,6 +52,12 @@ from .initial_data import ( |
| 52 | 52 | test_tender_data_planning, |
| 53 | 53 | test_tender_data_openua_defense, |
| 54 | 54 | test_bid_competitive_data, |
| 55 | + tets_monitoring_data, | |
| 56 | + test_party, | |
| 57 | + test_dialogue, | |
| 58 | + test_conclusion, | |
| 59 | + test_status_data, | |
| 60 | + test_elimination_report, | |
| 55 | 61 | create_fake_title, |
| 56 | 62 | create_fake_value_amount, |
| 57 | 63 | test_change_document_data, | ... | ... |
Please
register
or
login
to post a comment