def test_search_for_now_raises_ValueError_on_extra_results(self): image = testutil.cloud_object_mock(1) self.driver_mock().list_images.return_value = [image, image] driver = driver_base.BaseComputeNodeDriver({}, {}, {}, self.driver_mock) with self.assertRaises(ValueError) as test: driver.search_for_now('id_1', 'list_images')
def test_search_for_now_uses_public_method(self): image = testutil.cloud_object_mock(1) self.driver_mock().list_images.return_value = [image] driver = driver_base.BaseComputeNodeDriver({}, {}, {}, self.driver_mock) self.assertIs(image, driver.search_for_now('id_1', 'list_images')) self.assertEqual(1, self.driver_mock().list_images.call_count)
def test_search_for_now_uses_private_method(self): net = testutil.cloud_object_mock(1) self.driver_mock().ex_list_networks.return_value = [net] driver = driver_base.BaseComputeNodeDriver({}, {}, {}, self.driver_mock) self.assertIs(net, driver.search_for_now('id_1', 'ex_list_networks')) self.assertEqual(1, self.driver_mock().ex_list_networks.call_count)
def test_search_for_returns_cached_results(self): image1 = testutil.cloud_object_mock(1) image2 = testutil.cloud_object_mock(1) self.driver_mock().list_images.side_effect = [[image1], [image2]] driver = driver_base.BaseComputeNodeDriver({}, {}, {}, self.driver_mock) self.assertIs(driver.search_for('id_1', 'list_images'), driver.search_for('id_1', 'list_images')) self.assertEqual(1, self.driver_mock().list_images.call_count)
def test_search_for_now_raises_ValueError_on_zero_results(self): self.driver_mock().list_images.return_value = [] driver = driver_base.BaseComputeNodeDriver({}, {}, {}, self.driver_mock) with self.assertRaises(ValueError) as test: driver.search_for_now('id_1', 'list_images')