From b3a9b90343f70584b82df00a42cbcc5b97dbfc7f Mon Sep 17 00:00:00 2001 From: BonfaceKilz Date: Mon, 1 Nov 2021 09:22:23 +0300 Subject: use gn3 imports --- wqflask/tests/unit/wqflask/test_resource_manager.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) (limited to 'wqflask/tests') diff --git a/wqflask/tests/unit/wqflask/test_resource_manager.py b/wqflask/tests/unit/wqflask/test_resource_manager.py index 01fb2021..f4335425 100644 --- a/wqflask/tests/unit/wqflask/test_resource_manager.py +++ b/wqflask/tests/unit/wqflask/test_resource_manager.py @@ -3,10 +3,10 @@ import json import unittest from unittest import mock -from wqflask.resource_manager import get_user_membership -from wqflask.resource_manager import get_user_access_roles -from wqflask.resource_manager import DataRole -from wqflask.resource_manager import AdminRole +from gn3.authentication import get_user_membership +from gn3.authentication import get_highest_user_access_role +from gn3.authentication import DataRole +from gn3.authentication import AdminRole class TestGetUserMembership(unittest.TestCase): @@ -56,7 +56,7 @@ class TestGetUserMembership(unittest.TestCase): class TestCheckUserAccessRole(unittest.TestCase): - """Test cases for `get_user_access_roles`""" + """Test cases for `get_highest_user_access_role`""" @mock.patch("wqflask.resource_manager.requests.get") def test_edit_access(self, requests_mock): @@ -69,7 +69,7 @@ class TestCheckUserAccessRole(unittest.TestCase): } )) type(requests_mock.return_value).content = response - self.assertEqual(get_user_access_roles( + self.assertEqual(get_highest_user_access_role( resource_id="0196d92e1665091f202f", user_id="8ad942fe-490d-453e-bd37"), {"data": DataRole.EDIT, @@ -86,7 +86,7 @@ class TestCheckUserAccessRole(unittest.TestCase): } )) type(requests_mock.return_value).content = response - self.assertEqual(get_user_access_roles( + self.assertEqual(get_highest_user_access_role( resource_id="0196d92e1665091f202f", user_id=""), {"data": DataRole.NO_ACCESS, -- cgit v1.2.3