def test_should_write_no_dir(self, mock_exists, mock_mkdir, mock_pwn, mock_chown): mock_exists.side_effect = [False, False] self.assertTrue(host.should_write_ssh_config()) mock_pwn.assert_called_once() mock_mkdir.assert_called_once() mock_chown.assert_called_once()
def main(): parser = argparse.ArgumentParser( description='Manage docker containers that wrap a ChromeOS device.') parser.add_argument('-v', '--verbose', action='store_true', help='Enable verbose logging.') parser.add_argument( 'path_to_device_list', help='Path to json file containing list of CrOS device hostnames. Each ' 'device will be granted a container.') main_helpers.add_launch_arguments(parser) args = parser.parse_args() log_prefix = '%d' % os.getpid() main_helpers.configure_logging('cros_containers.log', log_prefix, args.verbose) docker_client = containers.CrosDockerClient() if not docker_client.ping(): logging.error('Docker engine unresponsive. Quitting early.') return 1 if host.should_write_ssh_config(): host.write_ssh_config() devices = host.read_device_list(args.path_to_device_list) container_descriptors = [ containers.CrosContainerDescriptor(d, host.SSH_IDENTITY_FILE_PATH) for d in devices ] main_helpers.launch_containers(docker_client, container_descriptors, args) return 0
def test_should_write_wrong_contents(self, mock_exists): mock_exists.return_value = True with mock.patch('__builtin__.open', mock.mock_open(read_data='this aint right')): self.assertTrue(host.should_write_ssh_config())
def test_should_write_no_file(self, mock_exists): mock_exists.side_effect = [True, False] self.assertTrue(host.should_write_ssh_config())
def test_should_write(self, mock_exists): mock_exists.side_effect = [True, True] with mock.patch('__builtin__.open', mock.mock_open(read_data=host.SSH_CONFIG_FILE_CONTENTS)): self.assertFalse(host.should_write_ssh_config())
def test_should_write_no_file(self, mock_exists): mock_exists.return_value = False self.assertTrue(host.should_write_ssh_config())
def test_should_write(self, mock_exists): mock_exists.return_value = True with mock.patch( '__builtin__.open', mock.mock_open(read_data=host.SSH_CONFIG_FILE_CONTENTS)): self.assertFalse(host.should_write_ssh_config())