Skip to content

Commit

Permalink
pr comments
Browse files Browse the repository at this point in the history
  • Loading branch information
AndrewChubatiuk committed May 1, 2024
1 parent b6cee83 commit d743625
Show file tree
Hide file tree
Showing 3 changed files with 28 additions and 6 deletions.
27 changes: 24 additions & 3 deletions redash/handlers/queries.py
Original file line number Diff line number Diff line change
Expand Up @@ -150,7 +150,14 @@ def get(self):
page = request.args.get("page", 1, type=int)
per_page = request.args.get("per_page", 25, type=int)

results = paginate(ordered_results, page=page, per_page=per_page, serializer=QuerySerializer)
results = paginate(
ordered_results,
page=page,
per_page=per_page,
serializer=QuerySerializer,
with_stats=True,
with_last_modified_by=False,
)

if search_term:
self.record_event({"action": "search", "object_type": "query", "term": search_term})
Expand Down Expand Up @@ -290,7 +297,14 @@ def get(self):

page = request.args.get("page", 1, type=int)
per_page = request.args.get("per_page", 25, type=int)
return paginate(ordered_results, page=page, per_page=per_page, serializer=QuerySerializer)
return paginate(
ordered_results,
page=page,
per_page=per_page,
serializer=QuerySerializer,
with_stats=True,
with_last_modified_by=False,
)


class QueryResource(BaseResource):
Expand Down Expand Up @@ -483,7 +497,14 @@ def get(self):

page = request.args.get("page", 1, type=int)
per_page = request.args.get("per_page", 25, type=int)
results = paginate(ordered_favorites, page=page, per_page=per_page, serializer=QuerySerializer)
results = paginate(
ordered_favorites,
page=page,
per_page=per_page,
serializer=QuerySerializer,
with_stats=True,
with_last_modified_by=False,
)

self.record_event(
{
Expand Down
1 change: 1 addition & 0 deletions redash/tasks/queries/execution.py
Original file line number Diff line number Diff line change
Expand Up @@ -165,6 +165,7 @@ def __init__(self, query, data_source_id, user_id, is_api_key, metadata, is_sche
self.query_hash = gen_query_hash(self.query)
self.is_scheduled_query = is_scheduled_query
if self.is_scheduled_query:
self.query_model = models.db.session.merge(self.query_model, load=False)
# Load existing tracker or create a new one if the job was created before code update:
models.scheduled_queries_executions.update(self.query_model.id)

Expand Down
6 changes: 3 additions & 3 deletions tests/test_authentication.py
Original file line number Diff line number Diff line change
Expand Up @@ -443,11 +443,11 @@ def tearDown(self):
org_settings["auth_jwt_auth_audience"] = ""
org_settings["auth_jwt_auth_header_name"] = ""

def jwt_no_token(self):
def test_jwt_no_token(self):
response = self.get_request("/data_sources", org=self.factory.org)
self.assertEqual(response.status_code, 302)

def jwt_from_pem_file(self):
def test_jwt_from_pem_file(self):
user = self.factory.create_user()

issued_at_timestamp = time.time()
Expand All @@ -468,7 +468,7 @@ def jwt_from_pem_file(self):
self.assertEqual(response.status_code, 200)

@patch.object(requests, "get")
def jwk_decode(self, mock_get):
def test_jwk_decode(self, mock_get):
with open(self.rsa_public_key, "rb") as keyfile:
public_key = jwcrypto.jwk.JWK.from_pem(keyfile.read())
jwk_keys = {"keys": [json.loads(public_key.export())]}
Expand Down

0 comments on commit d743625

Please sign in to comment.