1import responses 

2from tests.publisher.endpoint_testing import BaseTestCases 

3 

4 

5class GetRequestReservedNameNotAuth(BaseTestCases.EndpointLoggedOut): 

6 def setUp(self): 

7 endpoint_url = "/request-reserved-name" 

8 super().setUp(snap_name=None, endpoint_url=endpoint_url) 

9 

10 

11class GetRequestReservedName(BaseTestCases.BaseAppTesting): 

12 def setUp(self): 

13 self.store = "testing-store-id1" 

14 super().setUp( 

15 snap_name="test-snap", 

16 api_url=None, 

17 endpoint_url="/request-reserved-name", 

18 ) 

19 self.user_url = "https://dashboard.snapcraft.io/dev/api/account" 

20 self.user_payload = { 

21 "error_list": [], 

22 "stores": [ 

23 { 

24 "id": "testing-store-id1", 

25 "name": "test-store", 

26 "roles": ["admin", "review", "view", "access"], 

27 } 

28 ], 

29 } 

30 

31 @responses.activate 

32 def test_request_reserved_name_logged_in(self): 

33 self._log_in(self.client) 

34 responses.add( 

35 responses.GET, self.user_url, json=self.user_payload, status=200 

36 ) 

37 endpoint_url = "{}?snap-name={}&store={}".format( 

38 self.endpoint_url, self.snap_name, self.store 

39 ) 

40 response = self.client.get(endpoint_url) 

41 

42 self.assertEqual(response.status_code, 302) 

43 

44 @responses.activate 

45 def test_request_reserved_name_redirect_no_snap_name(self): 

46 self._log_in(self.client) 

47 responses.add( 

48 responses.GET, self.user_url, json=self.user_payload, status=200 

49 ) 

50 response = self.client.get(self.endpoint_url) 

51 

52 self.assertEqual(response.status_code, 302) 

53 self.assertEqual(response.headers["Location"], "/register-snap")