blob: 4cd474ccd0eb3d7326b9516c89d7bbf8e544d60d (
plain)
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
|
import os
import json
import pytest
import fatcat
import fatcat.sql
import unittest
import tempfile
# TODO: http://alextechrants.blogspot.com/2013/08/unit-testing-sqlalchemy-apps.html
## Helpers ##################################################################
def check_entity_fields(e):
for key in ('rev', 'is_live', 'redirect_id'):
assert key in e
for key in ('id',):
assert e[key] is not None
def check_release(e):
check_entity_fields(e)
for key in ('work', 'release_type'):
assert key in e
for key in ('title'):
assert e[key] is not None
def check_creator(e):
check_entity_fields(e)
for key in ('name'):
assert e[key] is not None
def check_container(e):
check_entity_fields(e)
for key in ('name'):
assert e[key] is not None
def check_file(e):
check_entity_fields(e)
for key in ('size', 'sha1'):
assert e[key] is not None
class FatcatTestCase(unittest.TestCase):
def setUp(self):
fatcat.app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite://'
fatcat.app.testing = True
self.app = fatcat.app.test_client()
fatcat.db.session.remove()
fatcat.db.drop_all()
fatcat.db.create_all()
fatcat.sql.populate_db()
## Model Tests ###############################################################
class ModelTestCase(FatcatTestCase):
def test_populate(self):
fatcat.sql.populate_db()
def test_example_works(self):
fatcat.dummy.insert_example_works()
def test_random_works(self):
fatcat.dummy.insert_random_works()
def test_load_crossref(self):
with open('./tests/files/crossref-works.2018-01-21.badsample.json', 'r') as f:
raw = [json.loads(l) for l in f.readlines() if len(l) > 3]
for obj in raw:
fatcat.sql.add_crossref(obj)
def test_hydrate_work(self):
fatcat.dummy.insert_random_works()
fatcat.sql.hydrate_work(1)
def test_hydrate_release(self):
fatcat.dummy.insert_random_works()
fatcat.sql.hydrate_release(1)
## API Tests #################################################################
class APITestCase(FatcatTestCase):
def test_health(self):
rv = self.app.get('/health')
obj = json.loads(rv.data.decode('utf-8'))
assert obj['ok']
def test_api_work(self):
fatcat.dummy.insert_example_works()
# Invalid Id
rv = self.app.get('/v0/work/_')
assert rv.status_code == 404
# Random
rv = self.app.get('/v0/work/random')
rv = self.app.get(rv.location)
work = json.loads(rv.data.decode('utf-8'))
check_entity_fields(work)
print(work)
assert work['title']
assert work['work_type']
# Valid Id (from random above)
rv = self.app.get('/v0/work/{}'.format(work['id']))
assert rv.status_code == 200
# Missing Id
rv = self.app.get('/v0/work/r3zga5b9cd7ef8gh084714iljk')
assert rv.status_code == 404
|
