def testPostIdConstruction(self):
"""Verify round-trip of various post-ids."""
def _RoundTripPostId(original_episode_id, original_photo_id):
post_id = Post.ConstructPostId(original_episode_id,
original_photo_id)
new_episode_id, new_photo_id = Post.DeconstructPostId(post_id)
self.assertEqual(original_episode_id, new_episode_id)
self.assertEqual(original_photo_id, new_photo_id)
_RoundTripPostId(Episode.ConstructEpisodeId(time.time(), 0, 0),
Photo.ConstructPhotoId(time.time(), 0, 0))
_RoundTripPostId(
Episode.ConstructEpisodeId(time.time(), 1, (127, 'extra')),
Photo.ConstructPhotoId(time.time(), 1, (127, 'extra')))
_RoundTripPostId(
Episode.ConstructEpisodeId(time.time(), 1, (128, None)),
Photo.ConstructPhotoId(time.time(), 1, (128, None)))
_RoundTripPostId(
Episode.ConstructEpisodeId(time.time(), 4000000000,
(5000000000, 'v123')),
Photo.ConstructPhotoId(time.time(), 6000000000,
(7000000000, 'v123')))
def _CreateEpisodeWithPosts(self, sharer_user, parent_ep_id, ph_dicts):
"""Creates a new episode containing the given photos."""
# Create the episode.
episode_id = Episode.ConstructEpisodeId(self._op.timestamp, self._new_user.webapp_dev_id, self._unique_id)
self._unique_id += 1
episode = yield gen.Task(Episode.CreateNew,
self._client,
episode_id=episode_id,
parent_ep_id=parent_ep_id,
user_id=sharer_user.user_id,
viewpoint_id=self._viewpoint_id,
publish_timestamp=util.GetCurrentTimestamp(),
timestamp=self._op.timestamp,
location=ph_dicts[0].get('location', None),
placemark=ph_dicts[0].get('placemark', None))
# Create the photos from photo dicts.
photo_ids = [ph_dict['photo_id'] for ph_dict in ph_dicts]
for photo_id in photo_ids:
yield gen.Task(Post.CreateNew, self._client, episode_id=episode_id, photo_id=photo_id)
# Update accounting, but only apply to the new user, since system users will remove
# themselves from the viewpoint.
yield self._acc_accum.SharePhotos(self._client,
sharer_user.user_id,
self._viewpoint_id,
photo_ids,
[self._new_user.user_id])
# Update viewpoint shared by total for the sharing user.
self._acc_accum.GetViewpointSharedBy(self._viewpoint_id, sharer_user.user_id).IncrementFromPhotoDicts(ph_dicts)
raise gen.Return(episode)
def testBatchQuery(self):
"""Test DBObject.BatchQuery."""
# Create some data to query.
keys = []
for i in xrange(3):
photo_id = Photo.ConstructPhotoId(time.time(),
self._mobile_dev.device_id, 1)
episode_id = Episode.ConstructEpisodeId(time.time(),
self._mobile_dev.device_id,
1)
ph_dict = {
'photo_id': photo_id,
'user_id': self._user.user_id,
'episode_id': episode_id
}
self._RunAsync(Photo.CreateNew, self._client, **ph_dict)
keys.append(DBKey(photo_id, None))
# Add a key that will not be found.
keys.append(DBKey('unk-photo', None))
photos = self._RunAsync(Photo.BatchQuery,
self._client,
keys,
None,
must_exist=False)
self.assertEqual(len(photos), 4)
for i in xrange(3):
self.assertEqual(photos[i].GetKey(), keys[i])
self.assertIsNone(photos[3])
def testCreateAndUpdate(self):
"""Creates a episode with id pre-allocated on mobile device. Then updates the episode."""
with EnterOpContext(Operation(1, 'o1')):
timestamp = time.time()
episode_id = Episode.ConstructEpisodeId(timestamp,
self._mobile_dev.device_id,
15)
ep_dict = {
'user_id': self._user.user_id,
'episode_id': episode_id,
'viewpoint_id': self._user.private_vp_id,
'timestamp': time.time(),
'publish_timestamp': time.time(),
'description': 'yada yada this is a episode',
'title': 'Episode #1'
}
episode = self._RunAsync(Episode.CreateNew, self._client,
**ep_dict)
episode._version = None
self.assertEqual(ep_dict, episode._asdict())
update_dict = {
'episode_id': episode.episode_id,
'user_id': episode.user_id,
'description': 'updated description',
'title': 'Episode #1a'
}
self._RunAsync(episode.UpdateExisting, self._client, **update_dict)
episode._version = None
ep_dict.update(update_dict)
self.assertEqual(ep_dict, episode._asdict())
def _FormatAttribute(self, name, value):
"""Returns the attribute value; If none, returns '-'. Formats by
default the following fields: 'viewpoint_id', 'episode_id',
'photo_id', 'timestamp', 'Location', 'Placemark'.
"""
if name == 'viewpoint_id' or name == 'private_vp_id':
did, (vid, sid) = Viewpoint.DeconstructViewpointId(value)
pretty = '%s/%d/%d' % (value, did, vid)
return FmtDefault._ViewpointLink(value, pretty)
elif name == 'user_id' or name == 'sender_id':
return self._UserLink(value)
elif name == 'episode_id' or name == 'parent_ep_id':
ts, did, (eid, sid) = Episode.DeconstructEpisodeId(value)
pretty = '%s/%d/%d' % (value, did, eid)
return self._EpisodeLink(value, pretty)
elif name == 'photo_id' or name == 'parent_id':
ts, did, (pid, sid) = Photo.DeconstructPhotoId(value)
pretty = '%s/%d/%d' % (value, did, pid)
return self._PhotoLink(value, pretty)
elif name == 'timestamp' or name == 'last_updated' or name == 'expires' or name == 'last_fetch':
return self._FormatTimestamp(value)
elif name == 'location':
return self._XEscape(', '.join([
'%s: %s' % (k, v)
for k, v in UnpackLocation(value)._asdict().items()
]))
elif name == 'placemark':
return self._XEscape(', '.join([
'%s: %s' % (k, v)
for k, v in UnpackPlacemark(value)._asdict().items()
]))
else:
return self._XEscape('%s' % value)
def _SetWelcomeIds(user, upload_request):
"""Assigns activity, episode, and photo ids for all welcome conversation requests. Assets
are assigned unique ids starting at 1.
"""
# Construct the activity id.
unique_id = 1
act_dict = upload_request['activity']
act_dict['activity_id'] = Activity.ConstructActivityId(
act_dict['timestamp'], user.webapp_dev_id, unique_id)
unique_id += 1
# Construct the episode id.
ep_dict = upload_request['episode']
ep_dict['episode_id'] = Episode.ConstructEpisodeId(ep_dict['timestamp'],
user.webapp_dev_id,
unique_id)
unique_id += 1
# Construct the photo ids.
for ph_dict in upload_request['photos']:
# Create metadata for each photo.
ph_dict['photo_id'] = Photo.ConstructPhotoId(ph_dict['timestamp'],
user.webapp_dev_id,
unique_id)
unique_id += 1
def _UploadEpisodeWithPhoto(self):
"""Create episode with photo and upload.
Returns: photo_id of created photo.
"""
timestamp = time.time()
episode_id = Episode.ConstructEpisodeId(timestamp, self._device_ids[0], 100)
ep_dict = {'episode_id': episode_id,
'timestamp': timestamp,
'title': 'Episode Title'}
photo_id = Photo.ConstructPhotoId(timestamp, self._device_ids[0], 100)
ph_dict = {'aspect_ratio': 1.3333,
'timestamp': timestamp,
'tn_md5': util.ComputeMD5Hex('thumbnail image data'),
'med_md5': util.ComputeMD5Hex('medium image data'),
'full_md5': util.ComputeMD5Hex('full image data'),
'orig_md5': util.ComputeMD5Hex('original image data'),
'tn_size': 5*1024,
'med_size': 10*1024,
'full_size': 150*1024,
'orig_size': 1200*1024,
'caption': 'a photo',
'photo_id': photo_id}
self._tester.UploadEpisode(self._cookie, ep_dict, [ph_dict])
return photo_id
def _AllocateTargetEpisodeIds(self, client, user_id, device_id,
target_viewpoint_id, source_ep_ids):
"""For each episode listed in "source_ep_ids", determines if a child episode already
exists in the given viewpoint. If not, allocates a new episode id using the user's asset
id allocator. The same timestamp used to create the source episode id is used to create
the target episode id.
Returns the list of target episodes ids, including both existing ids and allocated ids.
"""
# First check whether each episode has already been shared/saved into the target viewpoint.
tasks = []
for source_ep_id in source_ep_ids:
query_expr = ('episode.parent_ep_id={id}', {'id': source_ep_id})
tasks.append(gen.Task(Episode.IndexQuery, client, query_expr,
None))
target_ep_ids = []
allocate_ids_count = 0
target_episodes_list = yield tasks
for target_episodes in target_episodes_list:
found_match = False
for episode in target_episodes:
if episode.viewpoint_id == target_viewpoint_id:
target_ep_ids.append(episode.episode_id)
found_match = True
break
# If no matching child episode, then need to allocate an episode id.
if not found_match:
target_ep_ids.append(None)
allocate_ids_count += 1
if allocate_ids_count > 0:
# Allocate ids for any episodes which do not yet exist, and merge them into target_ep_ids.
id = yield gen.Task(User.AllocateAssetIds, client, user_id,
allocate_ids_count)
for i, source_ep_id in enumerate(source_ep_ids):
if target_ep_ids[i] is None:
timestamp, _, _ = source_ep_id = Episode.DeconstructEpisodeId(
source_ep_id)
target_ep_ids[i] = Episode.ConstructEpisodeId(
timestamp, device_id, id)
id += 1
raise gen.Return(target_ep_ids)
def _QueryAndVerify(episode_ids, barrier_cb, loc_search, matches):
def _Verify(keys):
ids = [key.hash_key for key in keys]
self.assertEqual(len(ids), len(matches))
[self.assertTrue(episode_ids[m] in ids) for m in matches]
barrier_cb()
Episode.IndexQueryKeys(self._client, 'episode.location="%f,%f,%f"' % \
(loc_search[0], loc_search[1], loc_search[2]), callback=_Verify)
def _QueryAndVerify(episode_ids, barrier_cb, search, matches):
def _Verify(keys):
ids = [key.hash_key for key in keys]
self.assertEqual(len(ids), len(matches))
[self.assertTrue(episode_ids[m] in ids) for m in matches]
barrier_cb()
Episode.IndexQueryKeys(self._client, ('episode.placemark={s}', {
's': search
}),
callback=_Verify)
def testSaveDuplicateIds(self):
"""Save duplicate episode and photo ids."""
# ------------------------------
# Duplicate photo ids.
# ------------------------------
self._tester.SavePhotos(
self._cookie,
[(self._existing_ep_id, self._photo_ids + self._photo_ids)])
self.assertEqual(
self._CountEpisodes(self._cookie, self._user.private_vp_id), 3)
# ------------------------------
# Duplicate episode ids.
# ------------------------------
self._tester.SavePhotos(self._cookie,
[(self._existing_ep_id, self._photo_ids),
(self._existing_ep_id, self._photo_ids)])
self.assertEqual(
self._CountEpisodes(self._cookie, self._user.private_vp_id), 5)
# ------------------------------
# Save same episode to same target episode multiple times.
# ------------------------------
new_episode_id = Episode.ConstructEpisodeId(time.time(),
self._device_ids[0],
self._test_id)
self._test_id += 1
self._tester.SavePhotos(self._cookie, [{
'existing_episode_id': self._existing_ep_id,
'new_episode_id': new_episode_id,
'photo_ids': self._photo_ids[:1]
}, {
'existing_episode_id': self._existing_ep_id,
'new_episode_id': new_episode_id,
'photo_ids': self._photo_ids
}])
self.assertEqual(
self._CountEpisodes(self._cookie, self._user.private_vp_id), 6)
# ------------------------------
# Duplicate viewpoint ids.
# ------------------------------
vp_id, ep_ids = self._tester.ShareNew(
self._cookie, [(self._episode_id2, self._photo_ids2)],
[self._user3.user_id], **self._CreateViewpointDict(self._cookie))
self._tester.SavePhotos(self._cookie,
ep_save_list=[
(self._existing_ep_id, self._photo_ids),
(self._existing_ep_id, self._photo_ids)
],
viewpoint_ids=[vp_id, vp_id])
self.assertEqual(
self._CountEpisodes(self._cookie, self._user.private_vp_id), 9)
def _CreateEpisodeDict(self, user_cookie, **update_ep_dict):
"""Create dict() for a test episode, overriding default values with
whatever is passed in "update_ep_dict"."""
user_id, device_id = self._tester.GetIdsFromCookie(user_cookie)
timestamp = time.time() - self._test_id
ep_dict = {'episode_id': Episode.ConstructEpisodeId(timestamp, device_id, self._test_id),
'timestamp': timestamp,
'title': 'Title %s' % self._test_id,
'description': 'Description %s. 朋友你好.' % self._test_id}
self._test_id += 1
ep_dict.update(**update_ep_dict)
return ep_dict
def testSaveDuplicatePhotos(self):
"""Save same photos from same source episode to same target episode in default viewpoint."""
new_episode_id = Episode.ConstructEpisodeId(time.time(),
self._device_ids[0],
self._test_id)
self._test_id += 1
share_list = [{
'existing_episode_id': self._existing_ep_id,
'new_episode_id': new_episode_id,
'photo_ids': self._photo_ids
}]
self._tester.SavePhotos(self._cookie, share_list)
self._tester.SavePhotos(self._cookie, share_list)
def testSuppressCopyTimestamp(self):
"""Test migrator that removes timestamp from share and save operations."""
self._CreateSimpleTestAssets()
# First do a share_new.
new_episode_id = Episode.ConstructEpisodeId(time.time(), self._device_ids[0], self._test_id)
self._test_id += 1
request_dict = {'activity': self._tester.CreateActivityDict(self._cookie),
'viewpoint': self._CreateViewpointDict(self._cookie),
'episodes': [{'existing_episode_id': self._episode_id,
'new_episode_id': new_episode_id,
'timestamp': time.time(),
'photo_ids': self._photo_ids}],
'contacts': self._tester.CreateContactDicts([self._user2.user_id])}
self._SendRequest('share_new', self._cookie, request_dict, version=Message.SUPPORT_REMOVED_FOLLOWERS)
# Now, do a share_existing.
new_episode_id2 = Episode.ConstructEpisodeId(time.time(), self._device_ids[0], self._test_id)
self._test_id += 1
request_dict = {'activity': self._tester.CreateActivityDict(self._cookie),
'viewpoint_id': request_dict['viewpoint']['viewpoint_id'],
'episodes': [{'existing_episode_id': self._episode_id2,
'new_episode_id': new_episode_id2,
'timestamp': time.time(),
'photo_ids': self._photo_ids2}]}
self._SendRequest('share_existing', self._cookie, request_dict, version=Message.SUPPORT_REMOVED_FOLLOWERS)
# Now, do a save_photos.
save_episode_id = Episode.ConstructEpisodeId(time.time(), self._device_ids[1], self._test_id)
self._test_id += 1
request_dict = {'activity': self._tester.CreateActivityDict(self._cookie2),
'episodes': [{'existing_episode_id': new_episode_id,
'new_episode_id': save_episode_id,
'timestamp': time.time(),
'photo_ids': self._photo_ids}]}
self._SendRequest('save_photos', self._cookie2, request_dict, version=Message.SUPPORT_REMOVED_FOLLOWERS)
def testShareSamePhotoSameEpisode(self):
"""ERROR: Try to share the same photo multiple times to same episode."""
timestamp = time.time()
new_episode_id = Episode.ConstructEpisodeId(timestamp,
self._device_ids[0],
self._test_id)
self._test_id += 1
share_dict = {
'existing_episode_id': self._episode_id,
'new_episode_id': new_episode_id,
'timestamp': timestamp,
'photo_ids': self._photo_ids
}
self.assertRaisesHttpError(400, self._tester.ShareNew, self._cookie,
[share_dict, share_dict], [])
def testWrongViewpoint(self):
"""ERROR: Try to share photos to existing episode that is not in the target viewpoint."""
timestamp = time.time()
new_episode_id = Episode.ConstructEpisodeId(timestamp,
self._device_ids[0],
self._test_id)
self._test_id += 1
share_dict = {
'existing_episode_id': self._episode_id,
'new_episode_id': self._episode_id2,
'photo_ids': self._photo_ids
}
self.assertRaisesHttpError(400, self._tester.ShareExisting,
self._cookie, self._existing_vp_id,
[share_dict])
def testWrongDeviceIds(self):
"""ERROR: Try to create an episode and photo using device ids that
are different than the ones in the user cookies.
"""
bad_episode_id = Episode.ConstructEpisodeId(100, 1000, 1)
self.assertRaisesHttpError(403,
self._tester.UploadEpisode,
self._cookie,
ep_dict={
'episode_id': bad_episode_id,
'timestamp': 100
},
ph_dict_list=[])
episode_id = Episode.ConstructEpisodeId(100, self._device_ids[0], 100)
bad_photo_id = Photo.ConstructPhotoId(100, 1000, 1)
self.assertRaisesHttpError(403,
self._tester.UploadEpisode,
self._cookie,
ep_dict={},
ph_dict_list=[
self._CreatePhotoDict(
self._cookie, photo_id=bad_photo_id)
])
def _UploadPhotoOperation(self,
user_id,
device_id,
seed,
callback,
photo_id=None):
"""Creates an upload photos operation using seed to create unique ids."""
request = {
'user_id':
user_id,
'activity': {
'activity_id':
Activity.ConstructActivityId(time.time(), device_id, seed),
'timestamp':
time.time()
},
'episode': {
'user_id':
user_id,
'episode_id':
Episode.ConstructEpisodeId(time.time(), device_id, seed),
'timestamp':
time.time()
},
'photos': [{
'photo_id':
Photo.ConstructPhotoId(time.time(), device_id, seed)
if photo_id is None else photo_id,
'aspect_ratio':
1.3333,
'timestamp':
time.time(),
'tn_size':
5 * 1024,
'med_size':
50 * 1024,
'full_size':
150 * 1024,
'orig_size':
1200 * 1024
}]
}
Operation.CreateAndExecute(self._client, user_id, device_id,
'UploadEpisodeOperation.Execute', request,
callback)
def testMultipleEpisodeUnshare(self):
"""Unshare photos from multiple episodes inside the same viewpoint."""
vp_id, ep_ids = self._tester.ShareNew(
self._cookie, [(self._new_ep_id, self._photo_ids)],
[self._user2.user_id])
timestamp = time.time()
episode_id = Episode.ConstructEpisodeId(timestamp, self._device_ids[0],
100)
ep_id, ph_ids = self._tester.UploadEpisode(
self._cookie, {}, [self._CreatePhotoDict(self._cookie)])
new_ep_ids = self._tester.ShareExisting(self._cookie, vp_id,
[(ep_id, ph_ids)])
self._tester.Unshare(self._cookie, vp_id,
[(ep_ids[0], self._photo_ids),
(new_ep_ids[0], ph_ids)])
def _GetShareInfo(cls, activity):
"""Returns a tuple with information about a "share_existing" or "share_new" operation:
episode_dates: List of timestamps for each episode in the share.
num_shares: Total count of photos shared.
"""
episode_dates = []
num_shares = 0
args_dict = json.loads(activity.json)
for ep_dict in args_dict['episodes']:
# Extract timestamps from episode id so that we don't need to query for episode object.
ts, dev_id, uniquifier = Episode.DeconstructEpisodeId(
ep_dict['episode_id'])
episode_dates.append(date.fromtimestamp(ts))
num_shares += len(ep_dict['photo_ids'])
return (episode_dates, num_shares)
def testSaveToSameEpisode(self):
"""Save multiple photos to same target episode in default viewpoint."""
timestamp = time.time()
new_episode_id = Episode.ConstructEpisodeId(timestamp,
self._device_ids[0],
self._test_id)
self._test_id += 1
share_dict1 = {
'existing_episode_id': self._existing_ep_id,
'new_episode_id': new_episode_id,
'photo_ids': self._photo_ids[:1]
}
share_dict2 = {
'existing_episode_id': self._existing_ep_id,
'new_episode_id': new_episode_id,
'photo_ids': self._photo_ids[1:]
}
self._tester.SavePhotos(self._cookie, [share_dict1, share_dict2])
def testFileSizeExtraction(self):
"""Ensure that the EXTRACT_FILE_SIZES migration works correctly."""
episode_id = Episode.ConstructEpisodeId(1, self._device_ids[0], 1)
photo_id = Photo.ConstructPhotoId(1, self._device_ids[0], 1)
# create request with size fields in client_data.
request = {'activity': self._tester.CreateActivityDict(self._cookie),
'episode': {'episode_id': episode_id,
'timestamp': 1},
'photos': [self._CreatePhotoDict(self._cookie, photo_id=photo_id,
client_data={'tn_size':'5', 'med_size':'40',
'full_size':'150', 'orig_size':'1200'})]}
# remove size fields from the photo metadata, leaving only those in client_data.
photo = request['photos'][0]
del photo['tn_size']
del photo['med_size']
del photo['full_size']
del photo['orig_size']
self._SendRequest('upload_episode', self._cookie, request, version=Message.INLINE_INVALIDATIONS)
def testSaveToSameEpisode(self):
"""ERROR: Try to save from two source episodes to the same target episode."""
share_ep_ids = self._tester.ShareExisting(
self._cookie, self._existing_vp_id,
[(self._episode_id2, self._photo_ids2)])
new_episode_id = Episode.ConstructEpisodeId(time.time(),
self._device_ids[0],
self._test_id)
self._test_id += 1
self.assertRaisesHttpError(
400, self._tester.SavePhotos, self._cookie,
[{
'existing_episode_id': self._existing_ep_id,
'new_episode_id': new_episode_id,
'photo_ids': self._photo_ids
}, {
'existing_episode_id': share_ep_ids[0],
'new_episode_id': new_episode_id,
'photo_ids': self._photo_ids2
}])
def testQuery(self):
"""Verify photo creation and query by photo id."""
def _OnQuery(p, p2):
self.assertEqual(p2.caption, p.caption)
self.assertEqual(p2.photo_id, p.photo_id)
self.stop()
def _OnCreatePhoto(p):
Photo.Query(self._client, p.photo_id, None, partial(_OnQuery, p))
photo_id = Photo.ConstructPhotoId(time.time(),
self._mobile_dev.device_id, 1)
episode_id = Episode.ConstructEpisodeId(time.time(),
self._mobile_dev.device_id, 2)
p_dict = {
'photo_id': photo_id,
'episode_id': episode_id,
'user_id': self._user.user_id,
'caption': 'a photo'
}
Photo.CreateNew(self._client, callback=_OnCreatePhoto, **p_dict)
def testInlineInvalidationsMigration(self):
"""Ensure that the INLINE_INVALIDATIONS migration works correctly."""
episode_id = Episode.ConstructEpisodeId(1, self._device_ids[0], 1)
photo_id = Photo.ConstructPhotoId(1, self._device_ids[0], 1)
request = {'activity': self._tester.CreateActivityDict(self._cookie),
'episode': {'episode_id': episode_id,
'timestamp': 1},
'photos': [self._CreatePhotoDict(self._cookie, photo_id=photo_id)]}
self._SendRequest('upload_episode', self._cookie, request,
version=Message.EXTRACT_MD5_HASHES)
request = {'episodes': [{'episode_id': episode_id,
'photo_ids': [photo_id]}]}
self._SendRequest('remove_photos', self._cookie, request,
version=Message.EXTRACT_MD5_HASHES)
request = {'activity': self._tester.CreateActivityDict(self._cookie),
'viewpoint_id': self._user.private_vp_id,
'viewed_seq': 2}
self._SendRequest('update_viewpoint', self._cookie, request,
version=Message.EXTRACT_MD5_HASHES)
response = self._SendRequest('query_notifications', self._cookie, {},
version=Message.EXTRACT_MD5_HASHES)
# upload_episode notification.
notify_dict = response['notifications'][1]
self.assertFalse('inline' in notify_dict)
self.assertTrue('activity' in notify_dict)
self.assertEqual(notify_dict['activity']['update_seq'], 1)
# remove_photos notification.
notify_dict = response['notifications'][2]
self.assertFalse('inline' in notify_dict)
self.assertFalse('activity' in notify_dict)
# update_viewpoint notification.
notify_dict = response['notifications'][3]
self.assertFalse('inline' in notify_dict)
self.assertFalse('activity' in notify_dict)
def testUpdateAttribute(self):
"""Verify update of a photo attribute."""
def _OnUpdate(p):
p.aspect_ratio = None
p.Update(self._client, self.stop)
def _OnQuery(p):
p.content_type = 'image/png'
p.Update(self._client, partial(_OnUpdate, p))
def _OnCreatePhoto(p):
Photo.Query(self._client, p.photo_id, None, _OnQuery)
photo_id = Photo.ConstructPhotoId(time.time(),
self._mobile_dev.device_id, 1)
episode_id = Episode.ConstructEpisodeId(time.time(),
self._mobile_dev.device_id, 2)
p_dict = {
'photo_id': photo_id,
'episode_id': episode_id,
'user_id': self._user.user_id,
'caption': 'A Photo'
}
Photo.CreateNew(self._client, callback=_OnCreatePhoto, **p_dict)
def testUpdateEpisode(self):
"""Creates a new episode and photo and updates both."""
timestamp = time.time()
episode_id = Episode.ConstructEpisodeId(timestamp, self._device_ids[0], 100)
ep_dict = {'episode_id': episode_id,
'timestamp': timestamp,
'title': 'Episode Title'}
photo_id = Photo.ConstructPhotoId(timestamp, self._device_ids[0], 100)
ph_dict = {'aspect_ratio': 1.3333,
'timestamp': time.time(),
'tn_md5': util.ComputeMD5Hex('thumbnail image data'),
'med_md5': util.ComputeMD5Hex('medium image data'),
'full_md5': util.ComputeMD5Hex('full image data'),
'orig_md5': util.ComputeMD5Hex('original image data'),
'tn_size': 5 * 1024,
'med_size': 10 * 1024,
'full_size': 150 * 1024,
'orig_size': 1200 * 1024,
'photo_id': photo_id}
self._tester.UploadEpisode(self._cookie, ep_dict, [ph_dict])
self._tester.UpdateEpisode(self._cookie, episode_id, description='A newly added description')
def testPostIdOrdering(self):
"""Verify that post_id sorts like (episode_id, photo_id) does."""
def _Compare(episode_id1, photo_id1, episode_id2, photo_id2):
result = cmp(episode_id1, episode_id2)
if result == 0:
result = cmp(photo_id1, photo_id2)
post_id1 = Post.ConstructPostId(episode_id1, photo_id1)
post_id2 = Post.ConstructPostId(episode_id2, photo_id2)
self.assertEqual(cmp(post_id1, post_id2), result)
timestamp = time.time()
episode_id1 = Episode.ConstructEpisodeId(timestamp, 1, (127, None))
episode_id2 = Episode.ConstructEpisodeId(timestamp, 1, (128, None))
photo_id1 = Photo.ConstructPhotoId(timestamp, 1, 128)
photo_id2 = Photo.ConstructPhotoId(timestamp, 1, 127)
_Compare(episode_id1, photo_id1, episode_id2, photo_id2)
episode_id1 = Episode.ConstructEpisodeId(timestamp, 127, 1)
episode_id2 = Episode.ConstructEpisodeId(timestamp, 128, 1)
photo_id1 = Photo.ConstructPhotoId(timestamp, 128, (1, None))
photo_id2 = Photo.ConstructPhotoId(timestamp, 127, (1, None))
_Compare(episode_id1, photo_id1, episode_id2, photo_id2)
episode_id1 = Episode.ConstructEpisodeId(timestamp, 0, 0)
episode_id2 = Episode.ConstructEpisodeId(timestamp, 0, 0)
photo_id1 = Photo.ConstructPhotoId(timestamp, 0, 0)
photo_id2 = Photo.ConstructPhotoId(timestamp, 0, 0)
_Compare(episode_id1, photo_id1, episode_id2, photo_id2)
episode_id1 = Episode.ConstructEpisodeId(timestamp, 1, 0)
episode_id2 = Episode.ConstructEpisodeId(timestamp, 1, 1)
photo_id1 = Photo.ConstructPhotoId(timestamp, 1, 1)
photo_id2 = Photo.ConstructPhotoId(timestamp, 1, 0)
_Compare(episode_id1, photo_id1, episode_id2, photo_id2)
episode_id1 = Episode.ConstructEpisodeId(0, 0, 0)
episode_id2 = Episode.ConstructEpisodeId(1, 0, 0)
photo_id1 = Photo.ConstructPhotoId(1, 0, (0, None))
photo_id2 = Photo.ConstructPhotoId(0, 0, (0, None))
_Compare(episode_id1, photo_id1, episode_id2, photo_id2)
episode_id1 = Episode.ConstructEpisodeId(timestamp, 0, (0, '1'))
episode_id2 = Episode.ConstructEpisodeId(timestamp, 0, (0, '2'))
photo_id1 = Photo.ConstructPhotoId(timestamp, 0, (0, None))
photo_id2 = Photo.ConstructPhotoId(timestamp, 0, (0, None))
_Compare(episode_id1, photo_id1, episode_id2, photo_id2)
episode_id1 = Episode.ConstructEpisodeId(timestamp, 0, 0)
episode_id2 = Episode.ConstructEpisodeId(timestamp, 0, 0)
photo_id1 = Photo.ConstructPhotoId(timestamp, 0, (0, u'ab'))
photo_id2 = Photo.ConstructPhotoId(timestamp, 0, (0, u'cd'))
_Compare(episode_id1, photo_id1, episode_id2, photo_id2)
def disabled_t_estPlacemarkQueries(self):
"""Tests placemark queries."""
def _QueryAndVerify(episode_ids, barrier_cb, search, matches):
def _Verify(keys):
ids = [key.hash_key for key in keys]
self.assertEqual(len(ids), len(matches))
[self.assertTrue(episode_ids[m] in ids) for m in matches]
barrier_cb()
Episode.IndexQueryKeys(self._client, ('episode.placemark={s}', {
's': search
}),
callback=_Verify)
def _OnCreate(locations, episodes):
with util.Barrier(self.stop) as b:
episode_ids = dict([(v.title, v.episode_id) for v in episodes])
_QueryAndVerify(episode_ids, b.Callback(), 'Broadway',
['kimball ph'])
_QueryAndVerify(episode_ids, b.Callback(), '682 Broadway',
['kimball ph'])
_QueryAndVerify(episode_ids, b.Callback(), 'Broadway 682', [])
_QueryAndVerify(episode_ids, b.Callback(),
'new york, ny, united states', [
'kimball ph', 'bond st sushi',
'viewfinder', 'soho house', 'google'
])
_QueryAndVerify(episode_ids, b.Callback(), 'new york, ny', [
'kimball ph', 'bond st sushi', 'viewfinder', 'soho house',
'google'
])
_QueryAndVerify(
episode_ids, b.Callback(), 'NY, United States', [
'kimball ph', 'bond st sushi', 'viewfinder',
'soho house', 'google', 'kimball east', 'surf lodge'
])
_QueryAndVerify(episode_ids, b.Callback(), 'United States', [
'kimball ph', 'bond st sushi', 'viewfinder', 'soho house',
'google', 'kimball east', 'surf lodge'
])
_QueryAndVerify(episode_ids, b.Callback(), 'Bahamas',
['atlantis'])
_QueryAndVerify(episode_ids, b.Callback(), 'Dominican',
['casa kimball'])
_QueryAndVerify(episode_ids, b.Callback(),
'Dominican Republic', ['casa kimball'])
_QueryAndVerify(episode_ids, b.Callback(), 'Cabrera',
['casa kimball'])
_QueryAndVerify(episode_ids, b.Callback(), 'DR',
['casa kimball'])
locations = {
'kimball ph':
Placemark('US', 'United States', 'NY', 'New York', 'NoHo',
'Broadway', '682'),
'bond st sushi':
Placemark('US', 'United States', 'NY', 'New York', 'NoHo',
'Bond St', '6'),
'viewfinder':
Placemark('US', 'United States', 'NY', 'New York', 'SoHo',
'Grand St', '154'),
'soho house':
Placemark('US', 'United States', 'NY', 'New York',
'Meatpacking District', '9th Avenue', '29-35'),
'google':
Placemark('US', 'United States', 'NY', 'New York', 'Chelsea',
'8th Avenue', '111'),
'kimball east':
Placemark('US', 'United States', 'NY', 'East Hampton',
'Northwest Harbor', 'Milina', '35'),
'surf lodge':
Placemark('US', 'United States', 'NY', 'Montauk', '',
'Edgemere St', '183'),
'casa kimball':
Placemark('DR', 'Dominican Republic', 'Maria Trinidad Sanchez',
'Cabrera', 'Orchid Bay Estates', '', '5-6'),
'atlantis':
Placemark('BS', 'Bahamas', '', 'Paradise Island', '', '', '')
}
with util.ArrayBarrier(partial(_OnCreate, locations)) as b:
device_episode_id = 0
for place, placemark in locations.items():
device_episode_id += 1
timestamp = time.time()
episode_id = Episode.ConstructEpisodeId(
timestamp, 1, device_episode_id)
episode = Episode.CreateFromKeywords(
timestamp=timestamp,
episode_id=episode_id,
user_id=self._user.user_id,
viewpoint_id=self._user.private_vp_id,
publish_timestamp=timestamp,
title=place,
placemark=placemark)
episode.Update(self._client, b.Callback())
def testAssetIdAltDevice(self):
"""Test construction of assets using a different device than the calling device."""
# ------------------------------
# Try to upload using a device not owned by the user at all.
# ------------------------------
ep_dict = self._CreateEpisodeDict(self._cookie)
ep_dict['episode_id'] = Episode.ConstructEpisodeId(
time.time(), self._device_ids[2], self._test_id)
self._test_id += 1
ph_dict = self._CreatePhotoDict(self._cookie)
ph_dict['photo_id'] = Photo.ConstructPhotoId(time.time(),
self._device_ids[2],
self._test_id)
self._test_id += 1
self.assertRaisesHttpError(403, self._tester.UploadEpisode,
self._cookie, ep_dict, [ph_dict])
# ------------------------------
# Upload using alternate devices owned by the user.
# ------------------------------
ep_dict = self._CreateEpisodeDict(self._cookie)
ep_dict['episode_id'] = Episode.ConstructEpisodeId(
time.time(), self._extra_device_id1, self._test_id)
self._test_id += 1
ph_dict = self._CreatePhotoDict(self._cookie)
ph_dict['photo_id'] = Photo.ConstructPhotoId(time.time(),
self._extra_device_id2,
self._test_id)
self._test_id += 1
act_dict = self._tester.CreateActivityDict(self._cookie)
act_dict['activity_id'] = Activity.ConstructActivityId(
time.time(), self._extra_device_id1, self._test_id)
self._test_id += 1
self._tester.UploadEpisode(self._cookie, ep_dict, [ph_dict], act_dict)
# ------------------------------
# Share to a new viewpoint using alternate devices owned by the user.
# ------------------------------
viewpoint_id = Viewpoint.ConstructViewpointId(self._extra_device_id2,
self._test_id)
self._test_id += 1
self._tester.ShareNew(self._cookie,
[(ep_dict['episode_id'], [ph_dict['photo_id']])],
[self._user2.user_id],
viewpoint_id=viewpoint_id)
# ------------------------------
# Post to the new viewpoint using alternate devices owned by the user.
# ------------------------------
comment_id = Comment.ConstructCommentId(time.time(),
self._extra_device_id1,
self._test_id)
self._test_id += 1
self._tester.PostComment(self._cookie, viewpoint_id, 'hi')