1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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
|
