|
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