become_ready tests

meejah · Sep 21, 2017 · 0601710 · 0601710
1 parent 9025173
commit 0601710
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/test/test_controller.py b/test/test_controller.py
@@ -1097,6 +1097,7 @@ def setUp(self):
         self.proto = Mock()
         self.cfg = Mock()
         self.tor = Tor(reactor, self.proto, _tor_config=self.cfg)
+        self.tor.dns_resolve = Mock()
 
     @defer.inlineCallbacks
     def test_ready(self):
@@ -1108,6 +1109,27 @@ def test_ready(self):
         ready = yield self.tor.is_ready()
         self.assertTrue(ready, "should be ready")
 
+    @defer.inlineCallbacks
+    def test_become_ready_already(self):
+        self.proto.get_info = Mock(return_value={
+            "dormant": "0",
+            "status/enough-dir-info": "1",
+            "status/circuit-established": "1",
+        })
+        ready = yield self.tor.become_ready()
+        self.assertTrue(
+            self.tor.dns_resolve.mock_calls == []
+        )
+
+    @defer.inlineCallbacks
+    def test_become_ready_asleep(self):
+        self.proto.get_info = Mock(return_value={
+            "dormant": "1",
+            "status/enough-dir-info": "1",
+            "status/circuit-established": "1",
+        })
+        ready = yield self.tor.become_ready()
+        self.assertEqual(1, len(self.tor.dns_resolve.mock_calls))
 
 
 class TorStreamTests(unittest.TestCase):