Exemplo n.º 1
0
    def test_add_note(self):
        u1 = User(username='******', email='*****@*****.**')
        db.session.add(u1)
        db.session.commit()
        self.assertEqual(u1.notes.count(), 0)

        u1.add_note('test title', 'test description')
        self.assertEqual(u1.notes.count(), 1)
        self.assertEqual(u1.notes.filter_by(id=1).first().title, 'test title')