generic assertion for TRANSFER_FAILED reverts; _invalidStake generator

contracts/integrations/test/framework/assertions/generic_assertions.ts

@@ -0,0 +1,24 @@
+import { BaseContract } from '@0x/base-contract';
+import { expect } from '@0x/contracts-test-utils';
+import { RevertReason } from '@0x/types';
+import { StringRevertError } from '@0x/utils';
+
+import { FunctionAssertion, FunctionResult } from './function_assertion';
+
+/**
+ * Returns a generic FunctionAssertion for the given contract function, asserting that the
+ * function call reverts in an asset proxy contract with TRANSFER_FAILED.
+ */
+export function assetProxyTransferFailedAssertion<TArgs extends any[]>(
+    contract: BaseContract,
+    functionName: string,
+): FunctionAssertion<TArgs, void, void> {
+    return new FunctionAssertion(contract, functionName, {
+        after: async (_beforeInfo: void, result: FunctionResult) => {
+            // Ensure that the tx reverted.
+            expect(result.success).to.be.false();
+            // Check revert error
+            expect(result.data).to.equal(new StringRevertError(RevertReason.TransferFailed));
+        },
+    });
+}