jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-10-29 05:21:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

test remove

Browse Source
This commit is contained in:
Diogo Sousa
2023-02-02 12:45:01 +00:00
parent 57b4ba8bd1
commit d5dd7f44a0
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/openbb_terminal/session/test_local_model.py
View File

@@ -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)