@aztec/end-to-end 0.0.1-commit.35158ae7e → 0.0.1-commit.3750d92a7
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +27 -0
- package/dest/bench/client_flows/client_flows_benchmark.d.ts +1 -1
- package/dest/bench/client_flows/client_flows_benchmark.d.ts.map +1 -1
- package/dest/bench/client_flows/client_flows_benchmark.js +11 -26
- package/dest/e2e_blacklist_token_contract/blacklist_token_contract_test.d.ts +3 -2
- package/dest/e2e_blacklist_token_contract/blacklist_token_contract_test.d.ts.map +1 -1
- package/dest/e2e_blacklist_token_contract/blacklist_token_contract_test.js +1 -1
- package/dest/e2e_epochs/epochs_test.js +3 -3
- package/dest/e2e_fees/fees_test.js +1 -1
- package/dest/e2e_p2p/inactivity_slash_test.js +3 -3
- package/dest/e2e_p2p/p2p_network.d.ts +5 -7
- package/dest/e2e_p2p/p2p_network.d.ts.map +1 -1
- package/dest/e2e_p2p/p2p_network.js +14 -13
- package/dest/e2e_p2p/reqresp/utils.js +1 -1
- package/dest/e2e_p2p/shared.d.ts +5 -7
- package/dest/e2e_p2p/shared.d.ts.map +1 -1
- package/dest/e2e_p2p/shared.js +16 -42
- package/dest/fixtures/authwit_proxy.d.ts +1 -1
- package/dest/fixtures/authwit_proxy.d.ts.map +1 -1
- package/dest/fixtures/authwit_proxy.js +4 -0
- package/dest/fixtures/e2e_prover_test.d.ts +1 -1
- package/dest/fixtures/e2e_prover_test.d.ts.map +1 -1
- package/dest/fixtures/e2e_prover_test.js +4 -7
- package/dest/fixtures/setup.d.ts +11 -8
- package/dest/fixtures/setup.d.ts.map +1 -1
- package/dest/fixtures/setup.js +11 -9
- package/dest/fixtures/setup_p2p_test.d.ts +6 -6
- package/dest/fixtures/setup_p2p_test.d.ts.map +1 -1
- package/dest/fixtures/setup_p2p_test.js +8 -8
- package/dest/fixtures/token_utils.d.ts +1 -1
- package/dest/fixtures/token_utils.d.ts.map +1 -1
- package/dest/fixtures/token_utils.js +2 -5
- package/dest/legacy-jest-resolver.d.cts +3 -0
- package/dest/legacy-jest-resolver.d.cts.map +1 -0
- package/dest/spartan/setup_test_wallets.d.ts +1 -1
- package/dest/spartan/setup_test_wallets.d.ts.map +1 -1
- package/dest/spartan/setup_test_wallets.js +9 -42
- package/dest/test-wallet/test_wallet.d.ts +16 -8
- package/dest/test-wallet/test_wallet.d.ts.map +1 -1
- package/dest/test-wallet/test_wallet.js +82 -46
- package/dest/test-wallet/worker_wallet.d.ts +4 -4
- package/dest/test-wallet/worker_wallet.d.ts.map +1 -1
- package/dest/test-wallet/worker_wallet_schema.d.ts +7 -2
- package/dest/test-wallet/worker_wallet_schema.d.ts.map +1 -1
- package/package.json +40 -39
- package/src/bench/client_flows/client_flows_benchmark.ts +29 -21
- package/src/e2e_blacklist_token_contract/blacklist_token_contract_test.ts +3 -6
- package/src/e2e_epochs/epochs_test.ts +3 -3
- package/src/e2e_fees/fees_test.ts +1 -1
- package/src/e2e_p2p/inactivity_slash_test.ts +3 -3
- package/src/e2e_p2p/p2p_network.ts +22 -27
- package/src/e2e_p2p/reqresp/utils.ts +1 -1
- package/src/e2e_p2p/shared.ts +16 -57
- package/src/fixtures/authwit_proxy.ts +4 -0
- package/src/fixtures/e2e_prover_test.ts +7 -6
- package/src/fixtures/setup.ts +21 -17
- package/src/fixtures/setup_p2p_test.ts +9 -9
- package/src/fixtures/token_utils.ts +1 -4
- package/src/legacy-jest-resolver.cjs +135 -0
- package/src/spartan/setup_test_wallets.ts +13 -35
- package/src/test-wallet/test_wallet.ts +100 -50
- package/src/test-wallet/worker_wallet.ts +3 -2
|
@@ -7,7 +7,7 @@ import { SecretValue } from '@aztec/foundation/config';
|
|
|
7
7
|
import { withLoggerBindings } from '@aztec/foundation/log/server';
|
|
8
8
|
import { bufferToHex } from '@aztec/foundation/string';
|
|
9
9
|
import type { DateProvider } from '@aztec/foundation/timer';
|
|
10
|
-
import type {
|
|
10
|
+
import type { GenesisData } from '@aztec/stdlib/world-state';
|
|
11
11
|
|
|
12
12
|
import getPort from 'get-port';
|
|
13
13
|
|
|
@@ -40,7 +40,7 @@ export async function createNodes(
|
|
|
40
40
|
bootstrapNodeEnr: string,
|
|
41
41
|
numNodes: number,
|
|
42
42
|
bootNodePort: number,
|
|
43
|
-
|
|
43
|
+
genesis?: GenesisData,
|
|
44
44
|
dataDirectory?: string,
|
|
45
45
|
metricsPort?: number,
|
|
46
46
|
indexOffset = 0,
|
|
@@ -65,7 +65,7 @@ export async function createNodes(
|
|
|
65
65
|
port,
|
|
66
66
|
bootstrapNodeEnr,
|
|
67
67
|
validatorIndices,
|
|
68
|
-
|
|
68
|
+
genesis,
|
|
69
69
|
dataDir,
|
|
70
70
|
metricsPort,
|
|
71
71
|
);
|
|
@@ -97,7 +97,7 @@ export async function createNode(
|
|
|
97
97
|
tcpPort: number,
|
|
98
98
|
bootstrapNode: string | undefined,
|
|
99
99
|
addressIndex: number | number[],
|
|
100
|
-
|
|
100
|
+
genesis?: GenesisData,
|
|
101
101
|
dataDirectory?: string,
|
|
102
102
|
metricsPort?: number,
|
|
103
103
|
) {
|
|
@@ -108,7 +108,7 @@ export async function createNode(
|
|
|
108
108
|
return await AztecNodeService.createAndSync(
|
|
109
109
|
validatorConfig,
|
|
110
110
|
{ telemetry, dateProvider },
|
|
111
|
-
{
|
|
111
|
+
{ genesis, dontStartSequencer: config.dontStartSequencer },
|
|
112
112
|
);
|
|
113
113
|
});
|
|
114
114
|
}
|
|
@@ -119,7 +119,7 @@ export async function createNonValidatorNode(
|
|
|
119
119
|
dateProvider: DateProvider,
|
|
120
120
|
tcpPort: number,
|
|
121
121
|
bootstrapNode: string | undefined,
|
|
122
|
-
|
|
122
|
+
genesis?: GenesisData,
|
|
123
123
|
dataDirectory?: string,
|
|
124
124
|
metricsPort?: number,
|
|
125
125
|
) {
|
|
@@ -133,7 +133,7 @@ export async function createNonValidatorNode(
|
|
|
133
133
|
sequencerPublisherPrivateKeys: [],
|
|
134
134
|
};
|
|
135
135
|
const telemetry = await getEndToEndTestTelemetryClient(metricsPort);
|
|
136
|
-
return await AztecNodeService.createAndSync(config, { telemetry, dateProvider }, {
|
|
136
|
+
return await AztecNodeService.createAndSync(config, { telemetry, dateProvider }, { genesis });
|
|
137
137
|
});
|
|
138
138
|
}
|
|
139
139
|
|
|
@@ -143,7 +143,7 @@ export async function createProverNode(
|
|
|
143
143
|
bootstrapNode: string | undefined,
|
|
144
144
|
addressIndex: number,
|
|
145
145
|
deps: { dateProvider: DateProvider },
|
|
146
|
-
|
|
146
|
+
genesis?: GenesisData,
|
|
147
147
|
dataDirectory?: string,
|
|
148
148
|
metricsPort?: number,
|
|
149
149
|
): Promise<{ proverNode: AztecNodeService }> {
|
|
@@ -159,7 +159,7 @@ export async function createProverNode(
|
|
|
159
159
|
{ ...config, ...p2pConfig },
|
|
160
160
|
{ dataDirectory },
|
|
161
161
|
{ ...deps, telemetry },
|
|
162
|
-
{
|
|
162
|
+
{ genesis },
|
|
163
163
|
);
|
|
164
164
|
});
|
|
165
165
|
}
|
|
@@ -6,11 +6,8 @@ import { TokenContract } from '@aztec/noir-contracts.js/Token';
|
|
|
6
6
|
|
|
7
7
|
export async function deployToken(wallet: Wallet, admin: AztecAddress, initialAdminBalance: bigint, logger: Logger) {
|
|
8
8
|
logger.info(`Deploying Token contract...`);
|
|
9
|
-
const {
|
|
10
|
-
receipt: { contract, instance },
|
|
11
|
-
} = await TokenContract.deploy(wallet, admin, 'TokenName', 'TokenSymbol', 18).send({
|
|
9
|
+
const { contract, instance } = await TokenContract.deploy(wallet, admin, 'TokenName', 'TokenSymbol', 18).send({
|
|
12
10
|
from: admin,
|
|
13
|
-
wait: { returnReceipt: true },
|
|
14
11
|
});
|
|
15
12
|
|
|
16
13
|
if (initialAdminBalance > 0n) {
|
|
@@ -0,0 +1,135 @@
|
|
|
1
|
+
// Custom Jest resolver. When CONTRACT_ARTIFACTS_VERSION is set, redirects *only* JSON artifact files under
|
|
2
|
+
// @aztec/noir-contracts.js/artifacts/, @aztec/noir-test-contracts.js/artifacts/, and @aztec/accounts/artifacts/ to a local cache of the pinned
|
|
3
|
+
// legacy versions. TypeScript wrapper classes (e.g. Token.ts) continue to load from the current workspace and use the
|
|
4
|
+
// current @aztec/aztec.js — only the artifact JSON (the deployed-contract ABI / bytecode / notes surface) is swapped.
|
|
5
|
+
//
|
|
6
|
+
// Why JSON-only: the JSON artifact is the actual interchange surface a "deployed contract" exposes. The TS wrapper is
|
|
7
|
+
// generated client-side ergonomics that's tightly coupled to the current @aztec/aztec.js API. Redirecting the wrapper
|
|
8
|
+
// would couple this test to a moving aztec.js surface and break at import time on unrelated breaking changes; we want
|
|
9
|
+
// to fail only on actual artifact-compat regressions.
|
|
10
|
+
//
|
|
11
|
+
// The cache is populated on demand by running `npm install` into .legacy-contracts/<version>/.
|
|
12
|
+
//
|
|
13
|
+
// Activated by env var; passthrough otherwise.
|
|
14
|
+
/* eslint-disable @typescript-eslint/no-require-imports */
|
|
15
|
+
|
|
16
|
+
const path = require('path');
|
|
17
|
+
const fs = require('fs');
|
|
18
|
+
const { execSync } = require('child_process');
|
|
19
|
+
|
|
20
|
+
const version = process.env.CONTRACT_ARTIFACTS_VERSION;
|
|
21
|
+
const REDIRECTED = ['@aztec/noir-contracts.js', '@aztec/noir-test-contracts.js', '@aztec/accounts'];
|
|
22
|
+
|
|
23
|
+
// Jest sets rootDir to <e2e>/src; this file lives there too.
|
|
24
|
+
const e2eRoot = path.resolve(__dirname, '..');
|
|
25
|
+
const cacheRoot = version ? path.join(e2eRoot, '.legacy-contracts', version) : null;
|
|
26
|
+
|
|
27
|
+
function pkgJsonPath(name) {
|
|
28
|
+
return path.join(cacheRoot, 'node_modules', name, 'package.json');
|
|
29
|
+
}
|
|
30
|
+
|
|
31
|
+
function ensureCache() {
|
|
32
|
+
const missing = REDIRECTED.some(p => !fs.existsSync(pkgJsonPath(p)));
|
|
33
|
+
if (!missing) {
|
|
34
|
+
return;
|
|
35
|
+
}
|
|
36
|
+
fs.mkdirSync(cacheRoot, { recursive: true });
|
|
37
|
+
// Seed a standalone package.json so `npm install --prefix` treats cacheRoot as its own project. Without this, npm
|
|
38
|
+
// walks up and finds the yarn-project workspace root, which breaks on `workspace:` protocol deps and risks
|
|
39
|
+
// clobbering the monorepo's node_modules.
|
|
40
|
+
const seed = path.join(cacheRoot, 'package.json');
|
|
41
|
+
if (!fs.existsSync(seed)) {
|
|
42
|
+
fs.writeFileSync(seed, JSON.stringify({ name: 'legacy-contracts-cache', private: true }));
|
|
43
|
+
}
|
|
44
|
+
|
|
45
|
+
const specs = REDIRECTED.map(p => `${p}@${version}`).join(' ');
|
|
46
|
+
process.stderr.write(`[legacy-contracts] installing ${specs} into ${cacheRoot}\n`);
|
|
47
|
+
// --prefix: install into cacheRoot instead of cwd, so the cache is isolated from the monorepo.
|
|
48
|
+
// --no-save: don't write the installed packages back to the seeded package.json.
|
|
49
|
+
// --ignore-scripts: skip lifecycle scripts (preinstall/postinstall) of the legacy packages and their transitive
|
|
50
|
+
// deps; we only want the files on disk, not to run any build steps.
|
|
51
|
+
// --legacy-peer-deps: tolerate peer-dependency mismatches between the pinned legacy @aztec/* graph and whatever
|
|
52
|
+
// current versions npm would otherwise try to reconcile.
|
|
53
|
+
execSync(`npm install --prefix "${cacheRoot}" --no-save --ignore-scripts --legacy-peer-deps ${specs}`, {
|
|
54
|
+
stdio: 'inherit',
|
|
55
|
+
});
|
|
56
|
+
|
|
57
|
+
// Verify versions on disk match the requested version.
|
|
58
|
+
for (const p of REDIRECTED) {
|
|
59
|
+
const onDisk = JSON.parse(fs.readFileSync(pkgJsonPath(p), 'utf8')).version;
|
|
60
|
+
if (onDisk !== version) {
|
|
61
|
+
throw new Error(`[legacy-contracts] ${p} on disk is ${onDisk}, expected ${version}`);
|
|
62
|
+
}
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
|
|
66
|
+
if (version) {
|
|
67
|
+
ensureCache();
|
|
68
|
+
}
|
|
69
|
+
|
|
70
|
+
let bannerPrinted = false;
|
|
71
|
+
const seen = new Set();
|
|
72
|
+
|
|
73
|
+
function printBannerOnce() {
|
|
74
|
+
if (bannerPrinted || !version) {
|
|
75
|
+
return;
|
|
76
|
+
}
|
|
77
|
+
bannerPrinted = true;
|
|
78
|
+
const lines = ['='.repeat(60), `[legacy-contracts][jest] CONTRACT_ARTIFACTS_VERSION=${version}`];
|
|
79
|
+
for (const p of REDIRECTED) {
|
|
80
|
+
const v = JSON.parse(fs.readFileSync(pkgJsonPath(p), 'utf8')).version;
|
|
81
|
+
if (v !== version) {
|
|
82
|
+
throw new Error(`[legacy-contracts] ${p} on disk is ${v}, expected ${version}`);
|
|
83
|
+
}
|
|
84
|
+
lines.push(`[legacy-contracts][jest] redirecting ${p}/artifacts/*.json -> .legacy-contracts/${version}/...`);
|
|
85
|
+
}
|
|
86
|
+
lines.push('='.repeat(60));
|
|
87
|
+
process.stderr.write(lines.join('\n') + '\n');
|
|
88
|
+
}
|
|
89
|
+
|
|
90
|
+
// Match a resolved absolute path against the workspace artifacts dirs and return the legacy cache equivalent, or null
|
|
91
|
+
// if it's not an artifact path we should redirect.
|
|
92
|
+
function legacyArtifactPath(resolved) {
|
|
93
|
+
if (!resolved.endsWith('.json')) {
|
|
94
|
+
return null;
|
|
95
|
+
}
|
|
96
|
+
for (const pkg of REDIRECTED) {
|
|
97
|
+
// pkg = '@aztec/noir-contracts.js' -> match '/noir-contracts.js/artifacts/'
|
|
98
|
+
const dirName = pkg.split('/')[1];
|
|
99
|
+
const marker = `/${dirName}/artifacts/`;
|
|
100
|
+
const idx = resolved.indexOf(marker);
|
|
101
|
+
if (idx === -1) {
|
|
102
|
+
continue;
|
|
103
|
+
}
|
|
104
|
+
const basename = resolved.slice(idx + marker.length);
|
|
105
|
+
return path.join(cacheRoot, 'node_modules', pkg, 'artifacts', basename);
|
|
106
|
+
}
|
|
107
|
+
return null;
|
|
108
|
+
}
|
|
109
|
+
|
|
110
|
+
module.exports = function legacyResolver(request, options) {
|
|
111
|
+
// Always run the default resolver first. We only inspect (and possibly rewrite) the *result*; this catches both
|
|
112
|
+
// bare-specifier imports of `@aztec/noir-contracts.js/artifacts/foo.json` and the relative `../artifacts/foo.json`
|
|
113
|
+
// imports inside the workspace TS wrapper classes — both resolve to the same workspace artifact path that we then
|
|
114
|
+
// redirect.
|
|
115
|
+
const resolved = options.defaultResolver(request, options);
|
|
116
|
+
if (!version) {
|
|
117
|
+
return resolved;
|
|
118
|
+
}
|
|
119
|
+
printBannerOnce();
|
|
120
|
+
const legacy = legacyArtifactPath(resolved);
|
|
121
|
+
if (!legacy) {
|
|
122
|
+
return resolved;
|
|
123
|
+
}
|
|
124
|
+
if (!fs.existsSync(legacy)) {
|
|
125
|
+
throw new Error(
|
|
126
|
+
`[legacy-contracts] artifact ${path.basename(legacy)} not present in legacy cache @${version}; ` +
|
|
127
|
+
`the contract may have been added after that release. Pin a newer CONTRACT_ARTIFACTS_VERSION or skip this test.`,
|
|
128
|
+
);
|
|
129
|
+
}
|
|
130
|
+
if (!seen.has(resolved)) {
|
|
131
|
+
seen.add(resolved);
|
|
132
|
+
process.stderr.write(`[legacy-contracts][jest] redirected ${path.basename(legacy)} -> ${legacy}\n`);
|
|
133
|
+
}
|
|
134
|
+
return legacy;
|
|
135
|
+
};
|
|
@@ -138,37 +138,11 @@ async function deployAccountWithDiagnostics(
|
|
|
138
138
|
estimateGas?: boolean,
|
|
139
139
|
): Promise<void> {
|
|
140
140
|
const deployMethod = await account.getDeployMethod();
|
|
141
|
-
let txHash;
|
|
142
141
|
let gasSettings: any;
|
|
143
|
-
|
|
144
|
-
|
|
145
|
-
|
|
146
|
-
|
|
147
|
-
logger.info(`${accountLabel} estimated gas: DA=${gasSettings.gasLimits.daGas} L2=${gasSettings.gasLimits.l2Gas}`);
|
|
148
|
-
}
|
|
149
|
-
const deployResult = await deployMethod.send({
|
|
150
|
-
from: NO_FROM,
|
|
151
|
-
fee: { paymentMethod, gasSettings },
|
|
152
|
-
wait: NO_WAIT,
|
|
153
|
-
});
|
|
154
|
-
txHash = deployResult.txHash;
|
|
155
|
-
await waitForTx(aztecNode, txHash, { timeout: 2400 });
|
|
156
|
-
logger.info(`${accountLabel} deployed at ${account.address}`);
|
|
157
|
-
} catch (error) {
|
|
158
|
-
const blockNumber = await aztecNode.getBlockNumber();
|
|
159
|
-
let receipt;
|
|
160
|
-
try {
|
|
161
|
-
receipt = await aztecNode.getTxReceipt(txHash);
|
|
162
|
-
} catch {
|
|
163
|
-
receipt = 'unavailable';
|
|
164
|
-
}
|
|
165
|
-
logger.error(`${accountLabel} deployment failed`, {
|
|
166
|
-
txHash: txHash.toString(),
|
|
167
|
-
receipt: JSON.stringify(receipt),
|
|
168
|
-
currentBlockNumber: blockNumber,
|
|
169
|
-
error: String(error),
|
|
170
|
-
});
|
|
171
|
-
throw error;
|
|
142
|
+
if (estimateGas) {
|
|
143
|
+
const sim = await deployMethod.simulate({ from: NO_FROM, fee: { paymentMethod } });
|
|
144
|
+
gasSettings = sim.estimatedGas;
|
|
145
|
+
logger.info(`${accountLabel} estimated gas: DA=${gasSettings.gasLimits.daGas} L2=${gasSettings.gasLimits.l2Gas}`);
|
|
172
146
|
}
|
|
173
147
|
|
|
174
148
|
// Track the tx hash across retries so we don't re-send when the previous tx is still pending.
|
|
@@ -196,7 +170,7 @@ async function deployAccountWithDiagnostics(
|
|
|
196
170
|
|
|
197
171
|
if (!sentTxHash) {
|
|
198
172
|
const deployResult = await deployMethod.send({
|
|
199
|
-
from:
|
|
173
|
+
from: NO_FROM,
|
|
200
174
|
fee: { paymentMethod, gasSettings },
|
|
201
175
|
wait: NO_WAIT,
|
|
202
176
|
});
|
|
@@ -391,14 +365,18 @@ async function deployTokenAndMint(
|
|
|
391
365
|
logger: Logger,
|
|
392
366
|
) {
|
|
393
367
|
logger.verbose(`Deploying TokenContract...`);
|
|
394
|
-
const {
|
|
395
|
-
|
|
396
|
-
|
|
368
|
+
const { contract: tokenContract } = await TokenContract.deploy(
|
|
369
|
+
wallet,
|
|
370
|
+
admin,
|
|
371
|
+
TOKEN_NAME,
|
|
372
|
+
TOKEN_SYMBOL,
|
|
373
|
+
TOKEN_DECIMALS,
|
|
374
|
+
).send({
|
|
397
375
|
from: admin,
|
|
398
376
|
fee: {
|
|
399
377
|
paymentMethod,
|
|
400
378
|
},
|
|
401
|
-
wait: { timeout: 600
|
|
379
|
+
wait: { timeout: 600 },
|
|
402
380
|
});
|
|
403
381
|
|
|
404
382
|
const tokenAddress = tokenContract.address;
|
|
@@ -1,7 +1,9 @@
|
|
|
1
1
|
import { EcdsaKAccountContract, EcdsaRAccountContract } from '@aztec/accounts/ecdsa';
|
|
2
2
|
import { SchnorrAccountContract } from '@aztec/accounts/schnorr';
|
|
3
|
-
import {
|
|
3
|
+
import { StubEcdsaAccountContractArtifact, createStubEcdsaAccount } from '@aztec/accounts/stub/ecdsa';
|
|
4
|
+
import { StubSchnorrAccountContractArtifact, createStubSchnorrAccount } from '@aztec/accounts/stub/schnorr';
|
|
4
5
|
import { type Account, type AccountContract, NO_FROM } from '@aztec/aztec.js/account';
|
|
6
|
+
import type { CompleteAddress } from '@aztec/aztec.js/addresses';
|
|
5
7
|
import {
|
|
6
8
|
type CallIntent,
|
|
7
9
|
type ContractFunctionInteractionCallIntent,
|
|
@@ -13,6 +15,7 @@ import {
|
|
|
13
15
|
} from '@aztec/aztec.js/authorization';
|
|
14
16
|
import type { AztecNode } from '@aztec/aztec.js/node';
|
|
15
17
|
import { AccountManager, type SendOptions } from '@aztec/aztec.js/wallet';
|
|
18
|
+
import { TxSimulationResultWithAppOffset } from '@aztec/aztec.js/wallet';
|
|
16
19
|
import type { DefaultAccountEntrypointOptions } from '@aztec/entrypoints/account';
|
|
17
20
|
import { DefaultEntrypoint } from '@aztec/entrypoints/default';
|
|
18
21
|
import { Fq, Fr } from '@aztec/foundation/curves/bn254';
|
|
@@ -25,16 +28,17 @@ import { AztecAddress } from '@aztec/stdlib/aztec-address';
|
|
|
25
28
|
import { getContractInstanceFromInstantiationParams } from '@aztec/stdlib/contract';
|
|
26
29
|
import { deriveSigningKey } from '@aztec/stdlib/keys';
|
|
27
30
|
import type { NoteDao } from '@aztec/stdlib/note';
|
|
28
|
-
import
|
|
29
|
-
BlockHeader,
|
|
31
|
+
import {
|
|
32
|
+
type BlockHeader,
|
|
33
|
+
type ContractOverrides,
|
|
30
34
|
SimulationOverrides,
|
|
31
|
-
TxExecutionRequest,
|
|
32
|
-
TxHash,
|
|
33
|
-
TxReceipt,
|
|
34
|
-
TxSimulationResult,
|
|
35
|
+
type TxExecutionRequest,
|
|
36
|
+
type TxHash,
|
|
37
|
+
type TxReceipt,
|
|
35
38
|
} from '@aztec/stdlib/tx';
|
|
36
39
|
import { ExecutionPayload, mergeExecutionPayloads } from '@aztec/stdlib/tx';
|
|
37
40
|
import { BaseWallet, type SimulateViaEntrypointOptions } from '@aztec/wallet-sdk/base-wallet';
|
|
41
|
+
import type { AccountType } from '@aztec/wallets/embedded';
|
|
38
42
|
|
|
39
43
|
import { AztecNodeProxy, ProvenTx } from './utils.js';
|
|
40
44
|
|
|
@@ -44,6 +48,7 @@ import { AztecNodeProxy, ProvenTx } from './utils.js';
|
|
|
44
48
|
export interface AccountData {
|
|
45
49
|
secret: Fr;
|
|
46
50
|
salt: Fr;
|
|
51
|
+
type?: AccountType;
|
|
47
52
|
contract: AccountContract;
|
|
48
53
|
}
|
|
49
54
|
|
|
@@ -84,50 +89,81 @@ export class TestWallet extends BaseWallet {
|
|
|
84
89
|
|
|
85
90
|
createSchnorrAccount(secret: Fr, salt: Fr, signingKey?: Fq): Promise<AccountManager> {
|
|
86
91
|
signingKey = signingKey ?? deriveSigningKey(secret);
|
|
87
|
-
|
|
88
|
-
secret,
|
|
89
|
-
salt,
|
|
90
|
-
contract: new SchnorrAccountContract(signingKey),
|
|
91
|
-
};
|
|
92
|
-
return this.createAccount(accountData);
|
|
92
|
+
return this.createAccount({ secret, salt, type: 'schnorr', contract: new SchnorrAccountContract(signingKey) });
|
|
93
93
|
}
|
|
94
94
|
|
|
95
95
|
createECDSARAccount(secret: Fr, salt: Fr, signingKey: Buffer): Promise<AccountManager> {
|
|
96
|
-
|
|
96
|
+
return this.createAccount({
|
|
97
97
|
secret,
|
|
98
98
|
salt,
|
|
99
|
+
type: 'ecdsasecp256r1',
|
|
99
100
|
contract: new EcdsaRAccountContract(signingKey),
|
|
100
|
-
};
|
|
101
|
-
return this.createAccount(accountData);
|
|
101
|
+
});
|
|
102
102
|
}
|
|
103
103
|
|
|
104
104
|
createECDSAKAccount(secret: Fr, salt: Fr, signingKey: Buffer): Promise<AccountManager> {
|
|
105
|
-
|
|
105
|
+
return this.createAccount({
|
|
106
106
|
secret,
|
|
107
107
|
salt,
|
|
108
|
+
type: 'ecdsasecp256k1',
|
|
108
109
|
contract: new EcdsaKAccountContract(signingKey),
|
|
109
|
-
};
|
|
110
|
-
return this.createAccount(accountData);
|
|
110
|
+
});
|
|
111
111
|
}
|
|
112
112
|
|
|
113
|
-
|
|
114
|
-
|
|
115
|
-
|
|
116
|
-
|
|
117
|
-
|
|
118
|
-
|
|
113
|
+
/**
|
|
114
|
+
* Builds contract overrides for all provided addresses by replacing their account contracts with stub implementations.
|
|
115
|
+
*/
|
|
116
|
+
protected async buildAccountOverrides(addresses: AztecAddress[]): Promise<ContractOverrides> {
|
|
117
|
+
const accounts = await this.getAccounts();
|
|
118
|
+
const contracts: ContractOverrides = {};
|
|
119
|
+
|
|
120
|
+
const filtered = accounts.filter(acc => addresses.some(addr => addr.equals(acc.item)));
|
|
121
|
+
|
|
122
|
+
for (const account of filtered) {
|
|
123
|
+
const address = account.item;
|
|
124
|
+
const originalAccount = await this.getAccountFromAddress(address);
|
|
125
|
+
const completeAddress = originalAccount.getCompleteAddress();
|
|
126
|
+
const contractInstance = await this.pxe.getContractInstance(completeAddress.address);
|
|
127
|
+
if (!contractInstance) {
|
|
128
|
+
throw new Error(
|
|
129
|
+
`No contract instance found for address: ${completeAddress.address} during account override building. This is a bug!`,
|
|
130
|
+
);
|
|
131
|
+
}
|
|
132
|
+
|
|
133
|
+
const stubArtifact = this.getStubArtifactFor(address);
|
|
134
|
+
const stubConstructorArgs =
|
|
135
|
+
this.getTypeFor(address) === 'schnorr' ? [Fr.ZERO, Fr.ZERO] : [Buffer.alloc(32), Buffer.alloc(32)];
|
|
136
|
+
const stubInstance = await getContractInstanceFromInstantiationParams(stubArtifact, {
|
|
137
|
+
salt: Fr.random(),
|
|
138
|
+
constructorArgs: stubConstructorArgs,
|
|
139
|
+
});
|
|
140
|
+
|
|
141
|
+
contracts[address.toString()] = {
|
|
142
|
+
instance: stubInstance,
|
|
143
|
+
artifact: stubArtifact,
|
|
144
|
+
};
|
|
119
145
|
}
|
|
120
|
-
|
|
121
|
-
|
|
122
|
-
|
|
123
|
-
|
|
124
|
-
|
|
125
|
-
|
|
126
|
-
|
|
127
|
-
|
|
128
|
-
|
|
146
|
+
|
|
147
|
+
return contracts;
|
|
148
|
+
}
|
|
149
|
+
|
|
150
|
+
protected accounts: Map<string, { account: Account; type: AccountType }> = new Map();
|
|
151
|
+
|
|
152
|
+
private getTypeFor(address: AztecAddress): AccountType {
|
|
153
|
+
return this.accounts.get(address.toString())?.type ?? 'schnorr';
|
|
154
|
+
}
|
|
155
|
+
|
|
156
|
+
private getStubArtifactFor(address: AztecAddress) {
|
|
157
|
+
return this.getTypeFor(address) === 'schnorr'
|
|
158
|
+
? StubSchnorrAccountContractArtifact
|
|
159
|
+
: StubEcdsaAccountContractArtifact;
|
|
160
|
+
}
|
|
161
|
+
|
|
162
|
+
private getStubAccountFor(address: AztecAddress, completeAddress: CompleteAddress) {
|
|
163
|
+
return this.getTypeFor(address) === 'schnorr'
|
|
164
|
+
? createStubSchnorrAccount(completeAddress)
|
|
165
|
+
: createStubEcdsaAccount(completeAddress);
|
|
129
166
|
}
|
|
130
|
-
protected accounts: Map<string, Account> = new Map();
|
|
131
167
|
|
|
132
168
|
/**
|
|
133
169
|
* Controls how the test wallet simulates transactions:
|
|
@@ -146,22 +182,25 @@ export class TestWallet extends BaseWallet {
|
|
|
146
182
|
}
|
|
147
183
|
|
|
148
184
|
protected getAccountFromAddress(address: AztecAddress): Promise<Account> {
|
|
149
|
-
const
|
|
185
|
+
const entry = this.accounts.get(address?.toString() ?? '');
|
|
150
186
|
|
|
151
|
-
if (!
|
|
187
|
+
if (!entry) {
|
|
152
188
|
throw new Error(`Account not found in wallet for address: ${address}`);
|
|
153
189
|
}
|
|
154
190
|
|
|
155
|
-
return Promise.resolve(account);
|
|
191
|
+
return Promise.resolve(entry.account);
|
|
156
192
|
}
|
|
157
193
|
|
|
158
194
|
getAccounts() {
|
|
159
|
-
return Promise.resolve(
|
|
195
|
+
return Promise.resolve(
|
|
196
|
+
Array.from(this.accounts.values()).map(entry => ({ alias: '', item: entry.account.getAddress() })),
|
|
197
|
+
);
|
|
160
198
|
}
|
|
161
199
|
|
|
162
200
|
async createAccount(accountData?: AccountData): Promise<AccountManager> {
|
|
163
201
|
const secret = accountData?.secret ?? Fr.random();
|
|
164
202
|
const salt = accountData?.salt ?? Fr.random();
|
|
203
|
+
const type = accountData?.type ?? 'schnorr';
|
|
165
204
|
const contract = accountData?.contract ?? new SchnorrAccountContract(GrumpkinScalar.random());
|
|
166
205
|
|
|
167
206
|
const accountManager = await AccountManager.create(this, secret, contract, salt);
|
|
@@ -171,7 +210,8 @@ export class TestWallet extends BaseWallet {
|
|
|
171
210
|
|
|
172
211
|
await this.registerContract(instance, artifact, secret);
|
|
173
212
|
|
|
174
|
-
|
|
213
|
+
const address = accountManager.address.toString();
|
|
214
|
+
this.accounts.set(address, { account: await accountManager.getAccount(), type });
|
|
175
215
|
|
|
176
216
|
return accountManager;
|
|
177
217
|
}
|
|
@@ -217,9 +257,12 @@ export class TestWallet extends BaseWallet {
|
|
|
217
257
|
protected override async simulateViaEntrypoint(
|
|
218
258
|
executionPayload: ExecutionPayload,
|
|
219
259
|
opts: SimulateViaEntrypointOptions,
|
|
220
|
-
): Promise<
|
|
221
|
-
const { from, feeOptions,
|
|
260
|
+
): Promise<TxSimulationResultWithAppOffset> {
|
|
261
|
+
const { from, feeOptions, additionalScopes, skipTxValidation, skipFeeEnforcement } = opts;
|
|
262
|
+
const scopes = this.scopesFrom(from, additionalScopes);
|
|
222
263
|
const skipKernels = this.simulationMode !== 'full';
|
|
264
|
+
const useOverride = this.simulationMode === 'kernelless-override';
|
|
265
|
+
|
|
223
266
|
const feeExecutionPayload = await feeOptions.walletFeePaymentMethod?.getExecutionPayload();
|
|
224
267
|
const finalExecutionPayload = feeExecutionPayload
|
|
225
268
|
? mergeExecutionPayloads([feeExecutionPayload, executionPayload])
|
|
@@ -228,18 +271,19 @@ export class TestWallet extends BaseWallet {
|
|
|
228
271
|
|
|
229
272
|
let overrides: SimulationOverrides | undefined;
|
|
230
273
|
let txRequest: TxExecutionRequest;
|
|
274
|
+
if (useOverride) {
|
|
275
|
+
const accountOverrides = await this.buildAccountOverrides(scopes);
|
|
276
|
+
overrides = new SimulationOverrides(accountOverrides);
|
|
277
|
+
}
|
|
278
|
+
|
|
231
279
|
if (from === NO_FROM) {
|
|
232
280
|
const entrypoint = new DefaultEntrypoint();
|
|
233
281
|
txRequest = await entrypoint.createTxExecutionRequest(finalExecutionPayload, feeOptions.gasSettings, chainInfo);
|
|
234
282
|
} else {
|
|
235
|
-
const useOverride = this.simulationMode === 'kernelless-override';
|
|
236
283
|
let fromAccount: Account;
|
|
237
284
|
if (useOverride) {
|
|
238
|
-
const
|
|
239
|
-
fromAccount =
|
|
240
|
-
overrides = {
|
|
241
|
-
contracts: { [from.toString()]: { instance, artifact } },
|
|
242
|
-
};
|
|
285
|
+
const originalAccount = await this.getAccountFromAddress(from);
|
|
286
|
+
fromAccount = this.getStubAccountFor(from, originalAccount.getCompleteAddress());
|
|
243
287
|
} else {
|
|
244
288
|
fromAccount = await this.getAccountFromAddress(from);
|
|
245
289
|
}
|
|
@@ -257,7 +301,7 @@ export class TestWallet extends BaseWallet {
|
|
|
257
301
|
);
|
|
258
302
|
}
|
|
259
303
|
|
|
260
|
-
|
|
304
|
+
const result = await this.pxe.simulateTx(txRequest, {
|
|
261
305
|
simulatePublic: true,
|
|
262
306
|
skipKernels,
|
|
263
307
|
skipFeeEnforcement,
|
|
@@ -265,10 +309,16 @@ export class TestWallet extends BaseWallet {
|
|
|
265
309
|
overrides,
|
|
266
310
|
scopes,
|
|
267
311
|
});
|
|
312
|
+
const appCallOffset = await this.computeAppCallOffset(from, feeOptions);
|
|
313
|
+
return TxSimulationResultWithAppOffset.fromResultAndOffset(result, appCallOffset);
|
|
268
314
|
}
|
|
269
315
|
|
|
270
316
|
async proveTx(exec: ExecutionPayload, opts: Omit<SendOptions, 'wait'>): Promise<ProvenTx> {
|
|
271
|
-
const fee = await this.completeFeeOptions(
|
|
317
|
+
const fee = await this.completeFeeOptions({
|
|
318
|
+
from: opts.from,
|
|
319
|
+
feePayer: exec.feePayer,
|
|
320
|
+
gasSettings: opts.fee?.gasSettings,
|
|
321
|
+
});
|
|
272
322
|
const txRequest = await this.createTxExecutionRequestFromPayloadAndFee(exec, opts.from, fee);
|
|
273
323
|
const txProvingResult = await this.pxe.proveTx(txRequest, this.scopesFrom(opts.from, opts.additionalScopes));
|
|
274
324
|
return new ProvenTx(
|
|
@@ -13,6 +13,7 @@ import type {
|
|
|
13
13
|
ProfileOptions,
|
|
14
14
|
SendOptions,
|
|
15
15
|
SimulateOptions,
|
|
16
|
+
TxSimulationResultWithAppOffset,
|
|
16
17
|
Wallet,
|
|
17
18
|
WalletCapabilities,
|
|
18
19
|
} from '@aztec/aztec.js/wallet';
|
|
@@ -29,7 +30,7 @@ import type { ContractArtifact, EventMetadataDefinition, FunctionCall } from '@a
|
|
|
29
30
|
import type { AuthWitness } from '@aztec/stdlib/auth-witness';
|
|
30
31
|
import type { AztecAddress } from '@aztec/stdlib/aztec-address';
|
|
31
32
|
import type { ContractInstanceWithAddress } from '@aztec/stdlib/contract';
|
|
32
|
-
import type { ExecutionPayload, TxProfileResult,
|
|
33
|
+
import type { ExecutionPayload, TxProfileResult, UtilityExecutionResult } from '@aztec/stdlib/tx';
|
|
33
34
|
import { Tx } from '@aztec/stdlib/tx';
|
|
34
35
|
|
|
35
36
|
import { Worker } from 'worker_threads';
|
|
@@ -165,7 +166,7 @@ export class WorkerWallet implements Wallet {
|
|
|
165
166
|
return this.call('registerContract', instance, artifact, secretKey);
|
|
166
167
|
}
|
|
167
168
|
|
|
168
|
-
simulateTx(exec: ExecutionPayload, opts: SimulateOptions): Promise<
|
|
169
|
+
simulateTx(exec: ExecutionPayload, opts: SimulateOptions): Promise<TxSimulationResultWithAppOffset> {
|
|
169
170
|
return this.call('simulateTx', exec, opts);
|
|
170
171
|
}
|
|
171
172
|
|