def testListTemplatesResponse(self):
tmpls = [
template_messages.Template(name='TMPL-NAME-1'),
template_messages.Template(name='TMPL-NAME-2')
]
list_tmpl_resp = template_messages.ListTemplatesResponse(
templates=tmpls)
self.assertListEqual(list_tmpl_resp.templates, tmpls)
def testTemplate(self):
template = template_messages.Template(name='TMPL-NAME',
body='TMPL-BODY-CONTENT',
title='TMPL-TITLE')
self.assertEqual(template.name, 'TMPL-NAME')
self.assertEqual(template.body, 'TMPL-BODY-CONTENT')
self.assertEqual(template.title, 'TMPL-TITLE')
def test_create_bad_request(self):
"""Test create raises BadRequestException with required fields missing."""
request = template_messages.CreateTemplateRequest(
template=template_messages.Template(
name='',
body='',
title='test_title'))
with self.assertRaises(endpoints.BadRequestException):
self.service.create(request)
def list(self, request):
"""Gets a list of all template values."""
self.check_xsrf_token(self.request_state)
response_message = []
for template in template_model.Template.get_all():
response_message.append(
template_messages.Template(name=template.name,
body=template.body,
title=template.title))
return template_messages.ListTemplatesResponse(
templates=response_message)
def test_create(self):
request = template_messages.CreateTemplateRequest(
template=template_messages.Template(
name='test_create_template',
body='test create body',
title='test create title'))
with mock.patch.object(
self.service, 'check_xsrf_token', autospec=True) as mock_xsrf_token:
response = self.service.create(request)
self.assertEqual(mock_xsrf_token.call_count, 1)
self.assertIsInstance(response, message_types.VoidMessage)
def test_create_duplicate_name(self):
request = template_messages.CreateTemplateRequest(
template=template_messages.Template(
name='test_create_template',
body='test create body',
title='test create title'))
with mock.patch.object(
self.service, 'check_xsrf_token', autospec=True) as mock_xsrf_token:
response = self.service.create(request)
self.assertEqual(mock_xsrf_token.call_count, 1)
self.assertIsInstance(response, message_types.VoidMessage)
request = template_messages.CreateTemplateRequest(
template=template_messages.Template(
name='test_create_template',
body='test second body',
title='test second title'))
with mock.patch.object(
self.service, 'check_xsrf_token', autospec=True) as mock_xsrf_token:
with self.assertRaises(endpoints.BadRequestException):
response = self.service.create(request)
def testCreateTemplateRequest(self):
tmpl = template_messages.Template(name='TMPL-NAME')
create_tmpl_req = template_messages.CreateTemplateRequest(
template=tmpl)
self.assertEqual(create_tmpl_req.template, tmpl)