Exemplo n.º 1
0
class TestUser(unittest.TestCase):

    def setUp(self):
        self.user = User(
            'testuser',
            'password'
        )

        self.user.set_new_user()

    def tearDown(self):
        self.db = Database()
        self.db.make_query(
            '''
            DELETE FROM user WHERE username = "******"
            '''.format('testuser')
        )

    def test_check_username(self):
        self.assertTrue(
            self.user.check_username()
        )

    def test_check_password(self):
        self.assertTrue(
            self.user.check_password()
        )