def setUp(self, mock_db, mock_get_vim_account):
super().setUp()
self.mock_db = mock_db
+ mock_vim_session = mock.Mock()
mock_get_vim_account.return_value = VIM_ACCOUNT
- self.collector = VIOCollector(Config(), "9de6df67-b820-48c3-bcae-ee4838c5c5f4")
+ self.collector = VIOCollector(
+ Config(), "9de6df67-b820-48c3-bcae-ee4838c5c5f4", mock_vim_session
+ )
with open(
os.path.join(os.path.dirname(__file__), "osm_mocks", "VNFR.json"), "r"
) as f: