1
1
import { parse } from 'graphql' ;
2
- import ASTParser from '../../src/analysis/ASTParser ' ;
2
+ import QueryParser from '../../src/analysis/QueryParser ' ;
3
3
import { TypeWeightObject , Variables } from '../../src/@types/buildTypeWeights' ;
4
4
5
5
/**
@@ -122,7 +122,7 @@ let nonNullMockWeightFunction: jest.Mock<number, []>;
122
122
// this object is created by the schema above for use in all the tests below
123
123
let typeWeights : TypeWeightObject ;
124
124
125
- let queryParser : ASTParser ;
125
+ let queryParser : QueryParser ;
126
126
127
127
describe ( 'Test getQueryTypeComplexity function' , ( ) => {
128
128
beforeEach ( ( ) => {
@@ -281,7 +281,7 @@ describe('Test getQueryTypeComplexity function', () => {
281
281
282
282
describe ( 'Calculates the correct type complexity for queries' , ( ) => {
283
283
beforeEach ( ( ) => {
284
- queryParser = new ASTParser ( typeWeights , variables ) ;
284
+ queryParser = new QueryParser ( typeWeights , variables ) ;
285
285
} ) ;
286
286
test ( 'with one feild' , ( ) => {
287
287
query = `query { scalars { num } }` ;
@@ -325,7 +325,7 @@ describe('Test getQueryTypeComplexity function', () => {
325
325
expect ( queryParser . processQuery ( parse ( query ) ) ) . toBe ( 2 ) ; // Query 1 + human/character 1
326
326
// argument passed in as a variable
327
327
variables = { ep : 'EMPIRE' } ;
328
- queryParser = new ASTParser ( typeWeights , variables ) ;
328
+ queryParser = new QueryParser ( typeWeights , variables ) ;
329
329
query = `query variableQuery ($ep: Episode){ hero(episode: $ep) { id, name } }` ;
330
330
expect ( queryParser . processQuery ( parse ( query ) ) ) . toBe ( 2 ) ; // Query 1 + hero/character 1
331
331
} ) ;
@@ -370,7 +370,7 @@ describe('Test getQueryTypeComplexity function', () => {
370
370
}` ;
371
371
mockCharacterFriendsFunction . mockReturnValueOnce ( 3 ) ;
372
372
variables = { first : 3 } ;
373
- queryParser = new ASTParser ( typeWeights , variables ) ;
373
+ queryParser = new QueryParser ( typeWeights , variables ) ;
374
374
// Query 1 + 2*(character 1 + appearsIn/episode 0 + 3 * friends/character 1)
375
375
expect ( queryParser . processQuery ( parse ( query ) ) ) . toBe ( 9 ) ;
376
376
} ) ;
@@ -395,7 +395,7 @@ describe('Test getQueryTypeComplexity function', () => {
395
395
}` ;
396
396
mockCharacterFriendsFunction . mockReturnValueOnce ( 3 ) ;
397
397
variables = { first : 3 } ;
398
- queryParser = new ASTParser ( typeWeights , variables ) ;
398
+ queryParser = new QueryParser ( typeWeights , variables ) ;
399
399
// Query 1 + 2*(character 1 + appearsIn/episode 0 + 3 * friends/character 1)
400
400
expect ( queryParser . processQuery ( parse ( query ) ) ) . toBe ( 9 ) ;
401
401
} ) ;
@@ -421,7 +421,7 @@ describe('Test getQueryTypeComplexity function', () => {
421
421
mockCharacterFriendsFunction . mockReturnValueOnce ( 3 ) ;
422
422
423
423
variables = { first : 3 } ;
424
- queryParser = new ASTParser ( typeWeights , variables ) ;
424
+ queryParser = new QueryParser ( typeWeights , variables ) ;
425
425
// Query 1 + 2*(character 1 + appearsIn/episode 0 + 3 * friends/character 1)
426
426
expect ( queryParser . processQuery ( parse ( query ) ) ) . toBe ( 9 ) ;
427
427
@@ -441,7 +441,7 @@ describe('Test getQueryTypeComplexity function', () => {
441
441
}` ;
442
442
mockCharacterFriendsFunction . mockReturnValueOnce ( 3 ) ;
443
443
variables = { first : 3 } ;
444
- queryParser = new ASTParser ( typeWeights , variables ) ;
444
+ queryParser = new QueryParser ( typeWeights , variables ) ;
445
445
// Query 1 + 2*(character 1 + 0 selectionCost)
446
446
expect ( queryParser . processQuery ( parse ( query ) ) ) . toBe ( 3 ) ;
447
447
} ) ;
@@ -519,7 +519,7 @@ describe('Test getQueryTypeComplexity function', () => {
519
519
} ,
520
520
} ;
521
521
variables = { } ;
522
- queryParser = new ASTParser ( unionTypeWeights , variables ) ;
522
+ queryParser = new QueryParser ( unionTypeWeights , variables ) ;
523
523
} ) ;
524
524
test ( 'that have a complexity of zero' , ( ) => {
525
525
query = `
@@ -581,7 +581,7 @@ describe('Test getQueryTypeComplexity function', () => {
581
581
}` ;
582
582
mockCharacterFriendsFunction . mockReturnValueOnce ( 3 ) ;
583
583
variables = { first : 3 } ;
584
- queryParser = new ASTParser ( unionTypeWeights , variables ) ;
584
+ queryParser = new QueryParser ( unionTypeWeights , variables ) ;
585
585
// Query 1 + 1 hero + 3 friends/character
586
586
expect ( queryParser . processQuery ( parse ( query ) ) ) . toBe ( 5 ) ;
587
587
} ) ;
@@ -604,7 +604,7 @@ describe('Test getQueryTypeComplexity function', () => {
604
604
}` ;
605
605
mockDroidFriendsFunction . mockReturnValueOnce ( 3 ) ;
606
606
variables = { first : 3 } ;
607
- queryParser = new ASTParser ( unionTypeWeights , variables ) ;
607
+ queryParser = new QueryParser ( unionTypeWeights , variables ) ;
608
608
// Query 1 + 1 hero + max(Droid 3, Human 0) = 5
609
609
expect ( queryParser . processQuery ( parse ( query ) ) ) . toBe ( 5 ) ;
610
610
} ) ;
@@ -768,7 +768,7 @@ describe('Test getQueryTypeComplexity function', () => {
768
768
}` ;
769
769
mockDroidFriendsFunction . mockReturnValueOnce ( 3 ) ;
770
770
variables = { first : 3 } ;
771
- queryParser = new ASTParser ( typeWeights , variables ) ;
771
+ queryParser = new QueryParser ( typeWeights , variables ) ;
772
772
// Query 1 + 1 hero + max(Droid 3, Human 0) = 5
773
773
expect ( queryParser . processQuery ( parse ( query ) ) ) . toBe ( 5 ) ;
774
774
} ) ;
@@ -838,7 +838,7 @@ describe('Test getQueryTypeComplexity function', () => {
838
838
expect ( mockWeightFunction . mock . calls [ 0 ] . length ) . toBe ( 3 ) ; // calling with arguments and variables
839
839
840
840
variables = { first : 4 } ;
841
- queryParser = new ASTParser ( typeWeights , variables ) ;
841
+ queryParser = new QueryParser ( typeWeights , variables ) ;
842
842
mockWeightFunction . mockReturnValueOnce ( 4 ) ;
843
843
query = `query queryVariables($first: Int) {reviews(episode: EMPIRE, first: $first) { stars, commentary } }` ;
844
844
expect ( queryParser . processQuery ( parse ( query ) ) ) . toBe ( 5 ) ; // 1 Query + 4 reviews
@@ -989,7 +989,7 @@ describe('Test getQueryTypeComplexity function', () => {
989
989
} ) ;
990
990
test ( 'with arguments and varibales' , ( ) => {
991
991
variables = { directive : false } ;
992
- queryParser = new ASTParser ( typeWeights , variables ) ;
992
+ queryParser = new QueryParser ( typeWeights , variables ) ;
993
993
query = `query ($directive: Boolean!){
994
994
hero(episode: EMPIRE) {
995
995
id, name
@@ -1003,7 +1003,7 @@ describe('Test getQueryTypeComplexity function', () => {
1003
1003
// 1 query + 1 hero + 1 human
1004
1004
expect ( queryParser . processQuery ( parse ( query ) ) ) . toBe ( 3 ) ;
1005
1005
variables = { directive : true } ;
1006
- queryParser = new ASTParser ( typeWeights , variables ) ;
1006
+ queryParser = new QueryParser ( typeWeights , variables ) ;
1007
1007
query = `query ($directive: Boolean!){
1008
1008
hero(episode: EMPIRE) {
1009
1009
id, name
@@ -1099,7 +1099,7 @@ describe('Test getQueryTypeComplexity function', () => {
1099
1099
episode
1100
1100
}
1101
1101
}` ;
1102
- queryParser = new ASTParser ( typeWeights , variables ) ;
1102
+ queryParser = new QueryParser ( typeWeights , variables ) ;
1103
1103
expect ( queryParser . processQuery ( parse ( query ) ) ) . toBe ( 11 ) ; // Mutation 10 + review 1
1104
1104
} ) ;
1105
1105
@@ -1108,7 +1108,7 @@ describe('Test getQueryTypeComplexity function', () => {
1108
1108
query = `mutation createReviewMutation($review: ReviewInput!) {
1109
1109
createReview(episode: Empire, review: $review)
1110
1110
}` ;
1111
- queryParser = new ASTParser ( typeWeights , variables ) ;
1111
+ queryParser = new QueryParser ( typeWeights , variables ) ;
1112
1112
expect ( queryParser . processQuery ( parse ( query ) ) ) . toBe ( 11 ) ; // Mutation 10 + review 1
1113
1113
} ) ;
1114
1114
@@ -1127,14 +1127,14 @@ describe('Test getQueryTypeComplexity function', () => {
1127
1127
name
1128
1128
}
1129
1129
}` ;
1130
- queryParser = new ASTParser ( typeWeights , variables ) ;
1130
+ queryParser = new QueryParser ( typeWeights , variables ) ;
1131
1131
expect ( queryParser . processQuery ( parse ( query ) ) ) . toBe ( 13 ) ; // Mutation 10 + review 1 + query 1 + character 1
1132
1132
} ) ;
1133
1133
} ) ;
1134
1134
1135
1135
describe ( 'Calculates the depth of the query' , ( ) => {
1136
1136
beforeEach ( ( ) => {
1137
- queryParser = new ASTParser ( typeWeights , { } ) ;
1137
+ queryParser = new QueryParser ( typeWeights , { } ) ;
1138
1138
} ) ;
1139
1139
test ( 'with one feild' , ( ) => {
1140
1140
query = `query { scalars { num } }` ;
0 commit comments