|
6 | 6 | def test_article_field_args(): |
7 | 7 | field = MongoengineConnectionField(nodes.ArticleNode) |
8 | 8 |
|
9 | | - field_args = ['id', 'headline', 'pub_date'] |
10 | | - assert set(field.field_args.keys()) == set(field_args) |
| 9 | + field_args = {'id', 'headline', 'pub_date'} |
| 10 | + assert set(field.field_args.keys()) == field_args |
11 | 11 |
|
12 | | - reference_args = ['editor', 'reporter'] |
13 | | - assert set(field.reference_args.keys()) == set(reference_args) |
| 12 | + reference_args = {'editor', 'reporter'} |
| 13 | + assert set(field.reference_args.keys()) == reference_args |
14 | 14 |
|
15 | | - default_args = ['after', 'last', 'first', 'before'] |
16 | | - args = field_args + reference_args + default_args |
17 | | - assert set(field.args) == set(args) |
| 15 | + default_args = {'after', 'last', 'first', 'before'} |
| 16 | + args = field_args | reference_args | default_args |
| 17 | + assert set(field.args) == args |
18 | 18 |
|
19 | 19 |
|
20 | 20 | def test_reporter_field_args(): |
21 | 21 | field = MongoengineConnectionField(nodes.ReporterNode) |
22 | 22 |
|
23 | | - field_args = ['id', 'first_name', 'last_name', 'email', 'awards'] |
24 | | - assert set(field.field_args.keys()) == set(field_args) |
| 23 | + field_args = {'id', 'first_name', 'last_name', 'email', 'awards'} |
| 24 | + assert set(field.field_args.keys()) == field_args |
| 25 | + |
| 26 | + |
| 27 | +def test_editor_field_args(): |
| 28 | + field = MongoengineConnectionField(nodes.EditorNode) |
| 29 | + |
| 30 | + field_args = {'id', 'first_name', 'last_name', 'metadata'} |
| 31 | + assert set(field.field_args.keys()) == field_args |
25 | 32 |
|
26 | 33 |
|
27 | 34 | def test_field_args_with_property(): |
|
0 commit comments