test for poll state (red)

1 files changed, 21 insertions, 0 deletions

diff --git a/src/leap/eip/test_conductor.py b/src/leap/eip/test_conductor.py
index 051bac5..27460d5 100644
--- a/src/leap/eip/test_conductor.py
+++ b/src/leap/eip/test_conductor.py
@@ -129,6 +129,27 @@ class EIPConductorTest(unittest.TestCase):
         self.con.connect()
         self.con._launch_openvpn.assert_called_once_with()
 
+    # XXX tests breaking here ...
+
+    def test_good_poll_connection_state(self):
+        """
+        """
+        #@patch --
+        # self.manager.get_connection_state
+        state = self.con.poll_connection_state()
+        # too whitebox??
+        #self.con.status.set_vpn_state.assert_called_with(status_step)
+        self.assertEqual(state, 'connected')
+
+    def test_bad_poll_connection_state(self):
+        """
+        """
+        #@patch --
+        # self.manager.get_connection_state
+        # but raise ConnectionRefusedError
+        state = self.con.poll_connection_state()
+        self.assertEqual(state, None)
+
 
     # XXX more things to test:
     # - called config routinges during initz.