|
3 | 3 | from django.contrib.auth.models import User
|
4 | 4 | from supplier.models import Supplier
|
5 | 5 | from supplier.views import action_display_all_products, get_supplier
|
6 |
| -from supplier.views import get_product_by_name_and_supplier, action_search_product |
| 6 | +from supplier.views import get_product_by_name_and_supplier, action_search_product, show_deliveries |
7 | 7 | from django.test import RequestFactory
|
8 | 8 | from supplier import views
|
9 | 9 | from supplier_product.models import SupplierProduct
|
10 | 10 | from product.models import Product
|
| 11 | +from delivery_location.models import DeliveryLocation |
11 | 12 |
|
12 | 13 |
|
13 | 14 | @pytest.fixture
|
@@ -130,40 +131,82 @@ def test_get_product_return_none(request_searched_product_mock,
|
130 | 131 | @pytest.mark.django_db()
|
131 | 132 | def test_searched_product_with_data(get_request_with_data,
|
132 | 133 | render_mock,
|
133 |
| - saved_supplier_product0): |
134 |
| - response = action_search_product(get_request_with_data, saved_supplier_product0) |
| 134 | + saved_supplier_product0, saved_supplier0): |
| 135 | + |
| 136 | + context = show_deliveries(saved_supplier0) |
| 137 | + response = action_search_product(get_request_with_data, saved_supplier_product0, context) |
135 | 138 | assert response == render_mock.return_value
|
136 | 139 |
|
137 |
| - data = {'supplier_products': saved_supplier_product0} |
| 140 | + # data = {'supplier_products': saved_supplier_product0} |
| 141 | + context['supplier_products'] = saved_supplier_product0 |
138 | 142 | assert render_mock.call_args_list == [
|
139 |
| - call(get_request_with_data, 'supplier/suppliers.html', data) |
| 143 | + call(get_request_with_data, 'supplier/suppliers.html', context) |
140 | 144 | ]
|
141 | 145 |
|
142 | 146 |
|
143 | 147 | @pytest.mark.django_db()
|
144 | 148 | def test_searched_product_without_data(get_request_with_data,
|
145 | 149 | render_mock,
|
146 |
| - messages_mock): |
147 |
| - response = action_search_product(get_request_with_data, None) |
| 150 | + messages_mock, saved_supplier0): |
| 151 | + context = show_deliveries(saved_supplier0) |
| 152 | + response = action_search_product(get_request_with_data, None, context) |
148 | 153 | assert response == render_mock.return_value
|
149 | 154 | messages_mock.info.assert_called_with(
|
150 | 155 | get_request_with_data, 'Product Not Found'
|
151 | 156 | )
|
152 | 157 | assert render_mock.call_args_list == [
|
153 |
| - call(get_request_with_data, 'supplier/suppliers.html') |
| 158 | + call(get_request_with_data, 'supplier/suppliers.html', context) |
154 | 159 | ]
|
155 | 160 |
|
156 | 161 |
|
157 | 162 | @pytest.mark.django_db()
|
158 | 163 | def test_action_display_all_products(get_request,
|
159 | 164 | saved_supplier0,
|
160 |
| - render_mock, |
161 |
| - saved_supplier_product0, |
162 |
| - saved_supplier_product1): |
163 |
| - response = action_display_all_products(get_request, saved_supplier0) |
| 165 | + render_mock): |
| 166 | + context = show_deliveries(saved_supplier0) |
| 167 | + response = action_display_all_products(get_request, saved_supplier0, context) |
164 | 168 | assert response == render_mock.return_value
|
165 | 169 |
|
166 | 170 | data = set(SupplierProduct.objects.filter(user_name=saved_supplier0))
|
| 171 | + context['supplier_products'] = data |
167 | 172 | assert render_mock.call_args_list == [
|
168 |
| - call(get_request, 'supplier/suppliers.html', {'supplier_products': data}) |
| 173 | + call(get_request, 'supplier/suppliers.html', context) |
169 | 174 | ]
|
| 175 | + |
| 176 | + |
| 177 | +@pytest.mark.django_db() |
| 178 | +def test_supplier_deliveries(client, delivery_location0): |
| 179 | + delivery_location0.add_delivery_location() |
| 180 | + client.force_login(user=delivery_location0.user_name.supplier_account) |
| 181 | + response = client.get('/suppliers/') |
| 182 | + assert list(response.context['supplier_deliveries']) == [delivery_location0] |
| 183 | + |
| 184 | + |
| 185 | +@pytest.mark.django_db() |
| 186 | +def test_create_delivery(client, saved_supplier0): |
| 187 | + client.force_login(user=saved_supplier0.supplier_account) |
| 188 | + response = client.get('/suppliers/?location=tel+aviv&date=2023-01-18&add=Add') |
| 189 | + assert len(DeliveryLocation.objects.filter(user_name=saved_supplier0)) != 0 |
| 190 | + assert response.status_code == 302 |
| 191 | + assert client.get('/suppliers/').status_code == 200 |
| 192 | + |
| 193 | + |
| 194 | +@pytest.mark.django_db() |
| 195 | +def test_show_deliveries_with_no_deliveries(client, saved_supplier0): |
| 196 | + client.force_login(user=saved_supplier0.supplier_account) |
| 197 | + response = client.get('/suppliers/') |
| 198 | + assert len(DeliveryLocation.objects.filter(user_name=saved_supplier0)) == 0 |
| 199 | + assert response.status_code == 200 |
| 200 | + assert len(list(response.context['supplier_deliveries'])) == 0 |
| 201 | + assert response.context['message'] == "You don't have any deliveries yet" |
| 202 | + |
| 203 | + |
| 204 | +@pytest.mark.django_db() |
| 205 | +def test_remove_delivery(client, saved_supplier0): |
| 206 | + client.force_login(user=saved_supplier0.supplier_account) |
| 207 | + client.get('/suppliers/?location=tel+aviv&date=2023-01-18&add=Add') |
| 208 | + assert len(DeliveryLocation.objects.filter(user_name=saved_supplier0)) != 0 |
| 209 | + response = client.get('/suppliers/?location=tel+aviv&date=2023-01-18&delete=Delete') |
| 210 | + assert len(DeliveryLocation.objects.filter(user_name=saved_supplier0)) == 0 |
| 211 | + assert response.status_code == 302 |
| 212 | + assert client.get('/suppliers/').status_code == 200 |
0 commit comments