class TestKeysService(TestCase): def setUp(self): self.ks = Keys() def test_LIST(self, request_method): request_method.return_value = mock_response_result() self.ks.list().all() self.assertEqual(request_method.call_args[0], ('get', _('user/key'))) def test_GET(self, request_method): request_method.return_value = mock_response() self.ks.get(1) self.assertEqual(request_method.call_args[0], ('get', _('user/keys/1'))) def test_ADD_with_required(self, request_method): request_method.return_value = mock_response('post') self.ks.add({'key': 'ssh-rsa ...', 'title': 'test'}) self.assertEqual(request_method.call_args[0], ('post', _('user/keys'))) def test_ADD_without_required(self, request_method): self.assertRaises(ValidationError, self.ks.add, {}) def test_UPDATE_with_required(self, request_method): request_method.return_value = mock_response('patch') self.ks.update(1, {'key': 'ssh-rsa ...', 'title': 'test'}) self.assertEqual(request_method.call_args[0], ('patch', _('user/keys/1'))) def test_UPDATE_without_required(self, request_method): self.assertRaises(ValidationError, self.ks.update, 1, {}) def test_DELETE(self, request_method): request_method.return_value = mock_response('delete') self.ks.delete(1) self.assertEqual(request_method.call_args[0], ('delete', _('user/keys/1')))
def setUp(self): self.ks = Keys()