@cosmology/ast/module/encoding/encode.test.js

Cosmos TypeScript AST generation

import { createAminoConverter } from './index';
import { ProtoStore, parseProto } from '@cosmology/proto-parser';
import { prepareContext, expectCode } from '../../test-utils';
import { camel } from '@cosmology/utils';
const store = new ProtoStore();
store.protos = [];
const addRef = ({
  filename,
  content
}) => {
  const ref = {
    absolute: filename,
    filename,
    proto: parseProto(content)
  };
  store.protos.push(ref);
};
addRef({
  filename: 'cosmology/example/tx.proto',
  content: `
syntax = "proto3";

package cosmology.finance;
option go_package = "github.com/cosmology-finance/go";

enum FancyEnumType {
    NO_HASH = 0;
    SHA256 = 1;
    SHA512 = 2;
    KECCAK = 3;
    RIPEMD160 = 4;
    BITCOIN = 5;
}
`
});
addRef({
  filename: 'cosmology/example/msg.proto',
  content: `
syntax = "proto3";
package cosmology.finance;
option go_package = "github.com/cosmology-finance/go";

import "cosmology/example/tx.proto";

message MsgDoFunThing {
    string                              address        = 1;
    cosmology.finance.FancyEnumType     myEnumField    = 2;
}

service Msg {
    rpc JoinPool(MsgDoFunThing) returns (MsgDoFunThingResponse);
}
message MsgDoFunThingResponse {}

`
});
store.traverseAll();
describe('cosmology/example/msg', () => {
  const {
    context,
    root,
    protos
  } = prepareContext(store, 'cosmology/example/msg.proto');
  it('AminoConverter', () => {
    context.options.aminoEncoding.casingFn = camel;
    expectCode(createAminoConverter({
      context,
      root,
      name: 'AminoConverter',
      protos
    }));
  });
});