mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 04:52:07 +00:00
save session test
This commit is contained in:
22
tests/openbb_terminal/session/test_local_model.py
Normal file
22
tests/openbb_terminal/session/test_local_model.py
Normal file
@@ -0,0 +1,22 @@
|
||||
# IMPORTATION STANDARD
|
||||
|
||||
# IMPORTATION THIRDPARTY
|
||||
import json
|
||||
from unittest.mock import patch, mock_open
|
||||
|
||||
# IMPORTATION INTERNAL
|
||||
from openbb_terminal.session import local_model
|
||||
|
||||
|
||||
def test_save_good_session():
|
||||
open_mock = mock_open()
|
||||
with patch("openbb_terminal.session.local_model.open", open_mock, create=True):
|
||||
data = {
|
||||
"access_token": "test_token",
|
||||
"token_type": "bearer",
|
||||
"uuid": "test_uuid",
|
||||
}
|
||||
local_model.save_session(data=data)
|
||||
|
||||
open_mock.assert_called_with(local_model.SESSION_FILE_PATH, "w")
|
||||
open_mock.return_value.write.assert_called_once_with(json.dumps(data))
|
||||
Reference in New Issue
Block a user