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)