Add getComplexity test with variables #16

slicknode · May 18, 2019 · fdc2b1a · fdc2b1a
1 parent 1e7a5ee
commit fdc2b1a
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/src/__tests__/QueryComplexity-test.ts b/src/__tests__/QueryComplexity-test.ts
@@ -40,6 +40,27 @@ describe('QueryComplexity analysis', () => {
     expect(complexity).to.equal(1);
   });
 
+  it('should calculate complexity with variables', () => {
+    const ast = parse(`
+      query Q($count: Int) {
+        variableScalar(count: $count)
+      }
+    `);
+
+    const complexity = getComplexity({
+      estimators: [
+        fieldConfigEstimator(),
+        simpleEstimator({defaultComplexity: 1})
+      ],
+      schema,
+      query: ast,
+      variables: {
+        count: 5,
+      },
+    });
+    expect(complexity).to.equal(50);
+  });
+
   it('should not allow negative cost', () => {
     const ast = parse(`
       query {