1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
import { OpCode, StructLog } from '@0xproject/types';
import * as chai from 'chai';
import * as fs from 'fs';
import * as _ from 'lodash';
import 'mocha';
import * as path from 'path';
import { getTracesByContractAddress } from '../src/trace';
const expect = chai.expect;
const DEFAULT_STRUCT_LOG: StructLog = {
depth: 0,
error: '',
gas: 0,
gasCost: 0,
memory: [],
op: OpCode.Invalid,
pc: 0,
stack: [],
storage: {},
};
function addDefaultStructLogFields(compactStructLog: Partial<StructLog> & { op: OpCode; depth: number }): StructLog {
return { ...DEFAULT_STRUCT_LOG, ...compactStructLog };
}
describe('Trace', () => {
describe('#getTracesByContractAddress', () => {
it('correctly splits trace by contract address', () => {
const delegateCallAddress = '0x0000000000000000000000000000000000000002';
const trace = [
{
op: OpCode.DelegateCall,
stack: ['0x', '0x', '0x', '0x', delegateCallAddress],
depth: 0,
},
{
op: OpCode.Return,
depth: 1,
},
{
op: OpCode.Return,
depth: 0,
},
];
const fullTrace = _.map(trace, compactStructLog => addDefaultStructLogFields(compactStructLog));
const startAddress = '0x0000000000000000000000000000000000000001';
const traceByContractAddress = getTracesByContractAddress(fullTrace, startAddress);
const expectedTraceByContractAddress = {
[startAddress]: [fullTrace[0], fullTrace[2]],
[delegateCallAddress]: [fullTrace[1]],
};
expect(traceByContractAddress).to.be.deep.equal(expectedTraceByContractAddress);
});
});
});
|
