def test_imports(self):
random_json = json.dumps(generate_dict_for_json(10))
random_json2 = json.dumps(generate_dict_for_json(100))
response = self.app.post('/imports',
content_type='application/json',
data=random_json)
assert response.get_json()['data']['import_id'] == 1
assert response.status_code == 201
response = self.app.post('/imports',
content_type='application/json',
data=random_json2)
assert response.get_json()['data']['import_id'] == 2
assert response.status_code == 201
def test_birthdays_10000(self):
random_json = json.dumps(generate_dict_for_json(10000))
import_id = self.app.post(
'/imports', content_type='application/json',
data=random_json).get_json()['data']['import_id']
response = self.app.get(f'/imports/{import_id}/citizens/birthdays')
assert response.status_code == 200
def test_import_10000(self):
random_json = json.dumps(
generate_dict_for_json(10000, relations_count=5000))
response = self.app.post('/imports',
content_type='application/json',
data=random_json)
assert response.status_code == 201
def test_stat_10000(self):
random_json = json.dumps(generate_dict_for_json(10000))
import_id = self.app.post(
'/imports', content_type='application/json',
data=random_json).get_json()['data']['import_id']
response = self.app.get(
f'/imports/{import_id}/towns/stat/percentile/age')
assert response.status_code == 200
def test_get_import(self):
random_json = json.dumps(generate_dict_for_json(20))
import_id = self.app.post(
'/imports', content_type='application/json',
data=random_json).get_json()['data']['import_id']
response = self.app.get(f'/imports/{import_id}/citizens')
assert sorted(response.get_json()['data'],
key=lambda x: x['citizen_id']) == sorted(
json.loads(random_json)['citizens'],
key=lambda x: x['citizen_id'])
def test_parallel(self):
random_json = json.dumps(generate_dict_for_json(10000))
random_json2 = json.dumps(
generate_dict_for_json(10000, error_line=9800))
from threading import Thread
def imports_threading(json_):
req = self.app.post('/imports',
content_type='application/json',
data=json_).get_json()
t1, t2 = Thread(target=imports_threading,
args=(random_json, )), Thread(target=imports_threading,
args=(random_json2, ))
t1.start()
t2.start()
t1.join()
t2.join()
assert Citizen.query.count() == 10000
def test_patch(self):
random_json = json.dumps(generate_dict_for_json(20))
import_id = self.app.post(
'/imports', content_type='application/json',
data=random_json).get_json()['data']['import_id']
citizen_id = randint(0, 19)
response = self.app.patch(
f'/imports/{import_id}/citizens/{citizen_id}',
content_type='application/json',
data=
'{"town":"Керчь","street":"Иосифа Бродского","relatives":[1,2,3]}')
citizen = Citizen.query.filter_by(citizen_id=citizen_id,
import_id=import_id).one()
assert response.status_code == 200
assert citizen_id in Citizen.query.filter_by(
citizen_id=2, import_id=import_id).one().relatives
assert citizen_id in Citizen.query.filter_by(
citizen_id=3, import_id=import_id).one().relatives
assert citizen.town == 'Керчь'
assert citizen.street == 'Иосифа Бродского'
def test_stat(self):
random_dict = generate_dict_for_json(10)
dates = [
'06.02.1984', '06.11.2012', '01.06.1972', '12.12.1996',
'18.04.1982', '15.02.1993', '01.09.2005', '27.10.2002',
'21.08.1984', '24.02.2016'
]
cities = ['Москва'] * 5 + ['Санкт-Петербург'] * 5
for i in range(10):
random_dict['citizens'][i].update({
'birth_date': dates[i],
'town': cities[i]
})
right = '{"data":[{"p50":35,"p75":37,"p99":46.6,"town":"\u041c\u043e\u0441\u043a\u0432\u0430"},{"p50":16,' \
'"p75":26,"p99":33.68,"town":"\u0421\u0430\u043d\u043a\u0442-\u041f\u0435\u0442\u0435\u0440\u0431' \
'\u0443\u0440\u0433"}]}'
import_id = \
self.app.post('/imports', content_type='application/json', data=json.dumps(random_dict)).get_json()['data'][
'import_id']
response = self.app.get(
f'/imports/{import_id}/towns/stat/percentile/age')
assert response.get_json() == json.loads(right)
def test_birthdays(self):
random_dict = generate_dict_for_json(10)
dates = [
'06.02.1984', '06.11.2012', '01.06.1972', '12.12.1996',
'18.04.1982', '15.02.1993', '01.09.2005', '27.10.2002',
'21.08.1984', '24.02.2016'
]
relatives = [[9, 5, 6, 8], [2, 4], [1], [8], [1, 9], [0, 8], [0], [],
[5, 0, 3], [0, 4]]
for i in range(10):
random_dict['citizens'][i].update({
'birth_date': dates[i],
'relatives': relatives[i]
})
import_id = \
self.app.post('/imports', content_type='application/json', data=json.dumps(random_dict)).get_json()['data'][
'import_id']
response = self.app.get(
f'/imports/{import_id}/citizens/birthdays').get_json()['data']
assert response['1'] == []
assert response['7'] == []
assert response['6'][0]['citizen_id'] == 1
assert response['12'][0]['presents'] == 1
def generate(count):
"""Генерация JSON для импорта"""
return jsonify(generate_dict_for_json(count))