Testing compute_engine_id() in _implicit_environ.

dhermes · dhermes · commit 12e39dbc4d43 · 2015-02-24T18:14:18.000-08:00
Also fixing unused dataset_id=None in _callFUT in
Test__get_gcd_dataset_id and Test__get_gcd_dataset_id.
diff --git a/gcloud/datastore/test__implicit_environ.py b/gcloud/datastore/test__implicit_environ.py
@@ -67,7 +67,7 @@ def test_preset(self):
 
 class Test__get_production_dataset_id(unittest2.TestCase):
 
-    def _callFUT(self, dataset_id=None):
+    def _callFUT(self):
         from gcloud.datastore import _implicit_environ
         return _implicit_environ._get_production_dataset_id()
 
@@ -94,7 +94,7 @@ def test_value_set(self):
 
 class Test__get_gcd_dataset_id(unittest2.TestCase):
 
-    def _callFUT(self, dataset_id=None):
+    def _callFUT(self):
         from gcloud.datastore import _implicit_environ
         return _implicit_environ._get_gcd_dataset_id()
 
@@ -146,6 +146,43 @@ def test_value_set(self):
             self.assertEqual(dataset_id, APP_ENGINE_ID)
 
 
+class Test_compute_engine_id(unittest2.TestCase):
+
+    def _callFUT(self):
+        from gcloud.datastore import _implicit_environ
+        return _implicit_environ.compute_engine_id()
+
+    def _monkeyConnection(self, connection):
+        from gcloud._testing import _Monkey
+        from gcloud.datastore import _implicit_environ
+
+        def _factory(host, timeout):
+            connection.host = host
+            connection.timeout = timeout
+            return connection
+
+        return _Monkey(_implicit_environ, HTTPConnection=_factory)
+
+    def test_bad_status(self):
+        connection = _HTTPConnection(404, None)
+        with self._monkeyConnection(connection):
+            dataset_id = self._callFUT()
+            self.assertEqual(dataset_id, None)
+
+    def test_success(self):
+        COMPUTE_ENGINE_ID = object()
+        connection = _HTTPConnection(200, COMPUTE_ENGINE_ID)
+        with self._monkeyConnection(connection):
+            dataset_id = self._callFUT()
+            self.assertEqual(dataset_id, COMPUTE_ENGINE_ID)
+
+    def test_socket_raises(self):
+        connection = _TimeoutHTTPConnection()
+        with self._monkeyConnection(connection):
+            dataset_id = self._callFUT()
+            self.assertEqual(dataset_id, None)
+
+
 class Test__determine_default_dataset_id(unittest2.TestCase):
 
     def _callFUT(self, dataset_id=None):
