1from unittest.mock import patch 

2from canonicalwebteam.store_api.exceptions import StoreApiResponseErrorList 

3 

4from tests.admin.tests_models import TestModelServiceEndpoints 

5 

6 

7class TestGetBrandStoreEndpoint(TestModelServiceEndpoints): 

8 @patch("webapp.admin.views.admin_api.get_brand") 

9 def test_successful_get_brand_store(self, mock_get_brand): 

10 mock_get_brand.return_value = {"name": "BrandName"} 

11 

12 response = self.client.get("/admin/store/1/brand") 

13 data = response.get_json() 

14 

15 self.assertEqual(response.status_code, 200) 

16 self.assertTrue(data["success"]) 

17 self.assertEqual(data["data"], {"name": "BrandName"}) 

18 

19 @patch("webapp.admin.views.admin_api.get_brand") 

20 def test_failed_get_brand_store(self, mock_get_brand): 

21 mock_get_brand.side_effect = StoreApiResponseErrorList( 

22 "error", 400, [{"message": "error"}] 

23 ) 

24 

25 response = self.client.get("/admin/store/1/brand") 

26 data = response.get_json() 

27 

28 self.assertEqual(response.status_code, 200) 

29 self.assertFalse(data["success"]) 

30 self.assertEqual(data["message"], "error")