def test_destroy_tag(self): request = tag_messages.TagRequest( urlsafe_key=self.hidden_tag.key.urlsafe()) with mock.patch.object(self.service, 'check_xsrf_token', autospec=True) as mock_xsrf_token: response = self.service.destroy(request) self.assertEqual(mock_xsrf_token.call_count, 1) self.assertIsNone(tag_model.Tag.get(self.hidden_tag.key.urlsafe())) self.assertIsInstance(response, message_types.VoidMessage)
def test_get_tag(self): request = tag_messages.TagRequest( urlsafe_key=self.test_tag.key.urlsafe()) expected_response = tag_messages.Tag( name=self.test_tag.name, hidden=self.test_tag.hidden, color=self.test_tag.color, protect=self.test_tag.protect, description=self.test_tag.description) with mock.patch.object(self.service, 'check_xsrf_token', autospec=True) as mock_xsrf_token: response = self.service.get(request) self.assertEqual(mock_xsrf_token.call_count, 1) self.assertEqual(response, expected_response)
def test_get_tag_bad_request(self): request = tag_messages.TagRequest(urlsafe_key='fake_urlsafe_key') with self.assertRaises(endpoints.BadRequestException): self.service.get(request)
def test_destroy_not_existing(self): with self.assertRaises(endpoints.BadRequestException): self.service.destroy( tag_messages.TagRequest(urlsafe_key='nonexistent_tag'))
def test_destroy_protected(self): with self.assertRaises(endpoints.BadRequestException): self.service.destroy( tag_messages.TagRequest( urlsafe_key=self.protected_tag.key.urlsafe()))
def testTagRequest(self): tag_req = tag_messages.TagRequest(urlsafe_key='FAKE-URL-KEY') self.assertEqual(tag_req.urlsafe_key, 'FAKE-URL-KEY')