def test_files_get_module_name(self): """ Test getting module name from path """ files_view = FilesView() module = 'test-module' formatted_module_name = 'test/module' module_name = files_view._get_module_name(module) self.assertEqual(formatted_module_name, module_name)
def test_files_get_bad_credentials(self): """ Test getting improperly formatted credentials from header """ files_view = FilesView() encoded_creds = base64.encodestring('blah') headers = {'HTTP_AUTHORIZATION': encoded_creds} creds = files_view._get_credentials(headers) self.assertEqual(creds, None)
def test_files_get_module_slug_name(self): """ Test getting module slug from path """ files_view = FilesView() module = 'test-module' path = '/v3/files/test-module-1.2.3.tar.gz' module_name = files_view._get_module_slug_name(path) self.assertEqual(module, module_name)
def test_files_get_credentials(self): """ Test getting credentials from header """ files_view = FilesView() real_creds = ('test', '123') encoded_creds = base64.encodestring('test:123') headers = {'HTTP_AUTHORIZATION': encoded_creds} creds = files_view._get_credentials(headers) self.assertEqual(real_creds, creds)
def test_files_get_module_version(self): """ Test getting module version from path """ files_view = FilesView() version = '1.2.3' formatted_module_name = 'module' path = '/v3/files/test-module-1.2.3.tar.gz' get_version = files_view._get_module_version(path) self.assertEqual(version, get_version)
def test_files_bad_resource_type(self, mock_get_credentials, mock_get_module_name): """ Test that only consumer or repository resource type is allowed """ mock_get_module_name.return_value = 'fake-module' mock_get_credentials.return_value = () mock_request = mock.MagicMock() files_view = FilesView() response = files_view.get(mock_request, resource_type='foo') self.assertEqual(response.status_code, 404)
def test_files_missing_auth(self, mock_get_credentials, mock_get_module_name): """ Test that 401 is returned when basic auth is not used for pre 3.3 """ mock_get_module_name.return_value = 'fake-module' mock_get_credentials.return_value = () mock_request = mock.MagicMock() files_view = FilesView() response = files_view.get(mock_request) self.assertEqual(response.status_code, 401)
def test_files_missing_module(self, mock_get_credentials, mock_get_module_name): """ Test that proper response is returned when module name is not specified """ mock_get_module_name.return_value = '' mock_get_credentials.return_value = () mock_request = mock.MagicMock() files_view = FilesView() response = files_view.get(mock_request, resource_type='repository', resource='repo-id') self.assertEqual(response.status_code, 400) self.assertEqual(response.content, 'Module name is missing.')