diff --git a/tests/test_integration/test_graphql_query.py b/tests/test_integration/test_graphql_query.py index a2d6234c9..5db42cc91 100644 --- a/tests/test_integration/test_graphql_query.py +++ b/tests/test_integration/test_graphql_query.py @@ -168,7 +168,7 @@ def test_query_with_datapoints(turbine_client: WindTurbineClient) -> None: temperature{ externalId name - getDataPoints(granularity: "1d", aggregates: SUM){ + getDataPoints(granularity: "1d", aggregates: SUM, first: 100){ items{ timestamp sum @@ -182,8 +182,13 @@ def test_query_with_datapoints(turbine_client: WindTurbineClient) -> None: ) assert len(result) == 1 item = result[0] - # Currently (16/11/2024 - there is now datapoints in the timeseries) assert isinstance(item, wdc.MetmastGraphQL) + assert item.temperature is not None + assert item.temperature.data is not None + assert item.temperature.data.sum is not None + assert len(item.temperature.data.sum) > 0 + assert item.temperature.data.timestamp is not None + assert len(item.temperature.data.timestamp) > 0 def test_query_latest_datapoint(turbine_client: WindTurbineClient) -> None: @@ -209,5 +214,10 @@ def test_query_latest_datapoint(turbine_client: WindTurbineClient) -> None: ) assert len(result) == 1 item = result[0] - # Currently (16/11/2024 - there is now datapoints in the timeseries) assert isinstance(item, wdc.MetmastGraphQL) + assert item.temperature is not None + assert item.temperature.data is not None + assert item.temperature.data.timestamp is not None + assert item.temperature.data.value is not None + assert len(item.temperature.data.timestamp) == 1 + assert len(item.temperature.data.value) == 1