1 | import json |
---|
2 | from datetime import datetime, timedelta, timezone |
---|
3 | |
---|
4 | import pytest |
---|
5 | |
---|
6 | from ow.models.workout import Workout |
---|
7 | from ow.models.user import User |
---|
8 | from ow.models.root import OpenWorkouts |
---|
9 | |
---|
10 | |
---|
11 | class TestRootOpenWorkouts(object): |
---|
12 | |
---|
13 | @pytest.fixture |
---|
14 | def root(self): |
---|
15 | root = OpenWorkouts() |
---|
16 | root['john'] = User(firstname='John', lastname='Doe', |
---|
17 | email='john.doe@example.net') |
---|
18 | root['john'].password = 's3cr3t' |
---|
19 | workout = Workout( |
---|
20 | start=datetime(2015, 6, 28, 12, 55, tzinfo=timezone.utc), |
---|
21 | duration=timedelta(minutes=60), |
---|
22 | distance=30, sport='cycling' |
---|
23 | ) |
---|
24 | root['john'].add_workout(workout) |
---|
25 | return root |
---|
26 | |
---|
27 | def test_all_usernames(self, root): |
---|
28 | # the method returns an OOBtree object |
---|
29 | assert [u for u in root.all_usernames()] == ['john'] |
---|
30 | |
---|
31 | def test_lowercase_usernames(self, root): |
---|
32 | root.add_user(uid='Jack', firstname='Jack', lastname='Dumb', |
---|
33 | email='jack.dumb@example.net') |
---|
34 | assert root.lowercase_usernames() == ['jack', 'john'] |
---|
35 | |
---|
36 | def test_emails(self, root): |
---|
37 | assert root.emails() == ['john.doe@example.net'] |
---|
38 | |
---|
39 | def test_lowercase_emails(self, root): |
---|
40 | root.add_user(uid='Jack', firstname='Jack', lastname='Dumb', |
---|
41 | email='Jack.Dumb@example.net') |
---|
42 | assert root.lowercase_emails() == ['jack.dumb@example.net', |
---|
43 | 'john.doe@example.net'] |
---|
44 | |
---|
45 | def test_users(self, root): |
---|
46 | # the method returns an OOBtree object |
---|
47 | assert [u for u in root.users()] == [root['john']] |
---|
48 | |
---|
49 | def test_get_user(self, root): |
---|
50 | assert root.get_user('john') == root['john'] |
---|
51 | assert root.get_user('jack') is None |
---|
52 | with pytest.raises(TypeError): |
---|
53 | root.get_user() |
---|
54 | |
---|
55 | def test_add_user(self, root): |
---|
56 | assert len(root.users()) == 1 |
---|
57 | root.add_user(uid='jack', firstname='Jack', lastname='Dumb', |
---|
58 | email='jack.dumb@example.net') |
---|
59 | assert len(root.users()) == 2 |
---|
60 | assert 'jack' in root.all_usernames() |
---|
61 | |
---|
62 | def test_sports(self, root): |
---|
63 | assert root.sports == ['cycling'] |
---|
64 | workout = Workout( |
---|
65 | start=datetime(2015, 6, 29, 12, 55, tzinfo=timezone.utc), |
---|
66 | duration=timedelta(minutes=60), |
---|
67 | distance=10, sport='running') |
---|
68 | root['john'].add_workout(workout) |
---|
69 | assert root.sports == ['cycling', 'running'] |
---|
70 | |
---|
71 | def test_sports_json(self, root): |
---|
72 | assert root.sports_json == json.dumps(["cycling"]) |
---|
73 | workout = Workout( |
---|
74 | start=datetime(2015, 6, 29, 12, 55, tzinfo=timezone.utc), |
---|
75 | duration=timedelta(minutes=60), |
---|
76 | distance=10, sport='running') |
---|
77 | root['john'].add_workout(workout) |
---|
78 | assert root.sports_json == json.dumps(["cycling", "running"]) |
---|