def test_save_bookmark_conflict(self): with self.assertRaises(IntegrityError): query = Bookmark(url=v.url0, title=v.title0, category=self.category0, description=v.description0, owner=self.user0) query.save()
def test_save_bookmark_same_url_by_another_user(self): query = Bookmark(url=v.url0, title=v.title0, category=self.category0, description=v.description0, owner=self.user1) query.save() query = Bookmark.objects.get(url=v.url0, owner=self.user1) self.assertEqual(query.__str__(), v.title0)
def test_update_bookmark(self): query = Bookmark(id=self.bookmark.id, url=v.url1, title=v.title1, category=self.category1, description=v.description1, owner=self.user0) query.save() query2 = Bookmark.objects.get(url=v.url1, owner=self.user0) self.assertEqual(query2.id, self.bookmark.id)
def add_item(**kwargs): """ Adding bookmark. Arguments: url: feed entry url title: feed entry title category: feed etnry category owner: feed subscriber user Return: (Bool, error message) """ category = Category.objects.get(category=kwargs.get('category')) owner = User.objects.get(username=kwargs.get('owner')) bookmark = Bookmark(url=kwargs.get('url'), title=kwargs.get('title'), category=category, owner=owner) try: bookmark.save() return True, '' except IntegrityError as error: return False, "already registered: %s" % error