from pprint import pprint from app.database import build_query from app.dtos import User, Vendor import xml.dom.minidom def test_queries_build(): new_user = User( name="Taras Shevchenko", email="taras@gmail.com", brand='sme', audience_profile=False, exists=False, vendors=[] ) existing_user_without_profile = User( name="Vasyl Stus", email="stus@orchard.com", brand='orchard', identity="222-111", audience_profile=False, exists=True, vendors=[] ) existed_user = User( name="Lina Kostenko", email="linakostenko@awal.com", brand='awal', exists=True, audience_profile=True, identity="444-111", vendors=[Vendor(name="Timeless", id=4242, uuid="4242-4242")] ) existed_user_all_up_to_date = User( name="Oleksa Tykhy", email="tykhy@sonymusic.com", brand='sme', exists=True, audience_profile=True, identity="111-111", vendors=[ Vendor(name="Timeless", id=4242, uuid="4242-4242"), Vendor(name="ULabel", id=21111819145, uuid="21111819145"), ] ) migration = build_query( [new_user, existed_user, existing_user_without_profile, existed_user_all_up_to_date], vendors_ids=[4242, 21111819145], pr_name="FS-TEST_test_something" ) print(migration) test_queries_build()