From 111176f7681d7be7d9b12a30ac303447032a669e Mon Sep 17 00:00:00 2001 From: Aleksandr Liber Date: Tue, 20 Oct 2020 01:51:35 -0700 Subject: [PATCH] Fixes #262 Add test_virtual_initialize_return_none --- .../python/dlpx/virtualization/test_plugin.py | 22 +++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/platform/src/test/python/dlpx/virtualization/test_plugin.py b/platform/src/test/python/dlpx/virtualization/test_plugin.py index 078b54e7..8cfde728 100755 --- a/platform/src/test/python/dlpx/virtualization/test_plugin.py +++ b/platform/src/test/python/dlpx/virtualization/test_plugin.py @@ -705,6 +705,28 @@ def virtual_initialize_impl(virtual_source, repository): actual_source_config = initialize_response.return_value.source_config assert actual_source_config.parameters.json == TEST_REPOSITORY_JSON + @staticmethod + def test_virtual_initialize_return_none(my_plugin, virtual_source, + repository, source_config): + @my_plugin.virtual.initialize() + def virtual_initialize_impl(virtual_source, repository): + TestPlugin.assert_plugin_args(virtual_source=virtual_source, + repository=repository) + # Will return none. + + initialize_request = platform_pb2.InitializeRequest() + TestPlugin.setup_request(request=initialize_request, + virtual_source=virtual_source, + repository=repository) + + with pytest.raises(IncorrectReturnTypeError) as err_info: + my_plugin.virtual._internal_initialize(initialize_request) + message = err_info.value.message + assert message == ( + "The returned object for the virtual.initialize() operation was" + " type 'NoneType' but should be of class 'dlpx.virtualization." + "fake_generated_definitions.SourceConfigDefinition'.") + @staticmethod def test_virtual_mount_spec(my_plugin, virtual_source, repository):