@@ -3,6 +3,8 @@ import { describe, it } from 'mocha';
33
44import dedent from '../../__testUtils__/dedent' ;
55
6+ import mapValue from '../../jsutils/mapValue' ;
7+
68import { parse } from '../../language/parser' ;
79import { print } from '../../language/printer' ;
810
@@ -47,67 +49,64 @@ describe('separateOperations', () => {
4749 }
4850 ` ) ;
4951
50- const separatedASTs = separateOperations ( ast ) ;
51-
52- expect ( separatedASTs ) . to . have . all . keys ( '' , 'One' , 'Two' ) ;
53-
54- expect ( print ( separatedASTs [ '' ] ) ) . to . equal ( dedent `
55- {
56- ...Y
57- ...X
58- }
59-
60- fragment X on T {
61- fieldX
62- }
63-
64- fragment Y on T {
65- fieldY
66- }
67- ` ) ;
68-
69- expect ( print ( separatedASTs . One ) ) . to . equal ( dedent `
70- query One {
71- foo
72- bar
73- ...A
74- ...X
75- }
76-
77- fragment A on T {
78- field
79- ...B
80- }
81-
82- fragment X on T {
83- fieldX
84- }
85-
86- fragment B on T {
87- something
88- }
89- ` ) ;
90-
91- expect ( print ( separatedASTs . Two ) ) . to . equal ( dedent `
92- fragment A on T {
93- field
94- ...B
95- }
96-
97- query Two {
98- ...A
99- ...Y
100- baz
101- }
102-
103- fragment Y on T {
104- fieldY
105- }
106-
107- fragment B on T {
108- something
109- }
110- ` ) ;
52+ const separatedASTs = mapValue ( separateOperations ( ast ) , print ) ;
53+ expect ( separatedASTs ) . to . deep . equal ( {
54+ '' : dedent `
55+ {
56+ ...Y
57+ ...X
58+ }
59+
60+ fragment X on T {
61+ fieldX
62+ }
63+
64+ fragment Y on T {
65+ fieldY
66+ }
67+ ` ,
68+ One : dedent `
69+ query One {
70+ foo
71+ bar
72+ ...A
73+ ...X
74+ }
75+
76+ fragment A on T {
77+ field
78+ ...B
79+ }
80+
81+ fragment X on T {
82+ fieldX
83+ }
84+
85+ fragment B on T {
86+ something
87+ }
88+ ` ,
89+ Two : dedent `
90+ fragment A on T {
91+ field
92+ ...B
93+ }
94+
95+ query Two {
96+ ...A
97+ ...Y
98+ baz
99+ }
100+
101+ fragment Y on T {
102+ fieldY
103+ }
104+
105+ fragment B on T {
106+ something
107+ }
108+ ` ,
109+ } ) ;
111110 } ) ;
112111
113112 it ( 'survives circular dependencies' , ( ) => {
@@ -129,36 +128,34 @@ describe('separateOperations', () => {
129128 }
130129 ` ) ;
131130
132- const separatedASTs = separateOperations ( ast ) ;
133-
134- expect ( separatedASTs ) . to . have . all . keys ( 'One' , 'Two' ) ;
135-
136- expect ( print ( separatedASTs . One ) ) . to . equal ( dedent `
137- query One {
138- ...A
139- }
140-
141- fragment A on T {
142- ...B
143- }
144-
145- fragment B on T {
146- ...A
147- }
148- ` ) ;
149-
150- expect ( print ( separatedASTs . Two ) ) . to . equal ( dedent `
151- fragment A on T {
152- ...B
153- }
154-
155- fragment B on T {
156- ...A
157- }
158-
159- query Two {
160- ...B
161- }
162- ` ) ;
131+ const separatedASTs = mapValue ( separateOperations ( ast ) , print ) ;
132+ expect ( separatedASTs ) . to . deep . equal ( {
133+ One : dedent `
134+ query One {
135+ ...A
136+ }
137+
138+ fragment A on T {
139+ ...B
140+ }
141+
142+ fragment B on T {
143+ ...A
144+ }
145+ ` ,
146+ Two : dedent `
147+ fragment A on T {
148+ ...B
149+ }
150+
151+ fragment B on T {
152+ ...A
153+ }
154+
155+ query Two {
156+ ...B
157+ }
158+ ` ,
159+ } ) ;
163160 } ) ;
164161} ) ;
0 commit comments