def test_user(self):
self.handler = WeiboUserHandler()
self.handler.add_user(
uid='110',
followees=10,
fans=8,
posts=1000,
)
# add duplicate user
self.handler.add_user(
uid='110',
followees=10,
fans=8,
posts=1000,
)
u = self.handler.get_user_by_uid(uid='110')
u1 = self.handler.get_user_by_uid(uid='110')
u2 = self.handler.get_user_by_uid(uid='110')
self.assertFalse(u is u1 or u1 is u2)
self.assertTrue(u.uid == '110')
self.assertFalse(u.uid == 110)
self.assertTrue(type(u.uid) is str)
self.assertFalse(type(u.uid) is int)
self.assertFalse(type(u.uid) is long)
self.assertTrue(type(u.followees) is long)
self.handler.update_user(uid='110')
self.handler.update_user(uid='110', followees=100)
self.assertFalse(u.followees == 100)
u4 = self.handler.get_user_by_uid(uid='110')
self.assertTrue(u4.followees == 100)
self.handler.add_user(uid='111')
self.handler.add_user(uid='112')
self.handler.add_user(uid='113')
num, uids = self.handler.get_invalid_user(2)
self.assertTrue(num == 2)
self.assertTrue(isinstance(uids[0], str))
self.handler.delete_user(uid='111')
self.handler.delete_user(uid='112')
self.handler.delete_user(uid='113')
exist = self.handler.user_exist(uid='110')
self.assertTrue(exist)
self.handler.delete_user(uid='110')
exist = self.handler.user_exist(uid='110')
self.assertFalse(exist)
def test_blog(self):
# add blog with no user inserted before
self.bhandler = MicroblogHandler()
self.bhandler.add_blog(
uid='110',
mid='001',
created_time=(str(datetime.today())),
content='content',
favorites=1,
comments=1,
forwards=1,
forwarded_content=None)
self.assertFalse(self.bhandler.blog_exist('001'))
# add blog with user inserted
self.uhandler = WeiboUserHandler()
self.uhandler.add_user(
uid='110',
followees=10,
fans=8,
posts=1000,
)
self.bhandler.add_blog(
uid='110',
mid='001',
created_time=(str(datetime.today())),
content='content',
favorites=1,
comments=1,
forwards=1,
forwarded_content=None)
self.assertTrue(self.bhandler.blog_exist('001'))
# query blog
b = self.bhandler.get_blog_by_mid('001')
print 'b.content', b.content
self.assertTrue(b.content == 'content')
self.assertTrue(b.forwarded_content is None)
self.assertTrue(isinstance(b.created_time, str))
# delete all
self.bhandler.delete_blog(mid='001')
self.uhandler.delete_user(uid='110')
self.assertFalse(self.bhandler.blog_exist('001'))
self.assertFalse(self.uhandler.user_exist(uid='110'))
class TestDB(unittest.TestCase):
def setUp(self):
handler = DatabaseHandler()
handler.open()
def tearDown(self):
pass
def test_user(self):
self.handler = WeiboUserHandler()
self.handler.add_user(
uid='110',
followees=10,
fans=8,
posts=1000,
)
# add duplicate user
self.handler.add_user(
uid='110',
followees=10,
fans=8,
posts=1000,
)
u = self.handler.get_user_by_uid(uid='110')
u1 = self.handler.get_user_by_uid(uid='110')
u2 = self.handler.get_user_by_uid(uid='110')
self.assertFalse(u is u1 or u1 is u2)
self.assertTrue(u.uid == '110')
self.assertFalse(u.uid == 110)
self.assertTrue(type(u.uid) is str)
self.assertFalse(type(u.uid) is int)
self.assertFalse(type(u.uid) is long)
self.assertTrue(type(u.followees) is long)
self.handler.update_user(uid='110')
self.handler.update_user(uid='110', followees=100)
self.assertFalse(u.followees == 100)
u4 = self.handler.get_user_by_uid(uid='110')
self.assertTrue(u4.followees == 100)
self.handler.add_user(uid='111')
self.handler.add_user(uid='112')
self.handler.add_user(uid='113')
num, uids = self.handler.get_invalid_user(2)
self.assertTrue(num == 2)
self.assertTrue(isinstance(uids[0], str))
self.handler.delete_user(uid='111')
self.handler.delete_user(uid='112')
self.handler.delete_user(uid='113')
exist = self.handler.user_exist(uid='110')
self.assertTrue(exist)
self.handler.delete_user(uid='110')
exist = self.handler.user_exist(uid='110')
self.assertFalse(exist)
def test_blog(self):
# add blog with no user inserted before
self.bhandler = MicroblogHandler()
self.bhandler.add_blog(
uid='110',
mid='001',
created_time=(str(datetime.today())),
content='content',
favorites=1,
comments=1,
forwards=1,
forwarded_content=None)
self.assertFalse(self.bhandler.blog_exist('001'))
# add blog with user inserted
self.uhandler = WeiboUserHandler()
self.uhandler.add_user(
uid='110',
followees=10,
fans=8,
posts=1000,
)
self.bhandler.add_blog(
uid='110',
mid='001',
created_time=(str(datetime.today())),
content='content',
favorites=1,
comments=1,
forwards=1,
forwarded_content=None)
self.assertTrue(self.bhandler.blog_exist('001'))
# query blog
b = self.bhandler.get_blog_by_mid('001')
print 'b.content', b.content
self.assertTrue(b.content == 'content')
self.assertTrue(b.forwarded_content is None)
self.assertTrue(isinstance(b.created_time, str))
# delete all
self.bhandler.delete_blog(mid='001')
self.uhandler.delete_user(uid='110')
self.assertFalse(self.bhandler.blog_exist('001'))
self.assertFalse(self.uhandler.user_exist(uid='110'))