1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
import urllib
from mock import mock
from testapiclient import main
from testapiclient.tests.unit import utils
from testapiclient.tests.unit import fakes
class MainTest(utils.TestCommand):
def setUp(self):
super(MainTest, self).setUp()
self.app = main.TestAPIClient()
self.cas_sever = '{}{}{}'.format(
self.env_variables['testapi_cas_auth_url'],
urllib.quote(self.env_variables['testapi_url']),
self.env_variables['testapi_cas_signin_return'])
self.headers = {
'Content-type': 'application/json',
'Accept': 'text/plain'}
def test_auth_success(self):
self.post_mock.return_value = fakes.FakeResponse(
data={'text': "success"})
self.app.run(
['-u', 'user', '-p', 'pass', 'pod', 'create',
'{"name": "asfad"}'])
self.post_mock.assert_called_with(
'http://localhost:8000/api/v1/pods',
data='{"name": "asfad"}',
headers=self.headers)
def test_auth_failure(self):
self.post_mock.return_value = fakes.FakeResponse(
data={'text': "login"})
self.app.run(
['-u', 'user', '-p', 'pass', 'pod', 'create',
'{"name": "asfad"}'])
self.post_mock.assert_called_once_with(
self.cas_sever,
{'pass': 'pass', 'name': 'user', 'form_id': 'user_login'}
)
def test_auth_not_called(self):
self.auth_post = mock.patch(
'testapiclient.utils.clientmanager.ClientManager.auth').start()
self.app.run(['pod', 'get'])
self.auth_post.assert_not_called()
self.get_mock.assert_called_once_with(
'http://localhost:8000/api/v1/pods', headers=self.headers)
|
