コード例 #1
0
    def testUserIsAdmin(self):

        '''
        userIsAdmin() should check if user with a given id is an admin.
        '''

        admin = User('admin', 'password', True)
        user = User('user', 'password', False)
        db.session.add(admin)
        db.session.add(user)
        db.session.commit()

        # True with an administrator, false with user.
        self.assertTrue(User.userIsAdmin(admin.id))
        self.assertFalse(User.userIsAdmin(user.id))

        # False if an id is passed for which there is no record.
        self.assertFalse(User.userIsAdmin(3))