def test_get_transforms( self, mocked_load_transforms_config: MagicMock, mocked_load_transforms_filter_config: MagicMock, ) -> None: """Test that get_transforms works.""" mocked_load_transforms_config.return_value = [ { "name": "framework_foo", "help": None, "params": [ {"name": "transform1"}, {"name": "transform2"}, {"name": "transform3"}, ], }, { "name": "framework_bar", "help": None, "params": [ {"name": "transform4"}, {"name": "transform5"}, {"name": "transform6"}, ], }, ] mocked_load_transforms_filter_config.return_value = { "framework_foo": { "domain_foo": [ "transform1", "transform2", "transform3", "transform4", ], "domain_bar": [ "transform2", "transform3", "transform4", ], }, } expected = [ {"name": "transform2"}, {"name": "transform3"}, ] feeder = Feeder( data={ "config": { "framework": "framework_foo", "domain": "domain_bar", }, }, ) actual = feeder.get_transforms() mocked_load_transforms_config.assert_called_once() mocked_load_transforms_filter_config.assert_called_once() self.assertEqual(expected, actual)
def test_get_transforms_for_unknown_framework( self, mocked_load_transforms_config: MagicMock, ) -> None: """Test that get_transforms works when unknown framework requested.""" mocked_load_transforms_config.return_value = [ { "name": "framework_foo", "help": None, "params": [ { "name": "transform1" }, { "name": "transform2" }, { "name": "transform3" }, ], }, { "name": "framework_bar", "help": None, "params": [ { "name": "transform4" }, { "name": "transform5" }, { "name": "transform6" }, ], }, ] feeder = Feeder(data={ "config": { "framework": "framework_unknown", }, }, ) actual = feeder.get_transforms() mocked_load_transforms_config.assert_called_once() self.assertEqual([], actual)
def test_get_transforms_fails_without_framework(self) -> None: """Test that get_transforms fails when no config given.""" feeder = Feeder(data={}) with self.assertRaisesRegex(ClientErrorException, "Framework not set."): feeder.get_transforms()