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

1from unittest.mock import patch 

2 

3from tests.endpoints.endpoint_testing import TestEndpoints 

4 

5 

6class TestMacaroonInfo(TestEndpoints): 

7 @patch("canonicalwebteam.store_api.publishergw.PublisherGW.macaroon_info") 

8 def test_get_macaroon_info(self, mock_macaroon_info): 

9 mock_macaroon_info.return_value = { 

10 "account-id": "test-account-id", 

11 "permissions": ["package_access"], 

12 } 

13 

14 response = self.client.get("/api/whoami") 

15 data = response.get_json() 

16 

17 self.assertEqual(response.status_code, 200) 

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

19 self.assertEqual( 

20 data["data"], 

21 { 

22 "account-id": "test-account-id", 

23 "permissions": ["package_access"], 

24 }, 

25 ) 

26 mock_macaroon_info.assert_called_once_with("test_macaroon")