-
Notifications
You must be signed in to change notification settings - Fork 0
/
test.py
68 lines (53 loc) · 2.13 KB
/
test.py
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
import os
import time
import requests
# setup
os.system('docker-compose down')
os.system('FLASK_CONFIG_TESTING=True docker-compose up --build -d')
time.sleep(60)
# tests
response = requests.post('http://localhost:8080/api/documents',
json={'document': 'hello'})
assert not response.ok, \
'FAILED TEST no auth token is error'
response = requests.post('http://localhost:8080/api/documents',
headers={'X-Access-Token': 'wrong-token'},
json={'document': 'hello'})
assert not response.ok, \
'FAILED TEST wrong auth token is error'
response = requests.post(
'http://localhost:8080/api/documents',
headers={'X-Access-Token': 'test-access-token'},
json={'document': 'hello'})
assert response.ok and response.json()['status'] == 'created', \
'FAILED TEST can create documents 1'
response = requests.post(
'http://localhost:8080/api/documents',
headers={'X-Access-Token': 'test-access-token'},
json={'document': 'hello world'})
assert response.ok and response.json()['status'] == 'created', \
'FAILED TEST can create documents 2'
response = requests.post(
'http://localhost:8080/api/documents',
headers={'X-Access-Token': 'test-access-token'},
json={'document': 'world'})
assert response.ok and response.json()['status'] == 'created', \
'FAILED TEST can create documents 3'
time.sleep(10)
response = requests.get(
'http://localhost:8080/api/documents/search/hello',
headers={'X-Access-Token': 'test-access-token'})
assert response.ok and len(response.json()['results']) == 2, \
'FAILED TEST can query documents 1'
response = requests.get(
'http://localhost:8080/api/documents/search/hello%20world',
headers={'X-Access-Token': 'test-access-token'})
assert response.ok and len(response.json()['results']) == 3, \
'FAILED TEST can query documents 2'
response = requests.get(
'http://localhost:8080/api/documents/search/unknown%20word',
headers={'X-Access-Token': 'test-access-token'})
assert response.ok and len(response.json()['results']) == 0, \
'FAILED TEST can query documents 3'
# teardown
os.system('docker-compose down')