89 lines
3.4 KiB
Python
89 lines
3.4 KiB
Python
|
import json
|
||
|
|
import mock
|
||
|
|
from buildbot.process.properties import Secret
|
||
|
|
from buildbot.test.util.config import ConfigErrorsMixin
|
||
|
|
from buildbot.test.util.misc import TestReactorMixin
|
||
|
|
from buildbot.test.util import www
|
||
|
|
from twisted.internet import defer
|
||
|
|
from twisted.trial import unittest
|
||
|
|
from buildbot.secrets.manager import SecretManager
|
||
|
|
from buildbot.test.fake.secrets import FakeSecretStorage
|
||
|
|
|
||
|
|
from buildbot_gitea.auth import GiteaAuth
|
||
|
|
|
||
|
|
try:
|
||
|
|
import requests
|
||
|
|
except ImportError:
|
||
|
|
requests = None
|
||
|
|
|
||
|
|
|
||
|
|
class FakeResponse:
|
||
|
|
|
||
|
|
def __init__(self, _json):
|
||
|
|
self.json = lambda: _json
|
||
|
|
self.content = json.dumps(_json)
|
||
|
|
|
||
|
|
def raise_for_status(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class TestGiteaAuth(TestReactorMixin, www.WwwTestMixin, ConfigErrorsMixin,
|
||
|
|
unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
self.setUpTestReactor()
|
||
|
|
if requests is None:
|
||
|
|
raise unittest.SkipTest("Need to install requests to test oauth2")
|
||
|
|
|
||
|
|
self.patch(requests, 'request', mock.Mock(spec=requests.request))
|
||
|
|
self.patch(requests, 'post', mock.Mock(spec=requests.post))
|
||
|
|
self.patch(requests, 'get', mock.Mock(spec=requests.get))
|
||
|
|
|
||
|
|
self.giteaAuth = GiteaAuth(
|
||
|
|
'https://gitea.test',
|
||
|
|
'client-id',
|
||
|
|
'client-secret')
|
||
|
|
self._master = master = self.make_master(
|
||
|
|
url='h:/a/b/', auth=self.giteaAuth)
|
||
|
|
self.giteaAuth.reconfigAuth(master, master.config)
|
||
|
|
|
||
|
|
self.giteaAuth_secret = GiteaAuth(
|
||
|
|
'https://gitea.test',
|
||
|
|
Secret("client-id"),
|
||
|
|
Secret("client-secret"))
|
||
|
|
self._master = master = self.make_master(
|
||
|
|
url='h:/a/b/', auth=self.giteaAuth_secret)
|
||
|
|
fake_storage_service = FakeSecretStorage()
|
||
|
|
fake_storage_service.reconfigService(
|
||
|
|
secretdict={
|
||
|
|
"client-id": "secretClientId",
|
||
|
|
"client-secret": "secretClientSecret"
|
||
|
|
})
|
||
|
|
secret_service = SecretManager()
|
||
|
|
secret_service.services = [fake_storage_service]
|
||
|
|
secret_service.setServiceParent(self._master)
|
||
|
|
self.giteaAuth_secret.reconfigAuth(master, master.config)
|
||
|
|
|
||
|
|
@defer.inlineCallbacks
|
||
|
|
def test_getGiteaLoginURL(self):
|
||
|
|
res = yield self.giteaAuth.getLoginURL('http://redir')
|
||
|
|
exp = ("https://gitea.test/login/oauth/authorize?client_id=client-id&"
|
||
|
|
"redirect_uri=h%3A%2Fa%2Fb%2Fauth%2Flogin&response_type=code&"
|
||
|
|
"state=redirect%3Dhttp%253A%252F%252Fredir")
|
||
|
|
self.assertEqual(res, exp)
|
||
|
|
res = yield self.giteaAuth.getLoginURL(None)
|
||
|
|
exp = ("https://gitea.test/login/oauth/authorize?client_id=client-id&"
|
||
|
|
"redirect_uri=h%3A%2Fa%2Fb%2Fauth%2Flogin&response_type=code")
|
||
|
|
self.assertEqual(res, exp)
|
||
|
|
|
||
|
|
@defer.inlineCallbacks
|
||
|
|
def test_getGiteaLoginURL_with_secret(self):
|
||
|
|
res = yield self.giteaAuth_secret.getLoginURL('http://redir')
|
||
|
|
exp = ("https://gitea.test/login/oauth/authorize?client_id=secretClientId&"
|
||
|
|
"redirect_uri=h%3A%2Fa%2Fb%2Fauth%2Flogin&response_type=code&"
|
||
|
|
"state=redirect%3Dhttp%253A%252F%252Fredir")
|
||
|
|
self.assertEqual(res, exp)
|
||
|
|
res = yield self.giteaAuth_secret.getLoginURL(None)
|
||
|
|
exp = ("https://gitea.test/login/oauth/authorize?client_id=secretClientId&"
|
||
|
|
"redirect_uri=h%3A%2Fa%2Fb%2Fauth%2Flogin&response_type=code")
|
||
|
|
self.assertEqual(res, exp)
|