Coverage for tests/endpoints/tests_dns_verified_status.py: 100%

1import json 

2from json import dumps as _dumps 

3from unittest.mock import patch, Mock, MagicMock 

4from tests.endpoints.endpoint_testing import TestEndpoints 

5 

6 

7class TestDnsVerifiedStatus(TestEndpoints): 

8 @patch("webapp.helpers.get_dns_verification_token") 

9 @patch( 

10 "canonicalwebteam.store_api.devicegw.DeviceGW.get_item_details", 

11 Mock(return_value={"links": {}}), 

12 ) 

13 def test_dns_verified_status(self, mock_get_dns_verification_token): 

14 def dumps_wrapper(*args, **kwargs): 

15 return _dumps( 

16 *args, 

17 **(kwargs | {"default": lambda obj: json.dumps({"links": {}})}) 

18 ) 

19 

20 json.dumps = MagicMock(wraps=dumps_wrapper) 

21 

22 mock_get_dns_verification_token.return_value = "test-token" 

23 

24 response = self.client.get("/api/store/test-store-id") 

25 

26 self.assertEqual(response.status_code, 200)