Esempio n. 1
0
    def test_cleanup(self, mocks):
        subject = TrafficSshCommands(lambda: '1.2.3.4')
        self.mock_executor.return_value.execute_async.return_value = 'cleaned'

        result = subject.cleanup(['one', 'two'])

        self.mock_executor.return_value.execute_async.assert_called_with([
            'ssh', '-oStrictHostKeyChecking=no',
            '[email protected]',
            'one', 'two'
        ])
        assert result == 'cleaned'
Esempio n. 2
0
 def __init__(self, machine_name):
     self.client = boto3.client('ec2', EC2_CONFIG['region'])
     self.ip = ""
     self._machine_name = machine_name
     self.node = Node(machine_name, NodeSshCommands(self.get_external_ip))
     self.dns = Dns(machine_name, DnsSshCommands(self.get_external_ip))
     self.traffic = TrafficHandler(machine_name, TrafficSshCommands(self.get_external_ip))
Esempio n. 3
0
 def __init__(self, name):
     self.ip_pattern = re.compile(r".*value: (.+), time.*")
     self.command = "VBoxManage"
     self.ip = ""
     self.name = name
     self.node = Node(name, NodeSshCommands(self.get_external_ip))
     self.dns = Dns(name, DnsSshCommands(self.get_external_ip))
     self.traffic = TrafficHandler(name,
                                   TrafficSshCommands(self.get_external_ip))
Esempio n. 4
0
 def __init__(self,
              machine_name,
              project=COMPUTE_CONFIG['project'],
              zone=COMPUTE_CONFIG['zone']):
     self.compute = googleapiclient.discovery.build('compute', 'v1')
     self.project = project
     self.zone = zone
     self.ip = ""
     self._machine_name = machine_name
     self.node = Node(machine_name, NodeSshCommands(self.get_external_ip))
     self.dns = Dns(machine_name, DnsSshCommands(self.get_external_ip))
     self.traffic = TrafficHandler(machine_name,
                                   TrafficSshCommands(self.get_external_ip))
Esempio n. 5
0
    def test_init(self, mocks):
        subject = TrafficSshCommands(lambda: '1.2.3.4')

        assert subject.get_ip() == '1.2.3.4'