mirror of
https://github.com/apache/superset.git
synced 2026-04-19 08:04:53 +00:00
chore: set up ruff as a new linter/formatter (#28158)
This commit is contained in:
committed by
GitHub
parent
e8a678b75a
commit
2d63722150
@@ -19,7 +19,7 @@ from unittest.mock import patch
|
||||
|
||||
import pytest
|
||||
from flask_appbuilder.security.sqla.models import User
|
||||
from sqlalchemy.orm import Session
|
||||
from sqlalchemy.orm import Session # noqa: F401
|
||||
|
||||
from superset import db
|
||||
from superset.commands.explore.form_data.state import TemporaryExploreState
|
||||
@@ -28,8 +28,8 @@ from superset.explore.exceptions import DatasetAccessDeniedError
|
||||
from superset.extensions import cache_manager
|
||||
from superset.models.slice import Slice
|
||||
from tests.integration_tests.fixtures.world_bank_dashboard import (
|
||||
load_world_bank_dashboard_with_slices,
|
||||
load_world_bank_data,
|
||||
load_world_bank_dashboard_with_slices, # noqa: F401
|
||||
load_world_bank_data, # noqa: F401
|
||||
)
|
||||
from tests.integration_tests.test_app import app
|
||||
|
||||
@@ -38,22 +38,22 @@ FORM_DATA = {"test": "test value"}
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def chart_id(load_world_bank_dashboard_with_slices) -> int:
|
||||
with app.app_context() as ctx:
|
||||
def chart_id(load_world_bank_dashboard_with_slices) -> int: # noqa: F811
|
||||
with app.app_context(): # noqa: F841
|
||||
chart = db.session.query(Slice).filter_by(slice_name="World's Population").one()
|
||||
return chart.id
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def admin_id() -> int:
|
||||
with app.app_context() as ctx:
|
||||
with app.app_context(): # noqa: F841
|
||||
admin = db.session.query(User).filter_by(username="admin").one()
|
||||
return admin.id
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def dataset() -> int:
|
||||
with app.app_context() as ctx:
|
||||
with app.app_context(): # noqa: F841
|
||||
dataset = (
|
||||
db.session.query(SqlaTable)
|
||||
.filter_by(table_name="wb_health_population")
|
||||
@@ -79,9 +79,9 @@ def assert_dataset(result, dataset_id):
|
||||
dataset = result["dataset"]
|
||||
assert dataset["id"] == dataset_id
|
||||
assert dataset["datasource_name"] == "wb_health_population"
|
||||
assert dataset["is_sqllab_view"] == False
|
||||
assert dataset["is_sqllab_view"] is False # noqa: E712
|
||||
assert dataset["main_dttm_col"] == "year"
|
||||
assert dataset["sql"] == None
|
||||
assert dataset["sql"] is None # noqa: E711
|
||||
assert dataset["type"] == "table"
|
||||
assert dataset["uid"] == f"{dataset_id}__table"
|
||||
|
||||
@@ -90,7 +90,7 @@ def assert_dataset(result, dataset_id):
|
||||
def assert_slice(result, chart_id, dataset_id):
|
||||
slice = result["slice"]
|
||||
assert slice["edit_url"] == f"/chart/edit/{chart_id}"
|
||||
assert slice["is_managed_externally"] == False
|
||||
assert slice["is_managed_externally"] is False # noqa: E712
|
||||
assert slice["slice_id"] == chart_id
|
||||
assert slice["slice_name"] == "World's Population"
|
||||
assert slice["form_data"]["datasource"] == f"{dataset_id}__table"
|
||||
@@ -98,14 +98,14 @@ def assert_slice(result, chart_id, dataset_id):
|
||||
|
||||
|
||||
def test_no_params_provided(test_client, login_as_admin):
|
||||
resp = test_client.get(f"api/v1/explore/")
|
||||
resp = test_client.get("api/v1/explore/") # noqa: F541
|
||||
assert resp.status_code == 200
|
||||
data = json.loads(resp.data.decode("utf-8"))
|
||||
result = data.get("result")
|
||||
assert result["dataset"]["name"] == "[Missing Dataset]"
|
||||
assert result["form_data"]["datasource"] == "None__table"
|
||||
assert result["message"] == None
|
||||
assert result["slice"] == None
|
||||
assert result["message"] is None # noqa: E711
|
||||
assert result["slice"] is None # noqa: E711
|
||||
|
||||
|
||||
def test_get_from_cache(test_client, login_as_admin, dataset):
|
||||
@@ -118,8 +118,8 @@ def test_get_from_cache(test_client, login_as_admin, dataset):
|
||||
assert_dataset(result, dataset.id)
|
||||
assert result["form_data"]["datasource"] == f"{dataset.id}__table"
|
||||
assert result["form_data"]["test"] == "test value"
|
||||
assert result["message"] == None
|
||||
assert result["slice"] == None
|
||||
assert result["message"] is None # noqa: E711
|
||||
assert result["slice"] is None # noqa: E711
|
||||
|
||||
|
||||
def test_get_from_cache_unknown_key_chart_id(
|
||||
@@ -155,7 +155,7 @@ def test_get_from_cache_unknown_key_dataset(test_client, login_as_admin, dataset
|
||||
result["message"]
|
||||
== "Form data not found in cache, reverting to dataset metadata."
|
||||
)
|
||||
assert result["slice"] == None
|
||||
assert result["slice"] is None # noqa: E711
|
||||
|
||||
|
||||
def test_get_from_cache_unknown_key_no_extra_parameters(test_client, login_as_admin):
|
||||
@@ -166,8 +166,8 @@ def test_get_from_cache_unknown_key_no_extra_parameters(test_client, login_as_ad
|
||||
result = data.get("result")
|
||||
assert result["dataset"]["name"] == "[Missing Dataset]"
|
||||
assert result["form_data"]["datasource"] == "None__table"
|
||||
assert result["message"] == None
|
||||
assert result["slice"] == None
|
||||
assert result["message"] is None # noqa: E711
|
||||
assert result["slice"] is None # noqa: E711
|
||||
|
||||
|
||||
def test_get_from_permalink(test_client, login_as_admin, chart_id, dataset):
|
||||
@@ -176,7 +176,7 @@ def test_get_from_permalink(test_client, login_as_admin, chart_id, dataset):
|
||||
"datasource": f"{dataset.id}__{dataset.type}",
|
||||
**FORM_DATA,
|
||||
}
|
||||
resp = test_client.post(f"api/v1/explore/permalink", json={"formData": form_data})
|
||||
resp = test_client.post("api/v1/explore/permalink", json={"formData": form_data}) # noqa: F541
|
||||
data = json.loads(resp.data.decode("utf-8"))
|
||||
permalink_key = data["key"]
|
||||
resp = test_client.get(f"api/v1/explore/?permalink_key={permalink_key}")
|
||||
@@ -186,8 +186,8 @@ def test_get_from_permalink(test_client, login_as_admin, chart_id, dataset):
|
||||
assert_dataset(result, dataset.id)
|
||||
assert result["form_data"]["datasource"] == f"{dataset.id}__table"
|
||||
assert result["form_data"]["test"] == "test value"
|
||||
assert result["message"] == None
|
||||
assert result["slice"] == None
|
||||
assert result["message"] is None # noqa: E711
|
||||
assert result["slice"] is None # noqa: E711
|
||||
|
||||
|
||||
def test_get_from_permalink_unknown_key(test_client, login_as_admin):
|
||||
|
||||
Reference in New Issue
Block a user