tests: example rust contracts moved from warp-wasm-templates

crates/toy-contract/tests/contract.spec.ts

@@ -0,0 +1,116 @@
+import fs from 'fs';
+
+import ArLocal from 'arlocal';
+import Arweave from 'arweave';
+import { JWKInterface } from 'arweave/node/lib/wallet';
+import {
+  LoggerFactory,
+  Warp,
+  SmartWeaveTags,
+  WarpFactory,
+  TagsParser,
+  ArweaveWrapper,
+  WasmSrc,
+  Contract
+} from 'warp-contracts';
+import { DeployPlugin } from 'warp-contracts-plugin-deploy';
+import path from 'path';
+
+jest.setTimeout(30000);
+
+class State {
+  constructor(public x: number) {}
+}
+
+class View {
+  constructor(public x: number) {}
+}
+
+class Action {
+  constructor(public x: number) {}
+}
+
+describe('Testing the Rust WASM Profit Sharing Token', () => {
+  let contractSrc: Buffer;
+  let contractGlueCodeFile: string;
+
+  let wallet: JWKInterface;
+
+  const initialState = new State(0);
+
+  let arweave: Arweave;
+  let arlocal: ArLocal;
+  let warp: Warp;
+  let toyContract: Contract<State>;
+
+  let contractTxId: string;
+
+  let arweaveWrapper: ArweaveWrapper;
+  let tagsParser: TagsParser;
+
+  beforeAll(async () => {
+    arlocal = new ArLocal(1820, false);
+    await arlocal.start();
+
+    tagsParser = new TagsParser();
+    LoggerFactory.INST.logLevel('debug');
+    LoggerFactory.INST.logLevel('debug', 'WASM:Rust');
+    //LoggerFactory.INST.logLevel('debug', 'WasmContractHandlerApi');
+
+    warp = WarpFactory.forLocal(1820).use(new DeployPlugin());
+    ({ arweave } = warp);
+    arweaveWrapper = new ArweaveWrapper(arweave);
+
+    ({ jwk: wallet } = await warp.generateWallet());
+
+    contractSrc = fs.readFileSync(path.join(__dirname, '../pkg/rust-contract_bg.wasm'));
+    const contractSrcCodeDir: string = path.join(__dirname, '../src');
+    contractGlueCodeFile = path.join(__dirname, '../pkg/rust-contract.js');
+
+    // deploying contract using the new SDK.
+    ({ contractTxId } = await warp.deploy({
+      wallet,
+      initState: JSON.stringify(initialState),
+      src: contractSrc,
+      wasmSrcCodeDir: contractSrcCodeDir,
+      wasmGlueCode: contractGlueCodeFile
+    }));
+
+    toyContract = warp.contract<State>(contractTxId);
+
+    // connecting wallet to the PST contract
+    toyContract.connect(wallet);
+  });
+
+  afterAll(async () => {
+    await arlocal.stop();
+  });
+
+  it('should properly deploy contract', async () => {
+    const contractTx = await arweave.transactions.get(contractTxId);
+    expect(contractTx).not.toBeNull();
+
+    const contractSrcTxId = tagsParser.getTag(contractTx, SmartWeaveTags.CONTRACT_SRC_TX_ID);
+    const contractSrcTx = await arweave.transactions.get(contractSrcTxId);
+    expect(tagsParser.getTag(contractSrcTx, SmartWeaveTags.CONTENT_TYPE)).toEqual('application/wasm');
+    expect(tagsParser.getTag(contractSrcTx, SmartWeaveTags.WASM_LANG)).toEqual('rust');
+    expect(tagsParser.getTag(contractSrcTx, SmartWeaveTags.WASM_META)).toBeTruthy();
+
+    const srcTxData = await arweaveWrapper.txData(contractSrcTxId);
+    const wasmSrc = new WasmSrc(srcTxData);
+    expect(wasmSrc.wasmBinary()).not.toBeNull();
+    expect(wasmSrc.additionalCode()).toEqual(fs.readFileSync(contractGlueCodeFile, 'utf-8'));
+    expect((await wasmSrc.sourceCode()).size).toEqual(1);
+  });
+
+  it('should read state', async () => {
+    const { cachedValue } = await toyContract.readState();
+    expect(cachedValue.state).toEqual(initialState);
+  });
+
+  it('should write/view state', async () => {
+    await toyContract.writeInteraction<Action>(new Action(4));
+    const interactionResult = await toyContract.viewState<Action, View>(new Action(17));
+    expect(interactionResult.result.x).toEqual(4);
+  });
+});