fix(TestOperation): ensure operation-definition is fetched in runtime

Author: Frozen-byte

packages/apollo-angular/testing/src/operation.ts

@@ -1,5 +1,4 @@
 import { ExecutionResult, GraphQLError, Kind, OperationTypeNode } from 'graphql';
-import type { FragmentDefinitionNode, OperationDefinitionNode } from 'graphql/index';
 import { Observer } from 'rxjs';
 import { ApolloError, FetchResult, Operation as LinkOperation } from '@apollo/client/core';
 import { getMainDefinition } from '@apollo/client/utilities';
@@ -13,13 +12,10 @@ export type Operation = LinkOperation & {
 };
 
 export class TestOperation<T = { [key: string]: any }> {
-  private readonly definition: OperationDefinitionNode | FragmentDefinitionNode;
-
   constructor(
     public readonly operation: Operation,
     private readonly observer: Observer<FetchResult<T>>,
   ) {
-    this.definition = getMainDefinition(this.operation.query);
   }
 
   public flush(result: ExecutionResult<T> | ApolloError): void {
@@ -28,10 +24,12 @@ export class TestOperation<T = { [key: string]: any }> {
     } else {
       const fetchResult = result ? { ...result } : result;
       this.observer.next(fetchResult);
+      
+      const definition = getMainDefinition(this.operation.query);
 
       if (
-        this.definition.kind === Kind.OPERATION_DEFINITION &&
-        this.definition.operation !== OperationTypeNode.SUBSCRIPTION
+        definition.kind === Kind.OPERATION_DEFINITION &&
+        definition.operation !== OperationTypeNode.SUBSCRIPTION
       ) {
         this.complete();
       }