Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
46 changes: 46 additions & 0 deletions typescript/swap-vm/src/swap-vm/maker-traits.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -176,4 +176,50 @@ describe('MakerTraits', () => {
encodeDecodeTest(traits, maker)
})
})

it('should encode hooks where target is maker', () => {
const maker = new Address('0x742d35cc6634c0532925a3b844bc454e4438f44e')
const traits = MakerTraits.new({
shouldUnwrap: false,
useAquaInsteadOfSignature: true,
allowZeroAmountIn: false,
preTransferInHook: new Interaction(maker, new HexString('0xdeadbeef01')),
postTransferInHook: new Interaction(maker, new HexString('0xdeadbeef02')),
preTransferOutHook: new Interaction(maker, new HexString('0xdeadbeef03')),
postTransferOutHook: new Interaction(maker, new HexString('0xdeadbeef04')),
})

const encoded = traits.encode(maker)
const decoded = MakerTraits.decode(encoded.traits, encoded.hooksData)

expect(decoded.postTransferInHook?.data).toEqual(traits.postTransferInHook?.data)
expect(decoded.postTransferInHook?.data).toEqual(traits.postTransferInHook?.data)
expect(decoded.preTransferOutHook?.data).toEqual(traits.preTransferOutHook?.data)
expect(decoded.postTransferOutHook?.data).toEqual(traits.postTransferOutHook?.data)

// target is zero after decoding because target is same as maker
expect(decoded.postTransferInHook?.target).toEqual(Address.ZERO_ADDRESS)
expect(decoded.postTransferInHook?.target).toEqual(Address.ZERO_ADDRESS)
expect(decoded.preTransferOutHook?.target).toEqual(Address.ZERO_ADDRESS)
expect(decoded.postTransferOutHook?.target).toEqual(Address.ZERO_ADDRESS)
})

it('should encode hooks where target is not maker', () => {
const maker = new Address('0x742d35cc6634c0532925a3b844bc454e4438f44e')
const target = Address.fromBigInt(1n)
const traits = MakerTraits.new({
shouldUnwrap: false,
useAquaInsteadOfSignature: true,
allowZeroAmountIn: false,
preTransferInHook: new Interaction(target, new HexString('0xdeadbeef01')),
postTransferInHook: new Interaction(target, new HexString('0xdeadbeef02')),
preTransferOutHook: new Interaction(target, new HexString('0xdeadbeef03')),
postTransferOutHook: new Interaction(target, new HexString('0xdeadbeef04')),
})

const encoded = traits.encode(maker)
const decoded = MakerTraits.decode(encoded.traits, encoded.hooksData)

expect(decoded).toEqual(traits)
})
})
4 changes: 2 additions & 2 deletions typescript/swap-vm/src/swap-vm/maker-traits.ts
Original file line number Diff line number Diff line change
Expand Up @@ -345,7 +345,7 @@ export class MakerTraits {
const { data, offsets } = MakerTraits.HOOKS.reduce(
(acc, hookName, i) => {
const hook = this[hookName]
const hasTarget = this.hasTargetForHook(hookName)
const hasTarget = this.hasTargetForHook(hookName, maker)
const encoded = hook && hasTarget ? hook.encode() : hook?.data || HexString.EMPTY

acc.sum += BigInt(encoded.bytesCount())
Expand All @@ -367,7 +367,7 @@ export class MakerTraits {

private hasTargetForHook(
hookName: (typeof MakerTraits.HOOKS)[number],
maker: Address = Address.ZERO_ADDRESS,
maker = Address.ZERO_ADDRESS,
): boolean {
const hook = this[hookName]

Expand Down
Loading