def test_convert_speaker_list_full_image_to_mimas_image(self):
data = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
s1 = self.mk_speaker_with_image("mail@email", data)
s2 = self.mk_speaker_with_image("mail2@email", data)
s3 = self.mk_speaker_with_image("mail3@email", data)
spk_list = speaker.Speaker.query().fetch(keys_only=True)
image.convert_speaker_list_full_image_to_mimas_image(spk_list)
self.assertEquals(None, s1.fullsize_picture)
self.assertEquals(None, s2.fullsize_picture)
self.assertEquals(None, s3.fullsize_picture)
self.assertEquals(data, image.retrieve_image_key(s1.key).get().picture)
self.assertEquals(data, image.retrieve_image_key(s2.key).get().picture)
self.assertEquals(data, image.retrieve_image_key(s3.key).get().picture)
def get(self):
speaker_key = ndb.Key(urlsafe=self.request.get('spk_id'))
if image.image_exists(speaker_key):
self.response.headers['Content-Type'] = 'image/png'
self.response.out.write(
image.retrieve_image_key(speaker_key).get().picture)
else:
self.response.out.write('No image')
def test_convert_speaker_full_image_to_mimas_image(self):
data = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
s = self.mk_speaker_with_image("mail@email", data)
image.convert_speaker_full_image_to_mimas_image(s)
self.assertEquals(None, s.fullsize_picture)
img_key = image.retrieve_image_key(s.key)
self.assertEquals(data, img_key.get().picture)
def test_basics(self):
new_conf = conference.Conference()
new_conf.name = "WorldConf 2016"
new_conf.put()
self.assertFalse(image.image_exists(new_conf.key))
self.assertEquals(None, image.retrieve_image_key(new_conf.key))
image.store_only_image(new_conf.key, u"")
self.assertFalse(image.image_exists(new_conf.key))
some_image = BytesIO(b"some binary data: \x00\x01\x02")
image.store_only_image(new_conf.key, some_image.getvalue())
self.assertTrue(image.image_exists(new_conf.key))
self.assertEquals(some_image.getvalue(), image.retrieve_image_key(new_conf.key).get().picture)
self.assertFalse(image.image_exists(None))
image.delete_image(new_conf.key)
self.assertFalse(image.image_exists(new_conf.key))
def test_convert_speaker_null_image(self):
s = self.mk_speaker_with_image("mail@email", None)
image.convert_speaker_full_image_to_mimas_image(s)
self.assertEquals(None, s.fullsize_picture)
self.assertEquals(None, image.retrieve_image_key(s.key))
def get_image_key(self):
return image.retrieve_image_key(self.key)
def set_image(self, img):
img_key = image.retrieve_image_key(self.key)
if img_key != None:
img_key.delete()
return image.store_only_image(self.key, img)