예제 #1
0
    def test_bookmarker(self):
        with test_db.atomic():
            number = 6555
            title = "Happy Eyeballs test"
            text = "testing bookmark"
            category = "Best Current Practice"
            bookmark = False
            Data.create(
                number=number,
                title=title,
                text=text,
                category=category,
                bookmark=bookmark,
            )
            DataIndex.create(rowid=number,
                             title=title,
                             text=text,
                             category=category)

        query = Data.select().where(Data.number == 6555)
        for result in query:
            self.assertEqual(result.bookmark, False)
            Data.insert(
                title=result.title,
                text=result.text,
                number=result.number,
                category=result.category,
                bookmark=1,
            ).on_conflict("replace").execute()
        new = Data.select().where(Data.number == 6555)
        for result in new:
            self.assertEqual(result.bookmark, True)
예제 #2
0
 def test_delete_bookmark(self):
     exists = Data.select().where(Data.bookmark == 1
                                  and Data.number == 7540)
     for result in exists:
         self.assertEqual(result.bookmark, True)
         Data.insert(
             title=result.title,
             text=result.text,
             number=result.number,
             category=result.category,
             bookmark=0,
         ).on_conflict("replace").execute()
     new = Data.select().where(Data.bookmark == 1 and Data.number == 7540)
     for x in new:
         self.assertEqual(x.bookmark, False)