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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
###############################################################################
# Copyright (c) 2017 Koren Lev (Cisco Systems), Yaron Yogev (Cisco Systems) #
# and others #
# #
# All rights reserved. This program and the accompanying materials #
# are made available under the terms of the Apache License, Version 2.0 #
# which accompanies this distribution, and is available at #
# http://www.apache.org/licenses/LICENSE-2.0 #
###############################################################################
from unittest.mock import MagicMock
from discover.fetchers.api.api_fetch_projects import ApiFetchProjects
from test.fetch.test_fetch import TestFetch
from test.fetch.api_fetch.test_data.api_fetch_projects import *
from test.fetch.api_fetch.test_data.regions import REGIONS
from test.fetch.api_fetch.test_data.token import TOKEN
class TestApiFetchProjects(TestFetch):
def setUp(self):
super().setUp()
self.configure_environment()
self._v2_auth_pwd = ApiFetchProjects.v2_auth_pwd
ApiFetchProjects.v2_auth_pwd = MagicMock(return_value=TOKEN)
self.fetcher = ApiFetchProjects()
self.set_regions_for_fetcher(self.fetcher)
self.region = REGIONS[REGION_NAME]
self._get_region_url_nover = self.fetcher.get_region_url_nover
self.fetcher.get_region_url_nover = \
MagicMock(return_value=REGION_URL_NOVER)
def test_get_for_region(self):
# mock request endpoint
self.fetcher.get_region_url_nover = MagicMock(return_value=REGION_URL_NOVER)
self.fetcher.get_url = MagicMock(return_value=REGION_RESPONSE)
result = self.fetcher.get_for_region(self.region, TOKEN)
self.assertEqual(result, REGION_RESULT, "Can't get correct projects info")
# TODO does this test case make sense?
def test_get_for_region_with_error_region_response(self):
self.fetcher.get_region_url_nover = MagicMock(return_value=REGION_URL_NOVER)
self.fetcher.get_url = MagicMock(return_value=REGION_ERROR_RESPONSE)
result = self.fetcher.get_for_region(self.region, TOKEN)
self.assertEqual(result, [], "Can't get [] when the " +
"region response is wrong")
def test_get_projects_for_api_user(self):
# mock the responses from OpenStack Api
self.fetcher.get_url = MagicMock(return_value=PROJECTS_CORRECT_RESPONSE)
result = self.fetcher.get_projects_for_api_user(self.region, TOKEN)
self.assertEqual(result, PROJECT_RESULT, "Can't get correct " +
"projects info for api user")
def test_get_projects_for_api_user_without_projects_response(self):
# the projects info from OpenStack Api will be None
self.fetcher.get_url = MagicMock(return_value=
PROJECTS_RESPONSE_WITHOUT_PROJECTS)
result = self.fetcher.get_projects_for_api_user(self.region, TOKEN)
self.assertIs(result, None, "Can't get None when the project " +
"response doesn't contain projects info")
def check_get_result(self, projects_for_api_user,
region_result,
token,
expected_result, error_msg):
self.fetcher.get_projects_for_api_user = MagicMock(return_value=
projects_for_api_user)
original_method = self.fetcher.get_for_region
# mock
self.fetcher.get_for_region = MagicMock(return_value=region_result)
self.fetcher.v2_auth_pwd = MagicMock(return_value=token)
result = self.fetcher.get(PROJECT_ID)
self.fetcher.get_for_region = original_method
self.assertEqual(result, expected_result, error_msg)
def test_get(self):
# test get method with different test cases
test_cases = [
{
"projects": PROJECT_RESULT,
"regions": REGION_RESULT,
"token": TOKEN,
"expected_result": REGION_RESULT,
"err_msg": "Can't get correct project result"
},
{
"projects": PROJECT_RESULT,
"regions": REGION_RESULT_WITH_NON_USER_PROJECT,
"token": TOKEN,
"expected_result": REGION_RESULT,
"err_msg": "Can't get correct project result" +
"when the region result contains project " +
"that doesn't belong to the user"
},
{
"projects": PROJECT_RESULT,
"regions": REGION_RESULT,
"token": None,
"expected_result": [],
"err_msg": "Can't get [] when the token is invalid"
},
{
"projects": None,
"regions": REGION_RESULT,
"token": TOKEN,
"expected_result": REGION_RESULT,
"err_msg": "Can't get the region " +
"result if the projects " +
"for the user doesn't exist"
}
]
for test_case in test_cases:
self.check_get_result(test_case["projects"],
test_case["regions"],
test_case["token"],
test_case["expected_result"],
test_case["err_msg"])
def tearDown(self):
super().tearDown()
ApiFetchProjects.v2_auth_pwd = self._v2_auth_pwd
self.reset_regions_for_fetcher(self.fetcher)
self.fetcher.get_region_url_nover = self._get_region_url_nover
|
