diff options
| author | kali <kali@leap.se> | 2012-12-10 19:51:53 +0900 |
|---|---|---|
| committer | kali <kali@leap.se> | 2012-12-10 19:51:53 +0900 |
| commit | 38cc1758240a3c64db387b0437dcf1517b52da15 (patch) | |
| tree | 7b9f98d0ec3af6bdd8fa77b94e83ceddb2be340d /src/leap/eip/tests/test_eipconnection.py | |
| parent | cd78d9d552977e8f8fb12b6a2ff56fda9c37bf35 (diff) | |
cleanup and rewrite eipconnection/openvpnconnection classes
Diffstat (limited to 'src/leap/eip/tests/test_eipconnection.py')
| -rw-r--r-- | src/leap/eip/tests/test_eipconnection.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/src/leap/eip/tests/test_eipconnection.py b/src/leap/eip/tests/test_eipconnection.py index aefca36f..4ee5ae30 100644 --- a/src/leap/eip/tests/test_eipconnection.py +++ b/src/leap/eip/tests/test_eipconnection.py @@ -123,9 +123,14 @@ class EIPConductorTest(BaseLeapTest): self.con.status.CONNECTED) # disconnect - self.con.cleanup = Mock() + self.con.terminate_openvpn_connection = Mock() self.con.disconnect() - self.con.cleanup.assert_called_once_with() + self.con.terminate_openvpn_connection.assert_called_once_with( + shutdown=False) + self.con.terminate_openvpn_connection = Mock() + self.con.disconnect(shutdown=True) + self.con.terminate_openvpn_connection.assert_called_once_with( + shutdown=True) # new status should be disconnected # XXX this should evolve and check no errors |