Fix bug 1884 MON openstack token optimization

[osm/MON.git] / osm_mon / tests / unit / collector / vnf_collectors / vmware / test_vcd_collector.py

diff --git a/osm_mon/tests/unit/collector/vnf_collectors/vmware/test_vcd_collector.py b/osm_mon/tests/unit/collector/vnf_collectors/vmware/test_vcd_collector.py
index 3cac999..3d92fd5 100644 (file)
--- a/osm_mon/tests/unit/collector/vnf_collectors/vmware/test_vcd_collector.py
+++ b/osm_mon/tests/unit/collector/vnf_collectors/vmware/test_vcd_collector.py
@@ -48,9 +48,10 @@ class CollectorTest(TestCase):
     @mock.patch("osm_mon.collector.vnf_collectors.vmware.CommonDbClient")
     def setUp(self, mock_db, mock_get_vim_account):
         super().setUp()
+        mock_vim_session = mock.Mock()
         mock_get_vim_account.return_value = VIM_ACCOUNT
         self.collector = VMwareCollector(
-            Config(), "9de6df67-b820-48c3-bcae-ee4838c5c5f4"
+            Config(), "9de6df67-b820-48c3-bcae-ee4838c5c5f4", mock_vim_session
         )
         self.mock_db = mock_db
         with open(
@@ -199,9 +200,10 @@ class VApp_Details_Test(TestCase):
     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 = VMwareCollector(
-            Config(), "9de6df67-b820-48c3-bcae-ee4838c5c5f4"
+            Config(), "9de6df67-b820-48c3-bcae-ee4838c5c5f4", mock_vim_session
         )
 
     def tearDown(self):
@@ -272,9 +274,10 @@ class Get_VM_Moref_Test(TestCase):
     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 = VMwareCollector(
-            Config(), "9de6df67-b820-48c3-bcae-ee4838c5c5f4"
+            Config(), "9de6df67-b820-48c3-bcae-ee4838c5c5f4", mock_vim_session
         )
 
     def tearDown(self):