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
|
import time
import requests
import json
import unittest
import re
import httpretty
class GenerateTests(type):
def __init__(cls, name, bases, attrs):
if name in ('GenerateTestMeta',): return
count = getattr(cls, '__generate_count__', attrs.get('__generate_count__'))
if not isinstance(count, int):
raise SyntaxError(f'Metaclass requires def `__generate_count__ = NUMBER_OF_TESTS` to be set to an integer')
generate_method = getattr(cls, '__generate_method__', attrs.get('__generate_method__'))
if not callable(generate_method):
raise SyntaxError(f'Metaclass requires def `__generate_method__(test_name):` to be implemented')
for x in range(count):
test_name = "test_{}".format(x)
def test_func(self, *args, **kwargs):
run_test = generate_method(test_name)
run_test(self, *args, **kwargs)
test_func.__name__ = test_name
attrs[test_name] = test_func
setattr(cls, test_name, test_func)
class TestBug426MypySegfaultWithCallbackAndPayload(unittest.TestCase, metaclass=GenerateTests):
__generate_count__ = 1000
def __generate_method__(test_name):
@httpretty.httprettified(allow_net_connect=False)
def test_func(self):
httpretty.register_uri(httpretty.GET, 'http://github.com', body=self.json_response_callback({"kind": "insecure"}))
httpretty.register_uri(httpretty.GET, 'https://github.com', body=self.json_response_callback({"kind": "secure"}))
httpretty.register_uri(httpretty.POST, re.compile('github.com/.*'), body=self.json_response_callback({"kind": "regex"}) )
response = requests.post(
'https://github.com/foo',
headers={
"Content-Type": "application/json"
},
data=json.dumps({test_name: time.time()}))
assert response.status_code == 200
try:
response = requests.get('https://gitlab.com')
assert response.status_code == 200
except Exception:
pass
return test_func
def json_response_callback(self, data):
payload = dict(data)
payload.update({
"time": time.time()
})
def request_callback(request, path, headers):
return [200, headers, json.dumps(payload)]
return request_callback
class TestBug426MypySegfaultWithEmptyMethod(unittest.TestCase, metaclass=GenerateTests):
__generate_count__ = 10000
def __generate_method__(test_name):
@httpretty.httprettified(allow_net_connect=False)
def test_func(self):
pass
return test_func
|
