fix: respect chart cache timeout setting (#21637)
diff --git a/tests/integration_tests/charts/data/api_tests.py b/tests/integration_tests/charts/data/api_tests.py
index 847c969..d360b15 100644
--- a/tests/integration_tests/charts/data/api_tests.py
+++ b/tests/integration_tests/charts/data/api_tests.py
@@ -21,7 +21,7 @@
import copy
from datetime import datetime
from io import BytesIO
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, Optional, List
from unittest import mock
from zipfile import ZipFile
diff --git a/tests/integration_tests/fixtures/energy_dashboard.py b/tests/integration_tests/fixtures/energy_dashboard.py
index c2f82b1..73f2f1e 100644
--- a/tests/integration_tests/fixtures/energy_dashboard.py
+++ b/tests/integration_tests/fixtures/energy_dashboard.py
@@ -15,7 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import random
-from typing import Dict, Set
+from typing import Dict, List, Set
import pandas as pd
import pytest
@@ -69,7 +69,7 @@
return pd.DataFrame.from_dict(data)
-def _create_energy_table():
+def _create_energy_table() -> List[Slice]:
table = create_table_metadata(
table_name=ENERGY_USAGE_TBL_NAME,
database=get_example_database(),
@@ -87,13 +87,17 @@
db.session.commit()
table.fetch_metadata()
+ slices = []
for slice_data in _get_energy_slices():
- _create_and_commit_energy_slice(
+
+ slice = _create_and_commit_energy_slice(
table,
slice_data["slice_title"],
slice_data["viz_type"],
slice_data["params"],
)
+ slices.append(slice)
+ return slices
def _create_and_commit_energy_slice(
