@graphprotocol/graph-cli 0.37.3 → 0.37.4-alpha-20230117095424-b5750b1

package/src/codegen/schema.test.ts

@@ -1,460 +0,0 @@
-import * as graphql from 'graphql/language';
-import immutable from 'immutable';
-import prettier from 'prettier';
-import Schema from '../schema';
-import SchemaCodeGenerator from './schema';
-import {
-  ArrayType,
-  Class,
-  Method,
-  NamedType,
-  NullableType,
-  Param,
-  StaticMethod,
-} from './typescript';
-
-const formatTS = (code: string) => prettier.format(code, { parser: 'typescript', semi: false });
-
-const createSchemaCodeGen = (schema: string) =>
-  new SchemaCodeGenerator(new Schema('', schema, immutable.fromJS(graphql.parse(schema))));
-
-const testEntity = (generatedTypes: any[], expectedEntity: any) => {
-  const entity = generatedTypes.find(type => type.name === expectedEntity.name);
-
-  expect(entity instanceof Class).toBe(true);
-  expect(entity.extends).toBe('Entity');
-  expect(entity.export).toBe(true);
-
-  const { members, methods } = entity;
-
-  expect(members).toStrictEqual(expectedEntity.members);
-
-  for (const expectedMethod of expectedEntity.methods) {
-    const method = methods.find((method: any) => method.name === expectedMethod.name);
-
-    expectedMethod.static
-      ? expect(method instanceof StaticMethod).toBe(true)
-      : expect(method instanceof Method).toBe(true);
-
-    expect(method.params).toStrictEqual(expectedMethod.params);
-    expect(method.returnType).toStrictEqual(expectedMethod.returnType);
-    expect(formatTS(method.body)).toBe(formatTS(expectedMethod.body));
-  }
-
-  expect(methods.length).toBe(expectedEntity.methods.length);
-};
-
-describe('Schema code generator', () => {
-  test('Should generate nothing for non entity types', () => {
-    const codegen = createSchemaCodeGen(`
-      type Foo {
-        foobar: Int
-      }
-
-      type Bar {
-        barfoo: Int
-      }
-    `);
-
-    expect(codegen.generateTypes().size).toBe(0);
-  });
-
-  describe('Should generate correct classes for each entity', () => {
-    const codegen = createSchemaCodeGen(`
-      # just to be sure nothing will be generated from non-entity types alongside regular ones
-      type Foo {
-        foobar: Int
-      }
-
-      type Account @entity {
-        id: ID!
-
-        # two non primitive types
-        description: String
-        name: String!
-
-        # two primitive types (i32)
-        age: Int
-        count: Int!
-
-        # derivedFrom
-        wallets: [Wallet!] @derivedFrom(field: "account")
-      }
-
-      type Wallet @entity {
-        id: ID!
-        amount: BigInt!
-        account: Account!
-      }
-    `);
-
-    const generatedTypes = codegen.generateTypes();
-
-    test('Foo is NOT an entity', () => {
-      const foo = generatedTypes.find((type: any) => type.name === 'Foo');
-      expect(foo).toBe(undefined);
-      // Account and Wallet
-      expect(generatedTypes.size).toBe(2);
-    });
-
-    test('Account is an entity with the correct methods', () => {
-      testEntity(generatedTypes, {
-        name: 'Account',
-        members: [],
-        methods: [
-          {
-            name: 'constructor',
-            params: [new Param('id', new NamedType('string'))],
-            returnType: undefined,
-            body: `
-              super()
-              this.set('id', Value.fromString(id))
-            `,
-          },
-          {
-            name: 'save',
-            params: [],
-            returnType: new NamedType('void'),
-            body: `
-              let id = this.get('id')
-              assert(id != null, 'Cannot save Account entity without an ID')
-              if (id) {
-                assert(
-                  id.kind == ValueKind.STRING,
-                  \`Entities of type Account must have an ID of type String but the id '\${id.displayData()}' is of type \${id.displayKind()}\`)
-                store.set('Account', id.toString(), this)
-              }
-            `,
-          },
-          {
-            name: 'load',
-            static: true,
-            params: [new Param('id', new NamedType('string'))],
-            returnType: new NullableType(new NamedType('Account')),
-            body: `
-              return changetype<Account | null>(store.get('Account', id))
-            `,
-          },
-          {
-            name: 'get id',
-            params: [],
-            returnType: new NamedType('string'),
-            body: `
-              let value = this.get('id')
-              return value!.toString()
-            `,
-          },
-          {
-            name: 'set id',
-            params: [new Param('value', new NamedType('string'))],
-            returnType: undefined,
-            body: `
-              this.set('id', Value.fromString(value))
-            `,
-          },
-          {
-            name: 'get description',
-            params: [],
-            returnType: new NullableType(new NamedType('string')),
-            body: `
-              let value = this.get('description')
-              if (!value || value.kind == ValueKind.NULL) {
-                return null
-              } else {
-                return value.toString()
-              }
-            `,
-          },
-          {
-            name: 'set description',
-            params: [new Param('value', new NullableType(new NamedType('string')))],
-            returnType: undefined,
-            body: `
-              if (!value) {
-                this.unset('description')
-              } else {
-                this.set('description', Value.fromString(<string>value))
-              }
-            `,
-          },
-          {
-            name: 'get name',
-            params: [],
-            returnType: new NamedType('string'),
-            body: `
-              let value = this.get('name')
-              return value!.toString()
-            `,
-          },
-          {
-            name: 'set name',
-            params: [new Param('value', new NamedType('string'))],
-            returnType: undefined,
-            body: `
-              this.set('name', Value.fromString(value))
-            `,
-          },
-          {
-            name: 'get age',
-            params: [],
-            returnType: new NamedType('i32'),
-            body: `
-              let value = this.get('age')
-              return value!.toI32()
-            `,
-          },
-          {
-            name: 'set age',
-            params: [new Param('value', new NamedType('i32'))],
-            returnType: undefined,
-            body: `
-              this.set('age', Value.fromI32(value))
-            `,
-          },
-          {
-            name: 'get count',
-            params: [],
-            returnType: new NamedType('i32'),
-            body: `
-              let value = this.get('count')
-              return value!.toI32()
-            `,
-          },
-          {
-            name: 'set count',
-            params: [new Param('value', new NamedType('i32'))],
-            returnType: undefined,
-            body: `
-              this.set('count', Value.fromI32(value))
-            `,
-          },
-          {
-            name: 'get wallets',
-            params: [],
-            returnType: new NullableType(new ArrayType(new NamedType('string'))),
-            body: `
-              let value = this.get('wallets')
-              if (!value || value.kind == ValueKind.NULL) {
-                return null
-              } else {
-                return value.toStringArray()
-              }
-            `,
-          },
-          {
-            name: 'set wallets',
-            params: [new Param('value', new NullableType(new ArrayType(new NamedType('string'))))],
-            returnType: undefined,
-            body: `
-              if (!value) {
-                this.unset('wallets')
-              } else {
-                this.set('wallets', Value.fromStringArray(<Array<string>>value))
-              }
-            `,
-          },
-        ],
-      });
-    });
-
-    test('Wallet is an entity with the correct methods', () => {
-      testEntity(generatedTypes, {
-        name: 'Wallet',
-        members: [],
-        methods: [
-          {
-            name: 'constructor',
-            params: [new Param('id', new NamedType('string'))],
-            returnType: undefined,
-            body: `
-              super()
-              this.set('id', Value.fromString(id))
-            `,
-          },
-          {
-            name: 'save',
-            params: [],
-            returnType: new NamedType('void'),
-            body: `
-              let id = this.get('id')
-              assert(id != null, 'Cannot save Wallet entity without an ID')
-              if (id) {
-                assert(
-                  id.kind == ValueKind.STRING,
-                  \`Entities of type Wallet must have an ID of type String but the id '\${id.displayData()}' is of type \${id.displayKind()}\`)
-                store.set('Wallet', id.toString(), this)
-              }
-            `,
-          },
-          {
-            name: 'load',
-            static: true,
-            params: [new Param('id', new NamedType('string'))],
-            returnType: new NullableType(new NamedType('Wallet')),
-            body: `
-              return changetype<Wallet | null>(store.get('Wallet', id))
-            `,
-          },
-          {
-            name: 'get id',
-            params: [],
-            returnType: new NamedType('string'),
-            body: `
-              let value = this.get('id')
-              return value!.toString()
-            `,
-          },
-          {
-            name: 'set id',
-            params: [new Param('value', new NamedType('string'))],
-            returnType: undefined,
-            body: `
-              this.set('id', Value.fromString(value))
-            `,
-          },
-          {
-            name: 'get amount',
-            params: [],
-            returnType: new NamedType('BigInt'),
-            body: `
-              let value = this.get('amount')
-              return value!.toBigInt()
-            `,
-          },
-          {
-            name: 'set amount',
-            params: [new Param('value', new NamedType('BigInt'))],
-            returnType: undefined,
-            body: `
-              this.set('amount', Value.fromBigInt(value))
-            `,
-          },
-          {
-            name: 'get account',
-            params: [],
-            returnType: new NamedType('string'),
-            body: `
-              let value = this.get('account')
-              return value!.toString()
-            `,
-          },
-          {
-            name: 'set account',
-            params: [new Param('value', new NamedType('string'))],
-            returnType: undefined,
-            body: `
-              this.set('account', Value.fromString(value))
-            `,
-          },
-        ],
-      });
-    });
-  });
-
-  test('Should handle references with Bytes id types', () => {
-    const codegen = createSchemaCodeGen(`
-    interface Employee {
-      id: Bytes!
-      name: String!
-     }
-
-    type Worker implements Employee @entity {
-      id: Bytes!
-      name: String!
-   }
-
-    type Task @entity {
-      id: Bytes!
-      employee: Employee!
-      worker: Worker!
-   }
-`);
-
-    const generatedTypes = codegen.generateTypes();
-    testEntity(generatedTypes, {
-      name: 'Task',
-      members: [],
-      methods: [
-        {
-          name: 'constructor',
-          params: [new Param('id', new NamedType('Bytes'))],
-          returnType: undefined,
-          body: "\n      super()\n      this.set('id', Value.fromBytes(id))\n      ",
-        },
-        {
-          name: 'save',
-          params: [],
-          returnType: new NamedType('void'),
-          body:
-            '\n' +
-            "        let id = this.get('id')\n" +
-            '        assert(id != null,\n' +
-            "               'Cannot save Task entity without an ID')\n" +
-            '        if (id) {\n' +
-            '          assert(id.kind == ValueKind.BYTES,\n' +
-            "                 `Entities of type Task must have an ID of type Bytes but the id '${id.displayData()}' is of type ${id.displayKind()}`)\n" +
-            "          store.set('Task', id.toBytes().toHexString(), this)\n" +
-            '        }',
-        },
-        {
-          name: 'load',
-          static: true,
-          params: [new Param('id', new NamedType('Bytes'))],
-          returnType: new NullableType(new NamedType('Task')),
-          body:
-            '\n' +
-            "        return changetype<Task | null>(store.get('Task', id.toHexString()))\n" +
-            '        ',
-        },
-        {
-          name: 'get id',
-          params: [],
-          returnType: new NamedType('Bytes'),
-          body:
-            '\n' +
-            "       let value = this.get('id')\n" +
-            '       return value!.toBytes()\n' +
-            '      ',
-        },
-        {
-          name: 'set id',
-          params: [new Param('value', new NamedType('Bytes'))],
-          returnType: undefined,
-          body: "\n      this.set('id', Value.fromBytes(value))\n    ",
-        },
-        {
-          name: 'get employee',
-          params: [],
-          returnType: new NamedType('Bytes'),
-          body:
-            '\n' +
-            "       let value = this.get('employee')\n" +
-            '       return value!.toBytes()\n' +
-            '      ',
-        },
-        {
-          name: 'set employee',
-          params: [new Param('value', new NamedType('Bytes'))],
-          returnType: undefined,
-          body: "\n      this.set('employee', Value.fromBytes(value))\n    ",
-        },
-        {
-          name: 'get worker',
-          params: [],
-          returnType: new NamedType('Bytes'),
-          body:
-            '\n' +
-            "       let value = this.get('worker')\n" +
-            '       return value!.toBytes()\n' +
-            '      ',
-        },
-        {
-          name: 'set worker',
-          params: [new Param('value', new NamedType('Bytes'))],
-          returnType: undefined,
-          body: "\n      this.set('worker', Value.fromBytes(value))\n    ",
-        },
-      ],
-    });
-  });
-});