def test_import_not_admin(self, mock_xsrf_token, mock_importyaml):
self.login_endpoints_user()
request = datastore_messages.ImportYamlRequest(yaml='fake_yaml')
self.assertRaises(endpoints.ForbiddenException,
self.service.datastore_import, request)
self.assertFalse(mock_xsrf_token.called)
def test_import(self, mock_xsrf_token, mock_importyaml):
self.login_admin_endpoints_user()
request = datastore_messages.ImportYamlRequest(yaml='fake_yaml')
self.service.datastore_import(request)
self.assertEqual(mock_xsrf_token.call_count, 1)
mock_importyaml.assert_called_once_with('fake_yaml', wipe=False)
user_model.User.get_user(loanertest.USER_EMAIL).key.delete()
def testImportYamlRequest(self):
import_yaml_req = datastore_messages.ImportYamlRequest(
yaml='FAKE-YAML.yaml')
self.assertEqual(import_yaml_req.yaml, 'FAKE-YAML.yaml')