def testStoreObjectInGCS_NotResizableMimeType(self): guid = 'aaaaa' project_id = 100 object_id = '/%s/attachments/%s' % (project_id, guid) bucket_name = 'test_bucket' object_path = '/' + bucket_name + object_id mime_type = 'not_resizable_mime_type' content = 'content' self.mox.StubOutWithMock(app_identity, 'get_default_gcs_bucket_name') app_identity.get_default_gcs_bucket_name().AndReturn(bucket_name) self.mox.StubOutWithMock(uuid, 'uuid4') uuid.uuid4().AndReturn(guid) self.mox.StubOutWithMock(cloudstorage, 'open') cloudstorage.open(object_path, 'w', mime_type).AndReturn(fake.FakeFile()) self.mox.ReplayAll() ret_id = gcs_helpers.StoreObjectInGCS(content, mime_type, project_id, gcs_helpers.DEFAULT_THUMB_WIDTH, gcs_helpers.DEFAULT_THUMB_HEIGHT) self.mox.VerifyAll() self.assertEquals(object_id, ret_id)
def testStoreLogoInGCS(self): file_name = 'test_file.png' mime_type = 'image/png' content = 'test content' project_id = 100 object_id = 123 self.mox.StubOutWithMock(filecontent, 'GuessContentTypeFromFilename') filecontent.GuessContentTypeFromFilename(file_name).AndReturn( mime_type) self.mox.StubOutWithMock(gcs_helpers, 'StoreObjectInGCS') gcs_helpers.StoreObjectInGCS( content, mime_type, project_id, thumb_width=gcs_helpers.LOGO_THUMB_WIDTH, thumb_height=gcs_helpers.LOGO_THUMB_HEIGHT).AndReturn(object_id) self.mox.ReplayAll() ret_id = gcs_helpers.StoreLogoInGCS(file_name, content, project_id) self.mox.VerifyAll() self.assertEquals(object_id, ret_id)