diff --git a/src/rivian/rivian.py b/src/rivian/rivian.py index 1905376..aacc332 100644 --- a/src/rivian/rivian.py +++ b/src/rivian/rivian.py @@ -477,15 +477,15 @@ async def get_live_charging_session( for p in properties ) graphql_query = f""" - query getLiveSessionData($vehicleId: ID!) {{ - getLiveSessionData(vehicleId: $vehicleId) {{ + query getLiveSessionHistory($vehicleId: ID!) {{ + getLiveSessionHistory(vehicleId: $vehicleId) {{ __typename {fragment} }} }}""" graphql_json = { - "operationName": "getLiveSessionData", + "operationName": "getLiveSessionHistory", "query": graphql_query, "variables": {"vehicleId": vin}, } diff --git a/src/rivian/schemas/charging.graphql b/src/rivian/schemas/charging.graphql index 63bfe75..74a3c8a 100644 --- a/src/rivian/schemas/charging.graphql +++ b/src/rivian/schemas/charging.graphql @@ -1,7 +1,6 @@ type Query { chargepoint: ChargepointQuery getCompletedSessionSummaries($vehicleId: String): [ChargingSessionSummary] - getLiveSessionData(vehicleId: ID!): LiveSessionData getLiveSessionHistory(vehicleId: ID!): LiveSessionHistory getNonRivianUserSession: NonRivianUserSessionData getRegisteredWallboxes: [WallboxRecord] diff --git a/tests/responses.py b/tests/responses.py index d5dc6cc..f953f68 100644 --- a/tests/responses.py +++ b/tests/responses.py @@ -37,7 +37,7 @@ } LIVE_CHARGING_SESSION_RESPONSE = { "data": { - "getLiveSessionData": { + "getLiveSessionHistory": { "isRivianCharger": None, "isFreeSession": None, "vehicleChargerState": { diff --git a/tests/rivian_test.py b/tests/rivian_test.py index 7eb0f76..e1c0e88 100644 --- a/tests/rivian_test.py +++ b/tests/rivian_test.py @@ -175,7 +175,7 @@ async def test_get_vehicle_state(aresponses: ResponsesMockServer) -> None: async def test_get_live_charging_session(aresponses: ResponsesMockServer) -> None: - """Test GraphQL Response for a getLiveSessionData request""" + """Test GraphQL Response for a getLiveSessionHistory request""" aresponses.add( "rivian.com", "/api/gql/chrg/user/graphql", @@ -188,7 +188,7 @@ async def test_get_live_charging_session(aresponses: ResponsesMockServer) -> Non response_json = await response.json() assert response.status == 200 assert ( - response_json["data"]["getLiveSessionData"]["vehicleChargerState"]["value"] + response_json["data"]["getLiveSessionHistory"]["vehicleChargerState"]["value"] == "charging_active" ) await rivian.close()