initial import

2 files changed, 21 insertions, 0 deletions

diff --git a/src/leap/tests/mocks/__init__.py b/src/leap/tests/mocks/__init__.py
new file mode 100644
index 00000000..06f96870
--- /dev/null
+++ b/src/leap/tests/mocks/__init__.py
@@ -0,0 +1 @@
+import manager
diff --git a/src/leap/tests/mocks/manager.py b/src/leap/tests/mocks/manager.py
new file mode 100644
index 00000000..564631cd
--- /dev/null
+++ b/src/leap/tests/mocks/manager.py
@@ -0,0 +1,20 @@
+from mock import Mock
+
+from eip_client.vpnmanager import OpenVPNManager
+
+vpn_commands = {
+        'status': [
+            'OpenVPN STATISTICS', 'Updated,Mon Jun 25 11:51:21 2012',
+            'TUN/TAP read bytes,306170', 'TUN/TAP write bytes,872102',
+            'TCP/UDP read bytes,986177', 'TCP/UDP write bytes,439329',
+            'Auth read bytes,872102'],
+        'state': ['1340616463,CONNECTED,SUCCESS,172.28.0.2,198.252.153.38'],
+        # XXX add more tests
+        }
+
+
+def get_openvpn_manager_mocks():
+    manager = OpenVPNManager()
+    manager.status = Mock(return_value='\n'.join(vpn_commands['status']))
+    manager.state = Mock(return_value=vpn_commands['state'][0])
+    return manager