def test_select_and_save(self):
b2 = Bangumi(name='test_select_and_save', update_time='Sun')
d = b2.select()
self.assertEqual(d, [])
b2.save()
ret = b2.select(one=True)
self.assertIsInstance(ret, dict)
def test_update(self):
b1 = Bangumi(name='test_update', update_time='Sun')
b1.save()
b1.update_time = 'Mon'
b1.update()
ret = b1.select(one=True)
self.assertEqual(ret['update_time'], 'Mon')
b1.update(data={'name': 'test666', 'update_time': 'Sat'})
ret = b1.select(one=True)
self.assertIsInstance(ret, dict)
def test_add_and_remove_followed(self):
f = Followed(bangumi_name='test_add_and_remove_followed',
status=STATUS_FOLLOWED,
episode=6)
f.save()
b = Bangumi(name='test_add_and_remove_followed')
b.save()
bangumi_data = b.select(
one=True,
join='LEFT JOIN %s ON %s.bangumi_name=%s.name' %
(Followed.table, Followed.table, Bangumi.table))
self.assertEqual(bangumi_data['status'], STATUS_FOLLOWED)
f.delete()
bangumi_data = b.select(
one=True,
join='LEFT JOIN %s ON %s.bangumi_name=%s.name' %
(Followed.table, Followed.table, Bangumi.table))
self.assertEqual(bangumi_data['status'], STATUS_NORMAL)
def test_delete(self):
b1 = Bangumi(name='test_delete', update_time='Sun')
b1.save()
self.assertEqual(b1.select(one=True)['name'], 'test_delete')
b1.delete()
self.assertEqual(b1.select(one=True), None)
def save_data(data):
b = Bangumi(**data)
b.save()