def test_save_and_get(self):
timestamp = self.ts_now
following = HBaseFollowing(from_user_id=123, to_user_id=34, created_at=timestamp)
following.save()
instance = HBaseFollowing.get(from_user_id=123, created_at=timestamp)
self.assertEqual(instance.from_user_id, 123)
self.assertEqual(instance.to_user_id, 34)
self.assertEqual(instance.created_at, timestamp)
following.to_user_id = 456
following.save()
instance = HBaseFollowing.get(from_user_id=123, created_at=timestamp)
self.assertEqual(instance.to_user_id, 456)
# object does not exist, return None
instance = HBaseFollowing.get(from_user_id=123, created_at=self.ts_now)
self.assertEqual(instance, None)
def test_save_and_get(self):
now = self.ts_now()
following = HBaseFollowing(
from_user_id=123,
created_at=now,
to_user_id=456,
)
following.save()
following = HBaseFollowing.get(from_user_id=123, created_at=now)
self.assertEqual(following.from_user_id, 123)
self.assertEqual(following.created_at, now)
self.assertEqual(following.to_user_id, 456)
following.to_user_id = 789
following.save()
self.assertEqual(following.to_user_id, 789)
following = HBaseFollowing.get(from_user_id=123, created_at=now)
self.assertEqual(following.to_user_id, 789)
# get non-exist object get none
following = HBaseFollowing.get(from_user_id=123,
created_at=self.ts_now())
self.assertIsNone(following)
# wrong row_keys to creat
try:
following = HBaseFollowing(from_user_id=123, to_user_id=456)
following.save()
exception_raised = False
except BadRowKeyException as e:
exception_raised = True
self.assertEqual(str(e), 'created_at not defined')
self.assertEqual(exception_raised, True)
