# -*- coding: utf-8 -*- from unittest import mock from bthlabs_jsonrpc_django import auth_checks def test_has_perms_regular_user(rf, user): # Given request = rf.get('/') request.user = user check = auth_checks.has_perms(['can_use_rpc']) # When result = check(request) # Then assert result is False def test_has_perms_ok(rf, user): # Given request = rf.get('/') request.user = user check = auth_checks.has_perms(['can_use_rpc']) with mock.patch.object(user, 'has_perms') as mock_has_perms: mock_has_perms.return_value = True # When result = check(request) # Then assert result is True mock_has_perms.assert_called_with(['can_use_rpc']) def test_has_perms_ok_super_user(rf, super_user): # Given request = rf.get('/') request.user = super_user check = auth_checks.has_perms(['can_use_rpc']) # When result = check(request) # Then assert result is True