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
|
##############################################################################
# Copyright (c) 2017 ZTE Corp
# feng.xiaowei@zte.com.cn
# 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
##############################################################################
import functools
import httplib
from concurrent.futures import ThreadPoolExecutor
import mock
O_get_secure_cookie = (
'opnfv_testapi.handlers.base_handlers.GenericApiHandler.get_secure_cookie')
def thread_execute(method, *args, **kwargs):
with ThreadPoolExecutor(max_workers=2) as executor:
result = executor.submit(method, *args, **kwargs)
return result
def mock_invalid_lfid():
def _mock_invalid_lfid(xstep):
def wrap(self, *args, **kwargs):
with mock.patch(O_get_secure_cookie) as m_cookie:
m_cookie.return_value = 'InvalidUser'
return xstep(self, *args, **kwargs)
return wrap
return _mock_invalid_lfid
def mock_valid_lfid():
def _mock_valid_lfid(xstep):
def wrap(self, *args, **kwargs):
with mock.patch(O_get_secure_cookie) as m_cookie:
m_cookie.return_value = 'ValidUser'
return xstep(self, *args, **kwargs)
return wrap
return _mock_valid_lfid
def create(excepted_status, excepted_response):
def _create(create_request):
@functools.wraps(create_request)
def wrap(self):
request = create_request(self)
status, body = self.create(request)
if excepted_status == httplib.OK:
getattr(self, excepted_response)(body)
else:
self.assertIn(excepted_response, body)
return wrap
return _create
def get(excepted_status, excepted_response):
def _get(get_request):
@functools.wraps(get_request)
def wrap(self):
request = get_request(self)
status, body = self.get(request)
if excepted_status == httplib.OK:
getattr(self, excepted_response)(body)
else:
self.assertIn(excepted_response, body)
return wrap
return _get
def update(excepted_status, excepted_response):
def _update(update_request):
@functools.wraps(update_request)
def wrap(self):
request, resource = update_request(self)
status, body = self.update(request, resource)
if excepted_status == httplib.OK:
getattr(self, excepted_response)(request, body)
else:
self.assertIn(excepted_response, body)
return wrap
return _update
def delete(excepted_status, excepted_response):
def _delete(delete_request):
@functools.wraps(delete_request)
def wrap(self):
request = delete_request(self)
if isinstance(request, tuple):
status, body = self.delete(request[0], *(request[1]))
else:
status, body = self.delete(request)
if excepted_status == httplib.OK:
getattr(self, excepted_response)(body)
else:
self.assertIn(excepted_response, body)
return wrap
return _delete
def query(excepted_status, excepted_response, number=0):
def _query(get_request):
@functools.wraps(get_request)
def wrap(self):
request = get_request(self)
status, body = self.query(request)
if excepted_status == httplib.OK:
getattr(self, excepted_response)(body, number)
else:
self.assertIn(excepted_response, body)
return wrap
return _query
|
