|
29 | 29 | LOCATION = 'us-west1' |
30 | 30 |
|
31 | 31 | PRODUCT_SET_DISPLAY_NAME = 'fake_product_set_display_name_for_testing' |
32 | | -PRODUCT_SET_ID = 'test_set_{}'.format(uuid.uuid4()) |
33 | 32 |
|
34 | 33 | PRODUCT_DISPLAY_NAME = 'fake_product_display_name_for_testing' |
35 | 34 | PRODUCT_CATEGORY = 'homegoods' |
36 | | -PRODUCT_ID = 'test_product_{}'.format(uuid.uuid4()) |
37 | 35 |
|
38 | 36 |
|
39 | | -@pytest.fixture(scope="function", autouse=True) |
40 | | -def setup_teardown(): |
| 37 | +@pytest.fixture(scope="function") |
| 38 | +def test_resources(): |
41 | 39 | # set up |
| 40 | + product_set_id = f'test_set_{uuid.uuid4()}' |
| 41 | + product_id = f'test_product_{uuid.uuid4()}' |
42 | 42 | create_product_set( |
43 | | - PROJECT_ID, LOCATION, PRODUCT_SET_ID, PRODUCT_SET_DISPLAY_NAME) |
| 43 | + PROJECT_ID, LOCATION, product_set_id, PRODUCT_SET_DISPLAY_NAME) |
44 | 44 | create_product( |
45 | | - PROJECT_ID, LOCATION, PRODUCT_ID, |
| 45 | + PROJECT_ID, LOCATION, product_id, |
46 | 46 | PRODUCT_DISPLAY_NAME, PRODUCT_CATEGORY) |
47 | 47 |
|
48 | | - yield |
| 48 | + yield product_set_id, product_id |
49 | 49 |
|
50 | 50 | # tear down |
51 | | - delete_product(PROJECT_ID, LOCATION, PRODUCT_ID) |
52 | | - delete_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID) |
| 51 | + delete_product(PROJECT_ID, LOCATION, product_id) |
| 52 | + delete_product_set(PROJECT_ID, LOCATION, product_set_id) |
53 | 53 |
|
54 | 54 |
|
55 | | -def test_add_product_to_product_set(capsys): |
| 55 | +def test_add_product_to_product_set(capsys, test_resources): |
| 56 | + product_set_id, product_id = test_resources |
56 | 57 | add_product_to_product_set( |
57 | | - PROJECT_ID, LOCATION, PRODUCT_ID, PRODUCT_SET_ID) |
58 | | - list_products_in_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID) |
| 58 | + PROJECT_ID, LOCATION, product_id, product_set_id) |
| 59 | + list_products_in_product_set(PROJECT_ID, LOCATION, product_set_id) |
59 | 60 | out, _ = capsys.readouterr() |
60 | | - assert 'Product id: {}'.format(PRODUCT_ID) in out |
| 61 | + assert 'Product id: {}'.format(product_id) in out |
61 | 62 |
|
62 | 63 |
|
63 | | -def test_remove_product_from_product_set(capsys): |
| 64 | +def test_remove_product_from_product_set(capsys, test_resources): |
| 65 | + product_set_id, product_id = test_resources |
64 | 66 | add_product_to_product_set( |
65 | | - PROJECT_ID, LOCATION, PRODUCT_ID, PRODUCT_SET_ID) |
66 | | - list_products_in_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID) |
| 67 | + PROJECT_ID, LOCATION, product_id, product_set_id) |
| 68 | + list_products_in_product_set(PROJECT_ID, LOCATION, product_set_id) |
67 | 69 | out, _ = capsys.readouterr() |
68 | | - assert 'Product id: {}'.format(PRODUCT_ID) in out |
| 70 | + assert 'Product id: {}'.format(product_id) in out |
69 | 71 |
|
70 | 72 | remove_product_from_product_set( |
71 | | - PROJECT_ID, LOCATION, PRODUCT_ID, PRODUCT_SET_ID) |
72 | | - list_products_in_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID) |
| 73 | + PROJECT_ID, LOCATION, product_id, product_set_id) |
| 74 | + list_products_in_product_set(PROJECT_ID, LOCATION, product_set_id) |
73 | 75 | out, _ = capsys.readouterr() |
74 | | - assert 'Product id: {}'.format(PRODUCT_ID) not in out |
| 76 | + assert 'Product id: {}'.format(product_id) not in out |
75 | 77 |
|
76 | 78 |
|
77 | | -def test_purge_products_in_product_set(capsys): |
| 79 | +def test_purge_products_in_product_set(capsys, test_resources): |
| 80 | + product_set_id, product_id = test_resources |
78 | 81 | add_product_to_product_set( |
79 | | - PROJECT_ID, LOCATION, PRODUCT_ID, PRODUCT_SET_ID) |
| 82 | + PROJECT_ID, LOCATION, product_id, product_set_id) |
80 | 83 | list_products(PROJECT_ID, LOCATION) |
81 | 84 | out, _ = capsys.readouterr() |
82 | | - assert 'Product id: {}'.format(PRODUCT_ID) in out |
| 85 | + assert 'Product id: {}'.format(product_id) in out |
83 | 86 |
|
84 | 87 | purge_products_in_product_set( |
85 | | - PROJECT_ID, LOCATION, PRODUCT_SET_ID, force=True) |
| 88 | + PROJECT_ID, LOCATION, product_set_id, force=True) |
86 | 89 |
|
87 | 90 | list_products(PROJECT_ID, LOCATION) |
88 | 91 | out, _ = capsys.readouterr() |
89 | | - assert 'Product id: {}'.format(PRODUCT_ID) not in out |
| 92 | + assert 'Product id: {}'.format(product_id) not in out |
0 commit comments