class TestResourceTracker(base.TestCase): """Test Agent ResourceTracker """ def setUp(self): super(TestResourceTracker, self).setUp() self.host = CONF.host self.cond_api = cond_api.ConductorAPI() self.rt = ResourceTracker(self.host, self.cond_api) def test_update_usage(self): """Update the resource usage and stats after a change in an instance """ # FIXME(Shaohe Feng) need add testcase. How to check the fpgas # has stored into DB by conductor correctly? pass def test_initialize_acc_drivers(self): enabled_drivers = ['intel_fpga_driver'] self.rt._initialize_drivers(enabled_drivers=enabled_drivers) drivers = self.rt.acc_drivers self.assertEqual(len(drivers), len(enabled_drivers)) def test_initialize_invalid_driver(self): enabled_drivers = ['invalid_driver'] self.assertRaises(exception.InvalidDriver, self.rt._initialize_drivers, enabled_drivers)
class TestResourceTracker(base.TestCase): """Test Agent ResourceTracker """ def setUp(self): super(TestResourceTracker, self).setUp() self.host = CONF.host self.cond_api = cond_api.ConductorAPI() self.rt = ResourceTracker(self.host, self.cond_api) def test_update_usage(self): """Update the resource usage and stats after a change in an instance """ # FIXME(Shaohe Feng) need add testcase. How to check the fpgas # has stored into DB by conductor correctly? pass def test_initialize_acc_drivers(self): enabled_drivers = ['intel_fpga_driver'] self.rt._initialize_drivers(enabled_drivers=enabled_drivers) drivers = self.rt.acc_drivers self.assertEqual(len(drivers), len(enabled_drivers)) def test_initialize_invalid_driver(self): enabled_drivers = ['invalid_driver'] self.assertRaises(exception.InvalidDriver, self.rt._initialize_drivers, enabled_drivers) @mock.patch('cyborg.agent.resource_tracker.LOG') def test_update_usage_failed_parent_provider(self, mock_log): with mock.patch.object(self.rt.conductor_api, 'report_data') as m: m.side_effect = exception.PlacementResourceProviderNotFound( resource_provider='foo') self.rt.update_usage(None) m.assert_called_once_with(None, 'fake-mini', []) mock_log.error.assert_called_once_with('Unable to report usage: %s', m.side_effect)