Skip to content

Commit 0141ca3

Browse files
committed
Add a test for the examples so it's easier to test when they break
1 parent 797a4b3 commit 0141ca3

File tree

2 files changed

+30
-0
lines changed

2 files changed

+30
-0
lines changed

tests/test_examples/__init__.py

Whitespace-only changes.
Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
from examples import departments
2+
3+
4+
def test_departments():
5+
result = departments.main()
6+
assert not result.errors
7+
8+
deps = result.data["listDepartments"]
9+
assert len(deps) == 1
10+
11+
employees = deps[0]["employees"]
12+
assert len(employees) == 3
13+
14+
def employee_by_name(employees, name):
15+
return [e for e in employees if e["name"] == name][0]
16+
17+
jason = employee_by_name(employees, "Jason")
18+
carmen = employee_by_name(employees, "Carmen")
19+
derek = employee_by_name(employees, "Derek")
20+
21+
# Jason is a manager
22+
assert jason["teamSize"] == 2
23+
assert carmen.get("teamSize") is None
24+
25+
# some sanity checks on optional fields,
26+
# knowing what the test data is
27+
assert jason.get("hiredOn") is None
28+
assert carmen.get("hiredOn") is not None
29+
assert carmen["salary"]["rating"] == "GS-9"
30+
assert derek["salary"] is None

0 commit comments

Comments
 (0)