def stub_glance_api(self, count=1):
self.mox.StubOutWithMock(api, "glance_api")
glance_api = self.mox.CreateMock(glance_client.Client)
glance_api.token = TEST_TOKEN
for i in range(count):
api.glance_api(IsA(http.HttpRequest)).AndReturn(glance_api)
return glance_api
def index(request, tenant_id):
tenant = api.get_tenant(request, request.user.tenant)
all_images = []
try:
all_images = api.glance_api(request).get_images_detailed()
if not all_images:
messages.info(request, "There are currently no images.")
except GlanceClientConnectionError, e:
messages.error(request, "Error connecting to glance: %s" % e.message)
def test_get_glance_api(self):
self.mox.StubOutClassWithMocks(glance_client, 'Client')
glance_client.Client(TEST_HOSTNAME, TEST_PORT)
self.mox.StubOutWithMock(api, 'url_for')
api.url_for(IsA(http.HttpRequest), 'glance').AndReturn(TEST_URL)
self.mox.ReplayAll()
self.assertIsNotNone(api.glance_api(self.request))
self.mox.VerifyAll()
def test_get_glance_api(self):
self.mox.StubOutClassWithMocks(glance_client, "Client")
client_instance = glance_client.Client(TEST_HOSTNAME, TEST_PORT, auth_tok=TEST_TOKEN)
# Normally ``auth_tok`` is set in ``Client.__init__``, but mox doesn't
# duplicate that behavior so we set it manually.
client_instance.auth_tok = TEST_TOKEN
self.mox.StubOutWithMock(api, "url_for")
api.url_for(IsA(http.HttpRequest), "glance").AndReturn(TEST_URL)
self.mox.ReplayAll()
ret_val = api.glance_api(self.request)
self.assertIsNotNone(ret_val)
self.assertEqual(ret_val.auth_tok, TEST_TOKEN)
self.mox.VerifyAll()
def index(request):
for f in (DeleteImage, ToggleImage):
_, handled = f.maybe_handle(request)
if handled:
return handled
# We don't have any way of showing errors for these, so don't bother
# trying to reuse the forms from above
delete_form = DeleteImage()
toggle_form = ToggleImage()
images = []
try:
images = api.glance_api(request).get_images_detailed()
if not images:
messages.info(request, "There are currently no images.")
except GlanceClientConnectionError, e:
messages.error(request, "Error connecting to glance: %s" % e.message)
def handle(self, request, data):
try:
api.glance_api(request).update_image(image_id, image_meta={"is_public": False})
except GlanceClientConnectionError, e:
messages.error(request, "Error connecting to glance: %s" % e.message)
def handle(self, request, data):
image_id = data["image_id"]
try:
api.glance_api(request).delete_image(image_id)
except GlanceClientConnectionError, e:
messages.error(request, "Error connecting to glance: %s" % e.message)
