Method query

test/backend/database/test_token_db.py:220–225

(self, model_class)

        pass

    def query(self, model_class):
        if model_class == MockUserTokenInfo:
            return MockQuery(model_class, self._tokens)
        if model_class == MockUserTokenUsageLog:
            return MockQuery(model_class, self._usage_logs)
        return MockQuery(model_class, [])


@pytest.fixture