Open Source Monorail

appengine/monorail/project/test/projectexport_test.py

Index: appengine/monorail/project/test/projectexport_test.py
diff --git a/appengine/monorail/project/test/projectexport_test.py b/appengine/monorail/project/test/projectexport_test.py
new file mode 100644
index 0000000000000000000000000000000000000000..56e4c6319a0fb1cf2dcd8a924c48d1785332290a
--- /dev/null
+++ b/appengine/monorail/project/test/projectexport_test.py
@@ -0,0 +1,29 @@
+# Copyright 2016 The Chromium Authors. All rights reserved.
+# Use of this source code is govered by a BSD-style
+# license that can be found in the LICENSE file or at
+# https://developers.google.com/open-source/licenses/bsd
+
+"""Unittests for the projectexport servlet."""
+
+import unittest
+
+from framework import permissions
+from services import service_manager
+from testing import testing_helpers
+from project import projectexport
+
+
+class ProjectExportTest(unittest.TestCase):
+
+  def setUp(self):
+    self.services = service_manager.Services()
+    self.servlet = projectexport.ProjectExport(
+        'req', 'res', services=self.services)
+
+  def testAssertBasePermission(self):
+    mr = testing_helpers.MakeMonorailRequest(
+        perms=permissions.OWNER_ACTIVE_PERMISSIONSET)
+    self.assertRaises(permissions.PermissionException,
+                      self.servlet.AssertBasePermission, mr)
+    mr.auth.user_pb.is_site_admin = True
+    self.servlet.AssertBasePermission(mr)