def test_get_networks(self):
""" Test normal get_network workflow:
- call vim_client correctly.
- collect network types and translate them to thrift representation
correctly.
"""
vim_client = MagicMock()
vim_client.get_networks.return_value = ["VM Network", "VM Network 2"]
network_manager = NetworkManager(vim_client)
networks = network_manager.get_networks()
assert_that(networks, has_length(2))
# Verify 2 VM networks
assert_that(networks, has_item(Network("VM Network",
[NetworkType.VM])))
assert_that(networks, has_item(Network("VM Network 2",
[NetworkType.VM])))
