mirror of
https://github.com/apache/superset.git
synced 2024-09-19 12:09:42 -04:00
57 lines
2.3 KiB
Python
57 lines
2.3 KiB
Python
|
# Licensed to the Apache Software Foundation (ASF) under one
|
||
|
# or more contributor license agreements. See the NOTICE file
|
||
|
# distributed with this work for additional information
|
||
|
# regarding copyright ownership. The ASF licenses this file
|
||
|
# to you under the Apache License, Version 2.0 (the
|
||
|
# "License"); you may not use this file except in compliance
|
||
|
# with the License. You may obtain a copy of the License at
|
||
|
#
|
||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
#
|
||
|
# Unless required by applicable law or agreed to in writing,
|
||
|
# software distributed under the License is distributed on an
|
||
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
||
|
# KIND, either express or implied. See the License for the
|
||
|
# specific language governing permissions and limitations
|
||
|
# under the License.
|
||
|
|
||
|
from unittest.mock import call, Mock, patch
|
||
|
|
||
|
from superset.extensions import machine_auth_provider_factory
|
||
|
from tests.base_tests import SupersetTestCase
|
||
|
|
||
|
|
||
|
class MachineAuthProviderTests(SupersetTestCase):
|
||
|
def test_get_auth_cookies(self):
|
||
|
user = self.get_user("admin")
|
||
|
auth_cookies = machine_auth_provider_factory.instance.get_auth_cookies(user)
|
||
|
self.assertIsNotNone(auth_cookies["session"])
|
||
|
|
||
|
@patch("superset.utils.machine_auth.MachineAuthProvider.get_auth_cookies")
|
||
|
def test_auth_driver_user(self, get_auth_cookies):
|
||
|
user = self.get_user("admin")
|
||
|
driver = Mock()
|
||
|
get_auth_cookies.return_value = {
|
||
|
"session": "session_val",
|
||
|
"other_cookie": "other_val",
|
||
|
}
|
||
|
machine_auth_provider_factory.instance.authenticate_webdriver(driver, user)
|
||
|
driver.add_cookie.assert_has_calls(
|
||
|
[
|
||
|
call({"name": "session", "value": "session_val"}),
|
||
|
call({"name": "other_cookie", "value": "other_val"}),
|
||
|
]
|
||
|
)
|
||
|
|
||
|
@patch("superset.utils.machine_auth.request")
|
||
|
def test_auth_driver_request(self, request):
|
||
|
driver = Mock()
|
||
|
request.cookies = {"session": "session_val", "other_cookie": "other_val"}
|
||
|
machine_auth_provider_factory.instance.authenticate_webdriver(driver, None)
|
||
|
driver.add_cookie.assert_has_calls(
|
||
|
[
|
||
|
call({"name": "session", "value": "session_val"}),
|
||
|
call({"name": "other_cookie", "value": "other_val"}),
|
||
|
]
|
||
|
)
|