def test_count(self, query): count = Department.count(where_clause="Active=True", qb=self.qb_client) query.assert_called_once_with( "SELECT COUNT(*) FROM Department WHERE Active=True")