From d5dd7f44a0c8b011a9959ede93a018ec37a2f22f Mon Sep 17 00:00:00 2001 From: Diogo Sousa Date: Thu, 2 Feb 2023 12:45:01 +0000 Subject: [PATCH] test remove --- tests/openbb_terminal/session/test_local_model.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/tests/openbb_terminal/session/test_local_model.py b/tests/openbb_terminal/session/test_local_model.py index 526bd09215d..a9c3fad1db9 100644 --- a/tests/openbb_terminal/session/test_local_model.py +++ b/tests/openbb_terminal/session/test_local_model.py @@ -29,8 +29,17 @@ def test_get_session(): "uuid": "test_uuid", } open_mock = mock_open(read_data=json.dumps(data)) - with patch("openbb_terminal.session.local_model.open", open_mock, create=True): - assert local_model.get_session() == data + with patch("openbb_terminal.session.local_model.os.path") as path_mock: + with patch("openbb_terminal.session.local_model.open", open_mock, create=True): + assert local_model.get_session() == data + path_mock.isfile.assert_called_with(local_model.SESSION_FILE_PATH) open_mock.assert_called_with(local_model.SESSION_FILE_PATH) open_mock.return_value.read.assert_called_once() + + +def test_remove_session_file(): + with patch("openbb_terminal.session.local_model.os") as os_mock: + local_model.remove_session_file() + + os_mock.remove.assert_called_with(local_model.SESSION_FILE_PATH)