diff --git a/src/galaxy/api/types.py b/src/galaxy/api/types.py index fb3b908..21466ac 100644 --- a/src/galaxy/api/types.py +++ b/src/galaxy/api/types.py @@ -204,5 +204,5 @@ class GameTime(): :param last_time_played: last time the game was played (**unix timestamp**) """ game_id: str - time_played: int - last_played_time: int + time_played: Optional[int] + last_played_time: Optional[int] diff --git a/tests/test_game_times.py b/tests/test_game_times.py index 8f5d5ae..a9f11f7 100644 --- a/tests/test_game_times.py +++ b/tests/test_game_times.py @@ -16,7 +16,8 @@ def test_success(plugin, read, write): read.side_effect = [json.dumps(request).encode() + b"\n", b""] plugin.get_game_times.coro.return_value = [ GameTime("3", 60, 1549550504), - GameTime("5", 10, 1549550502) + GameTime("5", 10, None), + GameTime("7", None, 1549550502), ] asyncio.run(plugin.run()) plugin.get_game_times.assert_called_with() @@ -35,7 +36,10 @@ def test_success(plugin, read, write): { "game_id": "5", "time_played": 10, - "last_played_time": 1549550502 + }, + { + "game_id": "7", + "last_played_time": 1549550502 } ] }