def test_action_manager_calls_create_machines_with_create_action_and_machines_and_nohosts( self): manager = ActionManager(config_path="blaap", no_hosts=True) manager.machines = ["machine"] manager.execute("create") self.create_machines.assert_called_once_with( self.validator.updated_config, machines=["machine"])
def test_action_manager_calls_change_machine_status_with_stop_action_and_machines( self): manager = ActionManager(config_path="blaap") manager.machines = ["machine"] manager.execute("stop") self.change_machine_status.assert_called_once_with( self.validator.updated_config, machines=["machine"], status="stop")
def test_action_manager_calls_destroy_machines_with_destroy_action_and_machines( self): manager = ActionManager(config_path="blaap") manager.machines = ["machine"] manager.execute("destroy") self.destroy_machines.assert_called_once_with( self.validator.updated_config, machines=["machine"]) self.assertFalse(self.destroy_lxc_image.called)
def main(args=None): """ Program entry point :param list args: The pre-cooked arguments to pass to the ArgParser :return int: exit_code """ args = parse_args(args) # Set the VNET_FORCE variable, if --yes is given this will answer yes to all questions environ[settings.VNET_FORCE_ENV_VAR] = "true" if args.yes else "false" # Setup logging setup_console_logging(verbosity=DEBUG if args.verbose else INFO) # Most VNet operation require root. So, do a root check if not check_for_root_user(): logger.critical("This program should only be run as root") return EX_NOPERM # Let the action manager handle the rest manager = ActionManager(config_path=args.config, sniffer=args.sniffer, base_image=args.base_image, no_hosts=args.no_hosts) if args.machines: manager.machines = args.machines return manager.execute(args.action)
def test_action_manager_has_machine_property(self): manager = ActionManager() manager.machines = ["1", "2", "3"] self.assertEqual(manager.machines, ["1", "2", "3"]) self.assertIsInstance(ActionManager.machines, property)
def test_action_manager_does_not_call_delete_vnet_interfaces_with_destroy_action_and_machines( self): manager = ActionManager(config_path="blaap") manager.machines = ["machine"] manager.execute("destroy") self.assertFalse(self.delete_vnet_interfaces.called)
def test_action_manager_does_not_call_bring_down_vnet_interfaces_with_stop_action_and_machines( self): manager = ActionManager(config_path="blaap") manager.machines = ["machine"] manager.execute("stop") self.assertFalse(self.bring_down_vnet_interfaces.called)