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')