Skip to content

Commit 311c8c0

Browse files
author
Brett Chaldecott
committed
test: add framework initialization mocks for OAuth tests
1 parent fac5536 commit 311c8c0

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

testv2/testv2_auth/test_oauth.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -96,6 +96,21 @@ def mock_get_side_effect(key):
9696
# Make session manager return token manager
9797
self.oauth._session_manager.get_token_manager = MagicMock(return_value=self.mock_token_manager)
9898

99+
# Mock framework
100+
self.mock_framework = MagicMock()
101+
self.mock_framework.get_name.return_value = "flask"
102+
self.mock_framework.get_description.return_value = "Flask framework implementation"
103+
self.mock_framework.start = MagicMock()
104+
self.mock_framework.stop = MagicMock()
105+
self.mock_framework.get_app = MagicMock(return_value=None)
106+
self.mock_framework.get_request = MagicMock(return_value=None)
107+
self.mock_framework.get_user_id = MagicMock(return_value=None)
108+
self.mock_framework.set_oauth = MagicMock()
109+
self.mock_framework.can_auto_detect = MagicMock(return_value=True)
110+
111+
# Set the framework in OAuth
112+
self.oauth._framework = self.mock_framework
113+
99114
# Test for None value handling in auth_params
100115
def test_none_values_in_auth_params(self):
101116
"""Test how None values are handled in auth_params"""

0 commit comments

Comments
 (0)