# -*- coding: utf-8 -*-
#
# Copyright (C) 2014 Mika Menp ,
# Tampere University of Technology
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with this program. If not, see .
from __future__ import print_function
import pickle
try:
import unittest
except ImportError:
import unittest2 as unittest
from httmock import HTTMock # noqa
from httmock import response # noqa
from httmock import urlmatch # noqa
import requests
import six
import gitlab
from gitlab import * # noqa
class TestSanitize(unittest.TestCase):
def test_do_nothing(self):
self.assertEqual(1, gitlab._sanitize(1))
self.assertEqual(1.5, gitlab._sanitize(1.5))
self.assertEqual("foo", gitlab._sanitize("foo"))
def test_slash(self):
self.assertEqual("foo%2Fbar", gitlab._sanitize("foo/bar"))
def test_dict(self):
source = {"url": "foo/bar", "id": 1}
expected = {"url": "foo%2Fbar", "id": 1}
self.assertEqual(expected, gitlab._sanitize(source))
class TestGitlabRawMethods(unittest.TestCase):
def setUp(self):
self.gl = Gitlab("http://localhost", private_token="private_token",
email="testuser@test.com", password="testpassword",
ssl_verify=True)
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/known_path",
method="get")
def resp_get(self, url, request):
headers = {'content-type': 'application/json'}
content = 'response'.encode("utf-8")
return response(200, content, headers, None, 5, request)
def test_raw_get_unknown_path(self):
@urlmatch(scheme="http", netloc="localhost",
path="/api/v3/unknown_path",
method="get")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(404, content, headers, None, 5, request)
with HTTMock(resp_cont):
resp = self.gl._raw_get("/unknown_path")
self.assertEqual(resp.status_code, 404)
def test_raw_get_without_kwargs(self):
with HTTMock(self.resp_get):
resp = self.gl._raw_get("/known_path")
self.assertEqual(resp.content, b'response')
self.assertEqual(resp.status_code, 200)
def test_raw_get_with_kwargs(self):
with HTTMock(self.resp_get):
resp = self.gl._raw_get("/known_path", sudo="testing")
self.assertEqual(resp.content, b'response')
self.assertEqual(resp.status_code, 200)
def test_raw_post(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/known_path",
method="post")
def resp_post(url, request):
headers = {'content-type': 'application/json'}
content = 'response'.encode("utf-8")
return response(200, content, headers, None, 5, request)
with HTTMock(resp_post):
resp = self.gl._raw_post("/known_path")
self.assertEqual(resp.content, b'response')
self.assertEqual(resp.status_code, 200)
def test_raw_post_unknown_path(self):
@urlmatch(scheme="http", netloc="localhost",
path="/api/v3/unknown_path",
method="post")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(404, content, headers, None, 5, request)
with HTTMock(resp_cont):
resp = self.gl._raw_post("/unknown_path")
self.assertEqual(resp.status_code, 404)
def test_raw_put(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/known_path",
method="put")
def resp_put(url, request):
headers = {'content-type': 'application/json'}
content = 'response'.encode("utf-8")
return response(200, content, headers, None, 5, request)
with HTTMock(resp_put):
resp = self.gl._raw_put("/known_path")
self.assertEqual(resp.content, b'response')
self.assertEqual(resp.status_code, 200)
def test_raw_put_unknown_path(self):
@urlmatch(scheme="http", netloc="localhost",
path="/api/v3/unknown_path",
method="put")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(404, content, headers, None, 5, request)
with HTTMock(resp_cont):
resp = self.gl._raw_put("/unknown_path")
self.assertEqual(resp.status_code, 404)
def test_raw_delete(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/known_path",
method="delete")
def resp_delete(url, request):
headers = {'content-type': 'application/json'}
content = 'response'.encode("utf-8")
return response(200, content, headers, None, 5, request)
with HTTMock(resp_delete):
resp = self.gl._raw_delete("/known_path")
self.assertEqual(resp.content, b'response')
self.assertEqual(resp.status_code, 200)
def test_raw_delete_unknown_path(self):
@urlmatch(scheme="http", netloc="localhost",
path="/api/v3/unknown_path",
method="delete")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(404, content, headers, None, 5, request)
with HTTMock(resp_cont):
resp = self.gl._raw_delete("/unknown_path")
self.assertEqual(resp.status_code, 404)
class TestGitlabList(unittest.TestCase):
def setUp(self):
self.gl = Gitlab("http://localhost", private_token="private_token",
api_version=4)
def test_build_list(self):
@urlmatch(scheme='http', netloc="localhost", path="/api/v4/tests",
method="get")
def resp_1(url, request):
headers = {'content-type': 'application/json',
'X-Page': 1,
'X-Next-Page': 2,
'X-Per-Page': 1,
'X-Total-Pages': 2,
'X-Total': 2,
'Link': (
';'
' rel="next"')}
content = '[{"a": "b"}]'
return response(200, content, headers, None, 5, request)
@urlmatch(scheme='http', netloc="localhost", path="/api/v4/tests",
method='get', query=r'.*page=2')
def resp_2(url, request):
headers = {'content-type': 'application/json',
'X-Page': 2,
'X-Next-Page': 2,
'X-Per-Page': 1,
'X-Total-Pages': 2,
'X-Total': 2}
content = '[{"c": "d"}]'
return response(200, content, headers, None, 5, request)
with HTTMock(resp_1):
obj = self.gl.http_list('/tests', as_list=False)
self.assertEqual(len(obj), 2)
self.assertEqual(obj._next_url,
'http://localhost/api/v4/tests?per_page=1&page=2')
self.assertEqual(obj.current_page, 1)
self.assertEqual(obj.prev_page, None)
self.assertEqual(obj.next_page, 2)
self.assertEqual(obj.per_page, 1)
self.assertEqual(obj.total_pages, 2)
self.assertEqual(obj.total, 2)
with HTTMock(resp_2):
l = list(obj)
self.assertEqual(len(l), 2)
self.assertEqual(l[0]['a'], 'b')
self.assertEqual(l[1]['c'], 'd')
class TestGitlabHttpMethods(unittest.TestCase):
def setUp(self):
self.gl = Gitlab("http://localhost", private_token="private_token",
api_version=4)
def test_build_url(self):
r = self.gl._build_url('http://localhost/api/v4')
self.assertEqual(r, 'http://localhost/api/v4')
r = self.gl._build_url('https://localhost/api/v4')
self.assertEqual(r, 'https://localhost/api/v4')
r = self.gl._build_url('/projects')
self.assertEqual(r, 'http://localhost/api/v4/projects')
def test_http_request(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v4/projects",
method="get")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '[{"name": "project1"}]'
return response(200, content, headers, None, 5, request)
with HTTMock(resp_cont):
http_r = self.gl.http_request('get', '/projects')
http_r.json()
self.assertEqual(http_r.status_code, 200)
def test_http_request_404(self):
@urlmatch(scheme="http", netloc="localhost",
path="/api/v4/not_there", method="get")
def resp_cont(url, request):
content = {'Here is wh it failed'}
return response(404, content, {}, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabHttpError,
self.gl.http_request,
'get', '/not_there')
def test_get_request(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v4/projects",
method="get")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"name": "project1"}'
return response(200, content, headers, None, 5, request)
with HTTMock(resp_cont):
result = self.gl.http_get('/projects')
self.assertIsInstance(result, dict)
self.assertEqual(result['name'], 'project1')
def test_get_request_raw(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v4/projects",
method="get")
def resp_cont(url, request):
headers = {'content-type': 'application/octet-stream'}
content = 'content'
return response(200, content, headers, None, 5, request)
with HTTMock(resp_cont):
result = self.gl.http_get('/projects')
self.assertEqual(result.content.decode('utf-8'), 'content')
def test_get_request_404(self):
@urlmatch(scheme="http", netloc="localhost",
path="/api/v4/not_there", method="get")
def resp_cont(url, request):
content = {'Here is wh it failed'}
return response(404, content, {}, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabHttpError, self.gl.http_get, '/not_there')
def test_get_request_invalid_data(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v4/projects",
method="get")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '["name": "project1"]'
return response(200, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabParsingError, self.gl.http_get,
'/projects')
def test_list_request(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v4/projects",
method="get")
def resp_cont(url, request):
headers = {'content-type': 'application/json', 'X-Total': 1}
content = '[{"name": "project1"}]'
return response(200, content, headers, None, 5, request)
with HTTMock(resp_cont):
result = self.gl.http_list('/projects', as_list=True)
self.assertIsInstance(result, list)
self.assertEqual(len(result), 1)
with HTTMock(resp_cont):
result = self.gl.http_list('/projects', as_list=False)
self.assertIsInstance(result, GitlabList)
self.assertEqual(len(result), 1)
with HTTMock(resp_cont):
result = self.gl.http_list('/projects', all=True)
self.assertIsInstance(result, list)
self.assertEqual(len(result), 1)
def test_list_request_404(self):
@urlmatch(scheme="http", netloc="localhost",
path="/api/v4/not_there", method="get")
def resp_cont(url, request):
content = {'Here is why it failed'}
return response(404, content, {}, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabHttpError, self.gl.http_list, '/not_there')
def test_list_request_invalid_data(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v4/projects",
method="get")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '["name": "project1"]'
return response(200, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabParsingError, self.gl.http_list,
'/projects')
def test_post_request(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v4/projects",
method="post")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"name": "project1"}'
return response(200, content, headers, None, 5, request)
with HTTMock(resp_cont):
result = self.gl.http_post('/projects')
self.assertIsInstance(result, dict)
self.assertEqual(result['name'], 'project1')
def test_post_request_404(self):
@urlmatch(scheme="http", netloc="localhost",
path="/api/v4/not_there", method="post")
def resp_cont(url, request):
content = {'Here is wh it failed'}
return response(404, content, {}, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabHttpError, self.gl.http_post, '/not_there')
def test_post_request_invalid_data(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v4/projects",
method="post")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '["name": "project1"]'
return response(200, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabParsingError, self.gl.http_post,
'/projects')
def test_put_request(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v4/projects",
method="put")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"name": "project1"}'
return response(200, content, headers, None, 5, request)
with HTTMock(resp_cont):
result = self.gl.http_put('/projects')
self.assertIsInstance(result, dict)
self.assertEqual(result['name'], 'project1')
def test_put_request_404(self):
@urlmatch(scheme="http", netloc="localhost",
path="/api/v4/not_there", method="put")
def resp_cont(url, request):
content = {'Here is wh it failed'}
return response(404, content, {}, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabHttpError, self.gl.http_put, '/not_there')
def test_put_request_invalid_data(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v4/projects",
method="put")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '["name": "project1"]'
return response(200, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabParsingError, self.gl.http_put,
'/projects')
def test_delete_request(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v4/projects",
method="delete")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = 'true'
return response(200, content, headers, None, 5, request)
with HTTMock(resp_cont):
result = self.gl.http_delete('/projects')
self.assertIsInstance(result, requests.Response)
self.assertEqual(result.json(), True)
def test_delete_request_404(self):
@urlmatch(scheme="http", netloc="localhost",
path="/api/v4/not_there", method="delete")
def resp_cont(url, request):
content = {'Here is wh it failed'}
return response(404, content, {}, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabHttpError, self.gl.http_delete,
'/not_there')
class TestGitlabMethods(unittest.TestCase):
def setUp(self):
self.gl = Gitlab("http://localhost", private_token="private_token",
email="testuser@test.com", password="testpassword",
ssl_verify=True)
def test_list(self):
@urlmatch(scheme="http", netloc="localhost",
path="/api/v3/projects/1/repository/branches", method="get")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = ('[{"branch_name": "testbranch", '
'"project_id": 1, "ref": "a"}]').encode("utf-8")
return response(200, content, headers, None, 5, request)
with HTTMock(resp_cont):
data = self.gl.list(ProjectBranch, project_id=1, page=1,
per_page=20)
self.assertEqual(len(data), 1)
data = data[0]
self.assertEqual(data.branch_name, "testbranch")
self.assertEqual(data.project_id, 1)
self.assertEqual(data.ref, "a")
def test_list_next_link(self):
@urlmatch(scheme="http", netloc="localhost",
path='/api/v3/projects/1/repository/branches', method="get")
def resp_one(url, request):
"""First request:
http://localhost/api/v3/projects/1/repository/branches?per_page=1
"""
headers = {
'content-type': 'application/json',
'link': '; rel="next", ; rel="las'
't", ; rel="first"'
}
content = ('[{"branch_name": "otherbranch", '
'"project_id": 1, "ref": "b"}]').encode("utf-8")
resp = response(200, content, headers, None, 5, request)
return resp
@urlmatch(scheme="http", netloc="localhost",
path='/api/v3/projects/1/repository/branches', method="get",
query=r'.*page=2.*')
def resp_two(url, request):
headers = {
'content-type': 'application/json',
'link': '; rel="prev", ; rel="las'
't", ; rel="first"'
}
content = ('[{"branch_name": "testbranch", '
'"project_id": 1, "ref": "a"}]').encode("utf-8")
resp = response(200, content, headers, None, 5, request)
return resp
with HTTMock(resp_two, resp_one):
data = self.gl.list(ProjectBranch, project_id=1, per_page=1,
all=True)
self.assertEqual(data[1].branch_name, "testbranch")
self.assertEqual(data[1].project_id, 1)
self.assertEqual(data[1].ref, "a")
self.assertEqual(data[0].branch_name, "otherbranch")
self.assertEqual(data[0].project_id, 1)
self.assertEqual(data[0].ref, "b")
self.assertEqual(len(data), 2)
def test_list_recursion_limit_caught(self):
@urlmatch(scheme="http", netloc="localhost",
path='/api/v3/projects/1/repository/branches', method="get")
def resp_one(url, request):
"""First request:
http://localhost/api/v3/projects/1/repository/branches?per_page=1
"""
headers = {
'content-type': 'application/json',
'link': '; rel="next", ; rel="las'
't", ; rel="first"'
}
content = ('[{"branch_name": "otherbranch", '
'"project_id": 1, "ref": "b"}]').encode("utf-8")
resp = response(200, content, headers, None, 5, request)
return resp
@urlmatch(scheme="http", netloc="localhost",
path='/api/v3/projects/1/repository/branches', method="get",
query=r'.*page=2.*')
def resp_two(url, request):
# Mock a runtime error
raise RuntimeError("maximum recursion depth exceeded")
with HTTMock(resp_two, resp_one):
data = self.gl.list(ProjectBranch, project_id=1, per_page=1,
safe_all=True)
self.assertEqual(data[0].branch_name, "otherbranch")
self.assertEqual(data[0].project_id, 1)
self.assertEqual(data[0].ref, "b")
self.assertEqual(len(data), 1)
def test_list_recursion_limit_not_caught(self):
@urlmatch(scheme="http", netloc="localhost",
path='/api/v3/projects/1/repository/branches', method="get")
def resp_one(url, request):
"""First request:
http://localhost/api/v3/projects/1/repository/branches?per_page=1
"""
headers = {
'content-type': 'application/json',
'link': '; rel="next", ; rel="las'
't", ; rel="first"'
}
content = ('[{"branch_name": "otherbranch", '
'"project_id": 1, "ref": "b"}]').encode("utf-8")
resp = response(200, content, headers, None, 5, request)
return resp
@urlmatch(scheme="http", netloc="localhost",
path='/api/v3/projects/1/repository/branches', method="get",
query=r'.*page=2.*')
def resp_two(url, request):
# Mock a runtime error
raise RuntimeError("maximum recursion depth exceeded")
with HTTMock(resp_two, resp_one):
with six.assertRaisesRegex(self, GitlabError,
"(maximum recursion depth exceeded)"):
self.gl.list(ProjectBranch, project_id=1, per_page=1, all=True)
def test_list_401(self):
@urlmatch(scheme="http", netloc="localhost",
path="/api/v3/projects/1/repository/branches", method="get")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message":"message"}'.encode("utf-8")
return response(401, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabAuthenticationError, self.gl.list,
ProjectBranch, project_id=1)
def test_list_unknown_error(self):
@urlmatch(scheme="http", netloc="localhost",
path="/api/v3/projects/1/repository/branches", method="get")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message":"message"}'.encode("utf-8")
return response(405, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabListError, self.gl.list,
ProjectBranch, project_id=1)
def test_list_kw_missing(self):
self.assertRaises(GitlabListError, self.gl.list, ProjectBranch)
def test_list_no_connection(self):
self.gl._url = 'http://localhost:66000/api/v3'
self.assertRaises(GitlabConnectionError, self.gl.list, ProjectBranch,
project_id=1)
def test_get(self):
@urlmatch(scheme="http", netloc="localhost",
path="/api/v3/projects/1", method="get")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"name": "testproject"}'.encode("utf-8")
return response(200, content, headers, None, 5, request)
with HTTMock(resp_cont):
data = self.gl.get(Project, id=1)
expected = {"name": "testproject"}
self.assertEqual(expected, data)
def test_get_unknown_path(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/groups/1",
method="get")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(404, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabGetError, self.gl.get, Group, 1)
def test_get_missing_kw(self):
self.assertRaises(GitlabGetError, self.gl.get, ProjectBranch)
def test_get_401(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/projects/1",
method="get")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(401, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabAuthenticationError, self.gl.get,
Project, 1)
def test_get_404(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/projects/1",
method="get")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(404, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabGetError, self.gl.get,
Project, 1)
def test_get_unknown_error(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/projects/1",
method="get")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(405, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabGetError, self.gl.get,
Project, 1)
def test_delete_from_object(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/groups/1",
method="delete")
def resp_delete_group(url, request):
headers = {'content-type': 'application/json'}
content = ''.encode("utf-8")
return response(200, content, headers, None, 5, request)
obj = Group(self.gl, data={"name": "testname", "id": 1})
with HTTMock(resp_delete_group):
data = self.gl.delete(obj)
self.assertIs(data, True)
def test_delete_from_invalid_class(self):
class InvalidClass(object):
pass
self.assertRaises(GitlabError, self.gl.delete, InvalidClass, 1)
def test_delete_from_class(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/groups/1",
method="delete")
def resp_delete_group(url, request):
headers = {'content-type': 'application/json'}
content = ''.encode("utf-8")
return response(200, content, headers, None, 5, request)
with HTTMock(resp_delete_group):
data = self.gl.delete(Group, 1)
self.assertIs(data, True)
def test_delete_unknown_path(self):
obj = Project(self.gl, data={"name": "testname", "id": 1})
obj._from_api = True
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/projects/1",
method="delete")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(404, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabDeleteError, self.gl.delete, obj)
def test_delete_401(self):
obj = Project(self.gl, data={"name": "testname", "id": 1})
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/projects/1",
method="delete")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(401, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabAuthenticationError, self.gl.delete, obj)
def test_delete_unknown_error(self):
obj = Project(self.gl, data={"name": "testname", "id": 1})
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/projects/1",
method="delete")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(405, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabDeleteError, self.gl.delete, obj)
def test_create(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/projects",
method="post")
def resp_create_project(url, request):
headers = {'content-type': 'application/json'}
content = '{"name": "testname", "id": 1}'.encode("utf-8")
return response(201, content, headers, None, 5, request)
obj = Project(self.gl, data={"name": "testname"})
with HTTMock(resp_create_project):
data = self.gl.create(obj)
expected = {u"name": u"testname", u"id": 1}
self.assertEqual(expected, data)
def test_create_kw_missing(self):
obj = Group(self.gl, data={"name": "testgroup"})
self.assertRaises(GitlabCreateError, self.gl.create, obj)
def test_create_unknown_path(self):
obj = Project(self.gl, data={"name": "name"})
obj.id = 1
obj._from_api = True
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/projects/1",
method="delete")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(404, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabDeleteError, self.gl.delete, obj)
def test_create_401(self):
obj = Group(self.gl, data={"name": "testgroup", "path": "testpath"})
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/groups",
method="post")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(401, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabAuthenticationError, self.gl.create, obj)
def test_create_unknown_error(self):
obj = Group(self.gl, data={"name": "testgroup", "path": "testpath"})
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/groups",
method="post")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(405, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabCreateError, self.gl.create, obj)
def test_update(self):
obj = User(self.gl, data={"email": "testuser@testmail.com",
"password": "testpassword",
"name": u"testuser",
"username": "testusername",
"can_create_group": True,
"id": 1})
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/users/1",
method="put")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"first": "return1"}'.encode("utf-8")
return response(200, content, headers, None, 5, request)
with HTTMock(resp_cont):
data = self.gl.update(obj)
expected = {"first": "return1"}
self.assertEqual(expected, data)
def test_update_kw_missing(self):
obj = Hook(self.gl, data={"name": "testgroup"})
self.assertRaises(GitlabUpdateError, self.gl.update, obj)
def test_update_401(self):
obj = Group(self.gl, data={"name": "testgroup", "path": "testpath",
"id": 1})
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/groups/1",
method="put")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(401, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabAuthenticationError, self.gl.update, obj)
def test_update_unknown_error(self):
obj = Group(self.gl, data={"name": "testgroup", "path": "testpath",
"id": 1})
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/groups/1",
method="put")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(405, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabUpdateError, self.gl.update, obj)
def test_update_unknown_path(self):
obj = Group(self.gl, data={"name": "testgroup", "path": "testpath",
"id": 1})
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/groups/1",
method="put")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(404, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabUpdateError, self.gl.update, obj)
class TestGitlabAuth(unittest.TestCase):
def test_invalid_auth_args(self):
self.assertRaises(ValueError,
Gitlab,
"http://localhost", api_version='4',
private_token='private_token', oauth_token='bearer')
self.assertRaises(ValueError,
Gitlab,
"http://localhost", api_version='4',
oauth_token='bearer', http_username='foo',
http_password='bar')
self.assertRaises(ValueError,
Gitlab,
"http://localhost", api_version='4',
private_token='private_token', http_password='bar')
self.assertRaises(ValueError,
Gitlab,
"http://localhost", api_version='4',
private_token='private_token', http_username='foo')
def test_private_token_auth(self):
gl = Gitlab('http://localhost', private_token='private_token',
api_version='4')
self.assertEqual(gl.private_token, 'private_token')
self.assertEqual(gl.oauth_token, None)
self.assertEqual(gl._http_auth, None)
self.assertEqual(gl.headers['PRIVATE-TOKEN'], 'private_token')
self.assertNotIn('Authorization', gl.headers)
def test_oauth_token_auth(self):
gl = Gitlab('http://localhost', oauth_token='oauth_token',
api_version='4')
self.assertEqual(gl.private_token, None)
self.assertEqual(gl.oauth_token, 'oauth_token')
self.assertEqual(gl._http_auth, None)
self.assertEqual(gl.headers['Authorization'], 'Bearer oauth_token')
self.assertNotIn('PRIVATE-TOKEN', gl.headers)
def test_http_auth(self):
gl = Gitlab('http://localhost', private_token='private_token',
http_username='foo', http_password='bar', api_version='4')
self.assertEqual(gl.private_token, 'private_token')
self.assertEqual(gl.oauth_token, None)
self.assertIsInstance(gl._http_auth, requests.auth.HTTPBasicAuth)
self.assertEqual(gl.headers['PRIVATE-TOKEN'], 'private_token')
self.assertNotIn('Authorization', gl.headers)
class TestGitlab(unittest.TestCase):
def setUp(self):
self.gl = Gitlab("http://localhost", private_token="private_token",
email="testuser@test.com", password="testpassword",
ssl_verify=True)
def test_pickability(self):
original_gl_objects = self.gl._objects
pickled = pickle.dumps(self.gl)
unpickled = pickle.loads(pickled)
self.assertIsInstance(unpickled, Gitlab)
self.assertTrue(hasattr(unpickled, '_objects'))
self.assertEqual(unpickled._objects, original_gl_objects)
def test_credentials_auth_nopassword(self):
self.gl.email = None
self.gl.password = None
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/session",
method="post")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(404, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabAuthenticationError,
self.gl._credentials_auth)
def test_credentials_auth_notok(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/session",
method="post")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{"message": "message"}'.encode("utf-8")
return response(404, content, headers, None, 5, request)
with HTTMock(resp_cont):
self.assertRaises(GitlabAuthenticationError,
self.gl._credentials_auth)
def test_auth_with_credentials(self):
self.gl.private_token = None
self.test_credentials_auth(callback=self.gl.auth)
def test_auth_with_token(self):
self.test_token_auth(callback=self.gl.auth)
def test_credentials_auth(self, callback=None):
if callback is None:
callback = self.gl._credentials_auth
token = "credauthtoken"
id_ = 1
expected = {"PRIVATE-TOKEN": token}
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/session",
method="post")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{{"id": {0:d}, "private_token": "{1:s}"}}'.format(
id_, token).encode("utf-8")
return response(201, content, headers, None, 5, request)
with HTTMock(resp_cont):
callback()
self.assertEqual(self.gl.private_token, token)
self.assertDictContainsSubset(expected, self.gl.headers)
self.assertEqual(self.gl.user.id, id_)
def test_token_auth(self, callback=None):
if callback is None:
callback = self.gl._token_auth
name = "username"
id_ = 1
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/user",
method="get")
def resp_cont(url, request):
headers = {'content-type': 'application/json'}
content = '{{"id": {0:d}, "username": "{1:s}"}}'.format(
id_, name).encode("utf-8")
return response(200, content, headers, None, 5, request)
with HTTMock(resp_cont):
callback()
self.assertEqual(self.gl.user.username, name)
self.assertEqual(self.gl.user.id, id_)
self.assertEqual(type(self.gl.user), CurrentUser)
def test_hooks(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/hooks/1",
method="get")
def resp_get_hook(url, request):
headers = {'content-type': 'application/json'}
content = '{"url": "testurl", "id": 1}'.encode("utf-8")
return response(200, content, headers, None, 5, request)
with HTTMock(resp_get_hook):
data = self.gl.hooks.get(1)
self.assertEqual(type(data), Hook)
self.assertEqual(data.url, "testurl")
self.assertEqual(data.id, 1)
def test_projects(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/projects/1",
method="get")
def resp_get_project(url, request):
headers = {'content-type': 'application/json'}
content = '{"name": "name", "id": 1}'.encode("utf-8")
return response(200, content, headers, None, 5, request)
with HTTMock(resp_get_project):
data = self.gl.projects.get(1)
self.assertEqual(type(data), Project)
self.assertEqual(data.name, "name")
self.assertEqual(data.id, 1)
def test_userprojects(self):
@urlmatch(scheme="http", netloc="localhost",
path="/api/v3/projects/user/2", method="get")
def resp_get_userproject(url, request):
headers = {'content-type': 'application/json'}
content = '{"name": "name", "id": 1, "user_id": 2}'.encode("utf-8")
return response(200, content, headers, None, 5, request)
with HTTMock(resp_get_userproject):
self.assertRaises(NotImplementedError, self.gl.user_projects.get,
1, user_id=2)
def test_groups(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/groups/1",
method="get")
def resp_get_group(url, request):
headers = {'content-type': 'application/json'}
content = '{"name": "name", "id": 1, "path": "path"}'
content = content.encode('utf-8')
return response(200, content, headers, None, 5, request)
with HTTMock(resp_get_group):
data = self.gl.groups.get(1)
self.assertEqual(type(data), Group)
self.assertEqual(data.name, "name")
self.assertEqual(data.path, "path")
self.assertEqual(data.id, 1)
def test_issues(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/issues",
method="get")
def resp_get_issue(url, request):
headers = {'content-type': 'application/json'}
content = ('[{"name": "name", "id": 1}, '
'{"name": "other_name", "id": 2}]')
content = content.encode("utf-8")
return response(200, content, headers, None, 5, request)
with HTTMock(resp_get_issue):
data = self.gl.issues.get(2)
self.assertEqual(data.id, 2)
self.assertEqual(data.name, 'other_name')
def test_users(self):
@urlmatch(scheme="http", netloc="localhost", path="/api/v3/users/1",
method="get")
def resp_get_user(url, request):
headers = {'content-type': 'application/json'}
content = ('{"name": "name", "id": 1, "password": "password", '
'"username": "username", "email": "email"}')
content = content.encode("utf-8")
return response(200, content, headers, None, 5, request)
with HTTMock(resp_get_user):
user = self.gl.users.get(1)
self.assertEqual(type(user), User)
self.assertEqual(user.name, "name")
self.assertEqual(user.id, 1)
def test_teams(self):
@urlmatch(scheme="http", netloc="localhost",
path="/api/v3/user_teams/1", method="get")
def resp_get_group(url, request):
headers = {'content-type': 'application/json'}
content = '{"name": "name", "id": 1, "path": "path"}'
content = content.encode('utf-8')
return response(200, content, headers, None, 5, request)
with HTTMock(resp_get_group):
data = self.gl.teams.get(1)
self.assertEqual(type(data), Team)
self.assertEqual(data.name, "name")
self.assertEqual(data.path, "path")
self.assertEqual(data.id, 1)