1 files changed, 71 insertions, 0 deletions
diff --git a/test/appengine/__init__.py b/test/appengine/__init__.py
new file mode 100644
index 0000000..917544d
--- /dev/null
+++ b/test/appengine/__init__.py
@@ -0,0 +1,71 @@
+import os
+import sys
+import unittest
+from nose.plugins.skip import SkipTest
+
+
+def activate_sandbox():
+    """
+    Enables parts of the GAE sandbox that are relevant.
+
+    Inserts the stub module import hook which causes the usage of appengine-specific
+    httplib, httplib2, socket, etc.
+    """
+    from google.appengine.tools.devappserver2.python import sandbox
+
+    for name in list(sys.modules):
+        if name in sandbox.dist27.MODULE_OVERRIDES:
+            del sys.modules[name]
+    sys.meta_path.insert(0, sandbox.StubModuleImportHook())
+    sys.path_importer_cache = {}
+
+
+def deactivate_sandbox():
+    from google.appengine.tools.devappserver2.python import sandbox
+
+    sys.meta_path = [
+        x for x in sys.meta_path if not isinstance(x, sandbox.StubModuleImportHook)]
+    sys.path_importer_cache = {}
+
+    # Delete any instances of sandboxed modules.
+    for name in list(sys.modules):
+        if name in sandbox.dist27.MODULE_OVERRIDES:
+            del sys.modules[name]
+
+
+class AppEngineSandboxTest(unittest.TestCase):
+
+    @classmethod
+    def setUpClass(cls):
+
+        if sys.version_info[:2] != (2, 7):
+            raise SkipTest("App Engine only tests on py2.7")
+
+        if 'APPLICATION_ID' not in os.environ:
+            raise SkipTest("NoseGAE plugin not used.")
+
+        try:
+            activate_sandbox()
+        except ImportError:
+            raise SkipTest("App Engine SDK not available.")
+
+    @classmethod
+    def tearDownClass(self):
+        try:
+            deactivate_sandbox()
+        except ImportError:
+            pass
+
+
+class MockResponse(object):
+    def __init__(self, content, status_code, content_was_truncated, final_url, headers):
+        import httplib
+        from StringIO import StringIO
+
+        self.content = content
+        self.status_code = status_code
+        self.content_was_truncated = content_was_truncated
+        self.final_url = final_url
+        self.header_msg = httplib.HTTPMessage(StringIO(''.join(
+            ["%s: %s\n" % (k, v) for k, v in headers.iteritems()] + ["\n"])))
+        self.headers = self.header_msg.items()