def test_get_image(): kwargs = {'image': 'kathara/frr'} device = Machine(Lab("test_lab"), "test_machine", **kwargs) assert device.get_image() == "kathara/frr"
def test_get_image_default(mock_setting_get_instance, default_device: Machine): setting_mock = Mock() setting_mock.configure_mock(**{'image': "kathara/frr"}) mock_setting_get_instance.return_value = setting_mock assert default_device.get_image() == "kathara/frr"