def test_consumer_updates_version(self):
cv = version_manager.ResourceConsumerTracker()
for version in [TEST_RESOURCE_VERSION_A, TEST_RESOURCE_VERSION_B]:
cv.set_versions(CONSUMER_1, {TEST_RESOURCE_TYPE: version})
self.assertEqual(set([TEST_RESOURCE_VERSION_B]),
cv.get_resource_versions(TEST_RESOURCE_TYPE))
def test_multiple_consumer_version_update(self):
cv = version_manager.ResourceConsumerTracker()
cv.set_versions(CONSUMER_1, {TEST_RESOURCE_TYPE: TEST_VERSION_A})
cv.set_versions(CONSUMER_2, {TEST_RESOURCE_TYPE: TEST_VERSION_A})
cv.set_versions(CONSUMER_1, {TEST_RESOURCE_TYPE: TEST_VERSION_B})
self.assertEqual(set([TEST_VERSION_A, TEST_VERSION_B]),
cv.get_resource_versions(TEST_RESOURCE_TYPE))
def test_different_adds_triggers_recalculation(self):
cv = version_manager.ResourceConsumerTracker()
for version in [TEST_VERSION_A, TEST_VERSION_B]:
cv.set_versions(CONSUMER_1, {TEST_RESOURCE_TYPE: version})
self.assertTrue(cv._needs_recalculation)
cv._recalculate_versions = mock.Mock()
cv.get_resource_versions(TEST_RESOURCE_TYPE)
cv._recalculate_versions.assert_called_once_with()
def test_consumer_downgrades_stops_reporting(self):
cv = version_manager.ResourceConsumerTracker()
cv.set_versions(CONSUMER_1, {TEST_RESOURCE_TYPE: TEST_VERSION_B,
TEST_RESOURCE_TYPE_2: TEST_VERSION_A})
cv.set_versions(CONSUMER_1, {})
for resource_type in [TEST_RESOURCE_TYPE, TEST_RESOURCE_TYPE_2]:
self.assertEqual(set(),
cv.get_resource_versions(resource_type))
def test_consumer_downgrades_removing_resource(self):
cv = version_manager.ResourceConsumerTracker()
cv.set_versions(CONSUMER_1, {TEST_RESOURCE_TYPE: TEST_VERSION_B,
TEST_RESOURCE_TYPE_2: TEST_VERSION_A})
cv.set_versions(CONSUMER_1, {TEST_RESOURCE_TYPE: TEST_VERSION_A})
self.assertEqual(set(),
cv.get_resource_versions(TEST_RESOURCE_TYPE_2))
self.assertEqual(set([TEST_VERSION_A]),
cv.get_resource_versions(TEST_RESOURCE_TYPE))
def test_compatibility_liberty_sriov_and_ovs_agents(self):
def _fake_local_versions(self):
local_versions = collections.defaultdict(set)
local_versions[resources.QOS_POLICY].add('1.11')
return local_versions
for agent_type in version_manager.NON_REPORTING_AGENT_TYPES:
consumer_id = version_manager.AgentConsumer(
agent_type, AGENT_HOST_1)
cv = version_manager.ResourceConsumerTracker()
cv._get_local_resource_versions = _fake_local_versions
cv._versions = _fake_local_versions(mock.ANY)
cv.set_versions(consumer_id, {})
self.assertEqual(set(['1.0', '1.11']),
cv.get_resource_versions(resources.QOS_POLICY))
def test_consumer_set_versions(self):
cv = version_manager.ResourceConsumerTracker()
cv.set_versions(CONSUMER_1, {TEST_RESOURCE_TYPE: TEST_VERSION_A})
self.assertIn(TEST_VERSION_A,
cv.get_resource_versions(TEST_RESOURCE_TYPE))
