|
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