@aztec/txe 5.0.0-private.20260319 → 5.0.0-rc.2

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.
Files changed (160) hide show
  1. package/dest/AuthRegistry-URXCCPYF.js +3 -0
  2. package/dest/AuthRegistry-URXCCPYF.js.map +7 -0
  3. package/dest/ContractClassRegistry-5RDBM6TC.js +3 -0
  4. package/dest/ContractClassRegistry-5RDBM6TC.js.map +7 -0
  5. package/dest/ContractInstanceRegistry-GCR4G4D7.js +3 -0
  6. package/dest/ContractInstanceRegistry-GCR4G4D7.js.map +7 -0
  7. package/dest/FeeJuice-O3KOSZO5.js +3 -0
  8. package/dest/FeeJuice-O3KOSZO5.js.map +7 -0
  9. package/dest/HandshakeRegistry-TXSMHMNY.js +3 -0
  10. package/dest/HandshakeRegistry-TXSMHMNY.js.map +7 -0
  11. package/dest/MultiCallEntrypoint-P4FSRXAD.js +3 -0
  12. package/dest/MultiCallEntrypoint-P4FSRXAD.js.map +7 -0
  13. package/dest/SchnorrAccount-AQZX7ROU.js +3 -0
  14. package/dest/SchnorrAccount-AQZX7ROU.js.map +7 -0
  15. package/dest/SchnorrInitializerlessAccount-KMX2H3J6.js +3 -0
  16. package/dest/SchnorrInitializerlessAccount-KMX2H3J6.js.map +7 -0
  17. package/dest/bin/check_txe_oracle_version.d.ts +2 -0
  18. package/dest/bin/check_txe_oracle_version.d.ts.map +1 -0
  19. package/dest/bin/check_txe_oracle_version.js +61 -0
  20. package/dest/bin/index.js +3 -30
  21. package/dest/bin/index.js.map +7 -0
  22. package/dest/bin/oracle_test_server.d.ts +3 -0
  23. package/dest/bin/oracle_test_server.d.ts.map +1 -0
  24. package/dest/bin/oracle_test_server.js +41 -0
  25. package/dest/chunk-BJVAAXNA.js +3 -0
  26. package/dest/chunk-BJVAAXNA.js.map +7 -0
  27. package/dest/chunk-I6DL4FT2.js +304 -0
  28. package/dest/chunk-I6DL4FT2.js.map +7 -0
  29. package/dest/chunk-IAO54JS2.js +266 -0
  30. package/dest/chunk-IAO54JS2.js.map +7 -0
  31. package/dest/constants.d.ts +5 -1
  32. package/dest/constants.d.ts.map +1 -1
  33. package/dest/constants.js +9 -1
  34. package/dest/dispatcher_pool.d.ts +67 -0
  35. package/dest/dispatcher_pool.d.ts.map +1 -0
  36. package/dest/dispatcher_pool.js +286 -0
  37. package/dest/index.d.ts +51 -7
  38. package/dest/index.d.ts.map +1 -1
  39. package/dest/index.js +70 -190
  40. package/dest/metafile.json +39364 -0
  41. package/dest/msgpackr_fr_extension.d.ts +2 -0
  42. package/dest/msgpackr_fr_extension.d.ts.map +1 -0
  43. package/dest/msgpackr_fr_extension.js +21 -0
  44. package/dest/oracle/interfaces.d.ts +38 -8
  45. package/dest/oracle/interfaces.d.ts.map +1 -1
  46. package/dest/oracle/test-resolver/default_fixtures.d.ts +9 -0
  47. package/dest/oracle/test-resolver/default_fixtures.d.ts.map +1 -0
  48. package/dest/oracle/test-resolver/default_fixtures.js +164 -0
  49. package/dest/oracle/test-resolver/index.d.ts +9 -0
  50. package/dest/oracle/test-resolver/index.d.ts.map +1 -0
  51. package/dest/oracle/test-resolver/index.js +32 -0
  52. package/dest/oracle/test-resolver/resolver.d.ts +48 -0
  53. package/dest/oracle/test-resolver/resolver.d.ts.map +1 -0
  54. package/dest/oracle/test-resolver/resolver.js +144 -0
  55. package/dest/oracle/txe_oracle_public_context.d.ts +26 -2
  56. package/dest/oracle/txe_oracle_public_context.d.ts.map +1 -1
  57. package/dest/oracle/txe_oracle_public_context.js +43 -1
  58. package/dest/oracle/txe_oracle_registry.d.ts +167 -0
  59. package/dest/oracle/txe_oracle_registry.d.ts.map +1 -0
  60. package/dest/oracle/txe_oracle_registry.js +715 -0
  61. package/dest/oracle/txe_oracle_top_level_context.d.ts +42 -17
  62. package/dest/oracle/txe_oracle_top_level_context.d.ts.map +1 -1
  63. package/dest/oracle/txe_oracle_top_level_context.js +205 -70
  64. package/dest/oracle/txe_oracle_version.d.ts +17 -0
  65. package/dest/oracle/txe_oracle_version.d.ts.map +1 -0
  66. package/dest/oracle/txe_oracle_version.js +14 -0
  67. package/dest/oracle/txe_private_execution_oracle.d.ts +17 -0
  68. package/dest/oracle/txe_private_execution_oracle.d.ts.map +1 -0
  69. package/dest/oracle/txe_private_execution_oracle.js +15 -0
  70. package/dest/rpc_server.d.ts +14 -0
  71. package/dest/rpc_server.d.ts.map +1 -0
  72. package/dest/rpc_server.js +78 -0
  73. package/dest/rpc_translator.d.ts +111 -233
  74. package/dest/rpc_translator.d.ts.map +1 -1
  75. package/dest/rpc_translator.js +759 -636
  76. package/dest/server.bundle.js +3 -0
  77. package/dest/server.bundle.js.map +7 -0
  78. package/dest/state_machine/archiver.d.ts +4 -3
  79. package/dest/state_machine/archiver.d.ts.map +1 -1
  80. package/dest/state_machine/archiver.js +24 -14
  81. package/dest/state_machine/dummy_p2p_client.d.ts +15 -7
  82. package/dest/state_machine/dummy_p2p_client.d.ts.map +1 -1
  83. package/dest/state_machine/dummy_p2p_client.js +22 -4
  84. package/dest/state_machine/global_variable_builder.d.ts +10 -6
  85. package/dest/state_machine/global_variable_builder.d.ts.map +1 -1
  86. package/dest/state_machine/global_variable_builder.js +8 -5
  87. package/dest/state_machine/index.d.ts +9 -6
  88. package/dest/state_machine/index.d.ts.map +1 -1
  89. package/dest/state_machine/index.js +45 -12
  90. package/dest/state_machine/mock_epoch_cache.d.ts +1 -2
  91. package/dest/state_machine/mock_epoch_cache.d.ts.map +1 -1
  92. package/dest/state_machine/mock_epoch_cache.js +0 -3
  93. package/dest/state_machine/synchronizer.d.ts +3 -2
  94. package/dest/state_machine/synchronizer.d.ts.map +1 -1
  95. package/dest/state_machine/synchronizer.js +4 -3
  96. package/dest/txe_session.d.ts +89 -20
  97. package/dest/txe_session.d.ts.map +1 -1
  98. package/dest/txe_session.js +289 -64
  99. package/dest/utils/block_creation.d.ts +2 -2
  100. package/dest/utils/block_creation.d.ts.map +1 -1
  101. package/dest/utils/block_creation.js +2 -2
  102. package/dest/utils/encoding.d.ts +191 -0
  103. package/dest/utils/encoding.d.ts.map +1 -0
  104. package/dest/{util → utils}/encoding.js +7 -2
  105. package/dest/{util → utils}/expected_failure_error.d.ts +1 -1
  106. package/dest/utils/expected_failure_error.d.ts.map +1 -0
  107. package/dest/{util → utils}/txe_account_store.d.ts +1 -1
  108. package/dest/utils/txe_account_store.d.ts.map +1 -0
  109. package/dest/utils/txe_artifact_resolver.d.ts +37 -0
  110. package/dest/utils/txe_artifact_resolver.d.ts.map +1 -0
  111. package/dest/utils/txe_artifact_resolver.js +161 -0
  112. package/dest/utils/txe_public_contract_data_source.d.ts +20 -0
  113. package/dest/utils/txe_public_contract_data_source.d.ts.map +1 -0
  114. package/dest/{util → utils}/txe_public_contract_data_source.js +1 -3
  115. package/dest/worker.bundle.js +3 -0
  116. package/dest/worker.bundle.js.map +7 -0
  117. package/dest/worker.d.ts +2 -0
  118. package/dest/worker.d.ts.map +1 -0
  119. package/dest/worker.js +92 -0
  120. package/package.json +38 -21
  121. package/src/bin/check_txe_oracle_version.ts +70 -0
  122. package/src/bin/index.ts +11 -2
  123. package/src/bin/oracle_test_server.ts +51 -0
  124. package/src/constants.ts +11 -1
  125. package/src/dispatcher_pool.ts +317 -0
  126. package/src/index.ts +97 -227
  127. package/src/msgpackr_fr_extension.ts +23 -0
  128. package/src/oracle/interfaces.ts +35 -7
  129. package/src/oracle/test-resolver/default_fixtures.ts +226 -0
  130. package/src/oracle/test-resolver/index.ts +44 -0
  131. package/src/oracle/test-resolver/resolver.ts +198 -0
  132. package/src/oracle/txe_oracle_public_context.ts +60 -0
  133. package/src/oracle/txe_oracle_registry.ts +494 -0
  134. package/src/oracle/txe_oracle_top_level_context.ts +245 -81
  135. package/src/oracle/txe_oracle_version.ts +17 -0
  136. package/src/oracle/txe_private_execution_oracle.ts +30 -0
  137. package/src/rpc_server.ts +87 -0
  138. package/src/rpc_translator.ts +845 -905
  139. package/src/state_machine/archiver.ts +37 -16
  140. package/src/state_machine/dummy_p2p_client.ts +40 -11
  141. package/src/state_machine/global_variable_builder.ts +12 -11
  142. package/src/state_machine/index.ts +44 -30
  143. package/src/state_machine/mock_epoch_cache.ts +0 -4
  144. package/src/state_machine/synchronizer.ts +4 -3
  145. package/src/txe_session.ts +496 -80
  146. package/src/utils/block_creation.ts +2 -1
  147. package/src/{util → utils}/encoding.ts +8 -2
  148. package/src/utils/txe_artifact_resolver.ts +217 -0
  149. package/src/{util → utils}/txe_public_contract_data_source.ts +0 -2
  150. package/src/worker.ts +98 -0
  151. package/dest/util/encoding.d.ts +0 -720
  152. package/dest/util/encoding.d.ts.map +0 -1
  153. package/dest/util/expected_failure_error.d.ts.map +0 -1
  154. package/dest/util/txe_account_store.d.ts.map +0 -1
  155. package/dest/util/txe_public_contract_data_source.d.ts +0 -20
  156. package/dest/util/txe_public_contract_data_source.d.ts.map +0 -1
  157. /package/dest/{util → utils}/expected_failure_error.js +0 -0
  158. /package/dest/{util → utils}/txe_account_store.js +0 -0
  159. /package/src/{util → utils}/expected_failure_error.ts +0 -0
  160. /package/src/{util → utils}/txe_account_store.ts +0 -0
@@ -0,0 +1,226 @@
1
+ import { AztecAddress } from '@aztec/aztec.js/addresses';
2
+ import { BlockNumber } from '@aztec/foundation/branded-types';
3
+ import { Fr } from '@aztec/foundation/curves/bn254';
4
+ import { EthAddress } from '@aztec/foundation/eth-address';
5
+ import {
6
+ AZTEC_ADDRESS,
7
+ BIGINT,
8
+ BLOCK_HASH,
9
+ BLOCK_NUMBER,
10
+ BOOL,
11
+ BYTE,
12
+ BoundedVec,
13
+ ETH_ADDRESS,
14
+ FIELD,
15
+ FUNCTION_SELECTOR,
16
+ NOTE_SELECTOR,
17
+ Option,
18
+ type OracleRegistryEntry,
19
+ type TypeMapping,
20
+ U32,
21
+ } from '@aztec/pxe/simulator';
22
+ import { FunctionSelector, NoteSelector } from '@aztec/stdlib/abi';
23
+ import { BlockHash } from '@aztec/stdlib/block';
24
+
25
+ import type { OracleTestScenario } from './resolver.js';
26
+
27
+ /**
28
+ * Synthesizes fixture scenarios for every oracle in the registry whose parameters and return are built from scalars,
29
+ * arrays/bounded vecs of them, or `Option`s of them. The matching Noir serialization tests are auto-generated by
30
+ * `#[auto_serialization_test]` from the same seed/scenario convention.
31
+ */
32
+ export function synthesizeDefaultFixtures(
33
+ registry: Record<string, OracleRegistryEntry>,
34
+ ): Record<string, OracleTestScenario[]> {
35
+ const synthesized: Record<string, OracleTestScenario[]> = {};
36
+
37
+ for (const [name, entry] of Object.entries(registry)) {
38
+ const scenarios = synthesizeScenarios(entry);
39
+ if (scenarios !== undefined) {
40
+ synthesized[name] = scenarios;
41
+ }
42
+ }
43
+
44
+ return synthesized;
45
+ }
46
+
47
+ /**
48
+ * The test-value implementations, one per oracle type, mirroring the Noir `OracleTestValue` trait exactly
49
+ * (`noir-projects/aztec-nr/.../macros/oracle_testing.nr`).
50
+ */
51
+ const TEST_VALUE_IMPLS: TestValueImpl[] = [
52
+ scalar(FIELD, seed => new Fr(seed)),
53
+ scalar(U32, seed => seed),
54
+ scalar(BLOCK_NUMBER, seed => BlockNumber(seed)),
55
+ scalar(BIGINT, seed => BigInt(seed)),
56
+ scalar(BYTE, seed => seed),
57
+ scalar(BOOL, seed => seed % 2 !== 0),
58
+ scalar(AZTEC_ADDRESS, seed => AztecAddress.fromNumberUnsafe(seed)),
59
+ scalar(ETH_ADDRESS, seed => EthAddress.fromField(new Fr(seed))),
60
+ scalar(FUNCTION_SELECTOR, seed => FunctionSelector.fromField(new Fr(seed))),
61
+ scalar(NOTE_SELECTOR, seed => NoteSelector.fromField(new Fr(seed))),
62
+ scalar(BLOCK_HASH, seed => new BlockHash(new Fr(seed))),
63
+ composite(isOption, (type, seed) => [
64
+ named(Option.some(firstValue(type.inner, seed)), 'some'),
65
+ named(Option.none(firstValue(type.inner, seed)), 'none'),
66
+ ]),
67
+ composite(isArray, (type, seed) => [unnamed(collectionData(type.inner, seed, DEFAULT_ARRAY_LENGTH))]),
68
+ composite(isBoundedVec, (type, seed) => [
69
+ unnamed(
70
+ BoundedVec.from({
71
+ data: collectionData(type.inner, seed, DEFAULT_ARRAY_LENGTH),
72
+ maxLength: BOUNDED_VEC_MAX_LENGTH,
73
+ }),
74
+ ),
75
+ ]),
76
+ ];
77
+
78
+ /**
79
+ * The base every seed is offset by, applied once where a parameter's position becomes its seed, so generated values
80
+ * stay clear of 0. Must match `SEED_BASE` in the Noir `oracle_testing.nr`.
81
+ */
82
+ const SEED_BASE = 10;
83
+
84
+ /** Elements per synthesized array and bounded vec. Must match `TEST_COLLECTION_LEN` in the Noir `oracle_testing.nr`. */
85
+ const DEFAULT_ARRAY_LENGTH = 3;
86
+
87
+ /**
88
+ * Capacity of synthesized bounded vecs, deliberately larger than the data length so padding is exercised. Must match
89
+ * `TEST_BOUNDED_VEC_CAP` in the Noir `oracle_testing.nr`.
90
+ */
91
+ const BOUNDED_VEC_MAX_LENGTH = 5;
92
+
93
+ /**
94
+ * One canonical test case for a type: the `value` to put on the wire (or expect back), and the `name` of the
95
+ * serialization shape it exercises (unset for single-scenario types). Mirrors the Noir `Scenario<T>`.
96
+ */
97
+ interface Scenario {
98
+ value: unknown;
99
+ name?: string;
100
+ }
101
+
102
+ function unnamed(value: unknown): Scenario {
103
+ return { value };
104
+ }
105
+ function named(value: unknown, name: string): Scenario {
106
+ return { value, name };
107
+ }
108
+
109
+ /**
110
+ * One type's test-value implementation, mirroring a Noir `impl OracleTestValue<N> for T`: `match` selects the type and
111
+ * `scenarios` yields its [`Scenario`]s for `seed`, one per serialization shape, like Noir's
112
+ * `oracle_test_value(seed) -> [Scenario<Self>; N]`.
113
+ */
114
+ interface TestValueImpl {
115
+ match: (type: TypeMapping<any>) => boolean;
116
+ scenarios: (type: TypeMapping<any>, seed: number) => Scenario[];
117
+ }
118
+
119
+ /** A scalar impl: matches the singleton `type` by identity and yields a single unnamed scenario. */
120
+ function scalar(type: TypeMapping<any>, value: (seed: number) => unknown): TestValueImpl {
121
+ return { match: t => t === type, scenarios: (_type, seed) => [unnamed(value(seed))] };
122
+ }
123
+
124
+ /** A composite impl: selected by the `kind` type guard, recursing through its element/inner mapping. */
125
+ function composite<K extends CompositeMapping>(
126
+ guard: (type: TypeMapping<any>) => type is K,
127
+ scenarios: (type: K, seed: number) => Scenario[],
128
+ ): TestValueImpl {
129
+ return { match: guard, scenarios: (type, seed) => scenarios(type as K, seed) };
130
+ }
131
+
132
+ /** Synthesized element values for a collection. */
133
+ function collectionData(element: TypeMapping<any>, seed: number, length: number): unknown[] {
134
+ return Array.from({ length }, (_, i) => firstValue(element, seed + i));
135
+ }
136
+
137
+ /**
138
+ * The [`Scenario`]s for `type` at `seed`, one per serialization *shape*, mirroring the Noir
139
+ * `OracleTestValue::oracle_test_value(seed) -> [Scenario<Self>; N]`. Throws if the type has no matching impl.
140
+ */
141
+ function scenariosForType(type: TypeMapping<any>, seed: number): Scenario[] {
142
+ const impl = TEST_VALUE_IMPLS.find(i => i.match(type));
143
+ if (!impl) {
144
+ throw new UnsynthesizableTypeError(type);
145
+ }
146
+ return impl.scenarios(type, seed);
147
+ }
148
+
149
+ /** The first (representative) test value of `type`, used for collection elements and `Option` inners. */
150
+ function firstValue(type: TypeMapping<any>, seed: number): unknown {
151
+ return scenariosForType(type, seed)[0].value;
152
+ }
153
+
154
+ class UnsynthesizableTypeError extends Error {
155
+ constructor(type: TypeMapping<any>) {
156
+ super(`No test-value impl for type: ${JSON.stringify(Object.keys(type))}`);
157
+ this.name = 'UnsynthesizableTypeError';
158
+ }
159
+ }
160
+
161
+ /**
162
+ * Synthesizes the scenarios for one oracle by zipping every position's [`Scenario`] list to one case per test (a
163
+ * shorter list wraps around via mod), mirroring the Noir macro. Returns `undefined` if any param or return type is
164
+ * unsynthesizable (structs, ephemeral arrays).
165
+ */
166
+ function synthesizeScenarios(entry: OracleRegistryEntry): OracleTestScenario[] | undefined {
167
+ try {
168
+ return synthesizeScenariosOrThrow(entry);
169
+ } catch (e) {
170
+ if (e instanceof UnsynthesizableTypeError) {
171
+ return undefined;
172
+ }
173
+ throw e;
174
+ }
175
+ }
176
+
177
+ function synthesizeScenariosOrThrow(entry: OracleRegistryEntry): OracleTestScenario[] {
178
+ const params = entry.params.map((param, index) => ({
179
+ name: param.name,
180
+ scenarios: scenariosForType(param.type, SEED_BASE + index),
181
+ }));
182
+ // The return's seed is the slot right after the last parameter, so its value is distinct from every parameter.
183
+ const returnScenarios =
184
+ entry.returnType === undefined ? undefined : scenariosForType(entry.returnType, SEED_BASE + entry.params.length);
185
+
186
+ // Positions with more than one scenario drive the test count; their names are zipped together (joined with `+`) per
187
+ // test, matching the announcements the Noir tests emit and the resolver joins.
188
+ const positions = [...params.map(p => p.scenarios), ...(returnScenarios !== undefined ? [returnScenarios] : [])];
189
+ const namedPositions = positions.filter(s => s.length > 1);
190
+ const testCount = Math.max(1, ...positions.map(s => s.length));
191
+
192
+ const scenarios: OracleTestScenario[] = [];
193
+ for (let index = 0; index < testCount; index++) {
194
+ const inputs: Record<string, unknown> = {};
195
+ for (const { name, scenarios: cases } of params) {
196
+ inputs[name] = cases[index % cases.length].value;
197
+ }
198
+ const output = returnScenarios === undefined ? undefined : returnScenarios[index % returnScenarios.length].value;
199
+ const scenario =
200
+ namedPositions.length === 0
201
+ ? undefined
202
+ : namedPositions.map(cases => cases[index % cases.length].name!).join('+');
203
+ scenarios.push({ scenario, inputs, output });
204
+ }
205
+ return scenarios;
206
+ }
207
+
208
+ /**
209
+ * The composite `TypeMapping`s (`ARRAY`/`BOUNDED_VEC`/`OPTION`), discriminated by `kind`. The combinators attach `kind`
210
+ * plus the inner mapping at construction; the registry erases params to the base `TypeMapping`, so the guards below
211
+ * recover the structure for recursion.
212
+ */
213
+ type ArrayMapping = TypeMapping & { kind: 'array'; inner: TypeMapping };
214
+ type BoundedVecMapping = TypeMapping & { kind: 'bounded-vec'; inner: TypeMapping };
215
+ type OptionMapping = TypeMapping & { kind: 'option'; inner: TypeMapping };
216
+ type CompositeMapping = ArrayMapping | BoundedVecMapping | OptionMapping;
217
+
218
+ function isArray(type: TypeMapping<any>): type is ArrayMapping {
219
+ return 'kind' in type && type.kind === 'array';
220
+ }
221
+ function isBoundedVec(type: TypeMapping<any>): type is BoundedVecMapping {
222
+ return 'kind' in type && type.kind === 'bounded-vec';
223
+ }
224
+ function isOption(type: TypeMapping<any>): type is OptionMapping {
225
+ return 'kind' in type && type.kind === 'option';
226
+ }
@@ -0,0 +1,44 @@
1
+ /* eslint-disable camelcase */
2
+ import { createSafeJsonRpcServer } from '@aztec/foundation/json-rpc/server';
3
+ import type { Logger } from '@aztec/foundation/log';
4
+ import type { ApiSchemaFor } from '@aztec/foundation/schemas';
5
+ import { zodFor } from '@aztec/stdlib/schemas';
6
+
7
+ import { z } from 'zod';
8
+
9
+ import { ForeignCallArgsSchema, type ForeignCallResult, ForeignCallResultSchema } from '../../utils/encoding.js';
10
+ import { TXE_ORACLE_REGISTRY } from '../txe_oracle_registry.js';
11
+ import { type OracleTestCallInput, OracleTestResolver } from './resolver.js';
12
+
13
+ const OracleTestCallInputSchema = zodFor<OracleTestCallInput>()(
14
+ z.object({
15
+ session_id: z.number().nonnegative(),
16
+ function: z.string(),
17
+ root_path: z.string(),
18
+ package_name: z.string(),
19
+ inputs: ForeignCallArgsSchema,
20
+ }),
21
+ );
22
+
23
+ /** Narrowed interface exposing only the RPC method. */
24
+ interface OracleTestRpcHandler {
25
+ resolve_foreign_call(callData: OracleTestCallInput): Promise<ForeignCallResult>;
26
+ }
27
+
28
+ const OracleTestRpcHandlerSchema: ApiSchemaFor<OracleTestRpcHandler> = {
29
+ resolve_foreign_call: z.function({
30
+ input: z.tuple([OracleTestCallInputSchema]),
31
+ output: ForeignCallResultSchema,
32
+ }),
33
+ };
34
+
35
+ /** Creates an RPC server backed by the oracle test resolver. */
36
+ export function createOracleTestRpcServer(logger: Logger) {
37
+ const resolver = OracleTestResolver.fromRegistry(TXE_ORACLE_REGISTRY, logger);
38
+ const server = createSafeJsonRpcServer<OracleTestRpcHandler>(resolver, OracleTestRpcHandlerSchema, {
39
+ http200OnError: true,
40
+ });
41
+ return { server, resolver };
42
+ }
43
+
44
+ export { OracleTestResolver } from './resolver.js';
@@ -0,0 +1,198 @@
1
+ /* eslint-disable camelcase */
2
+ import type { Logger } from '@aztec/foundation/log';
3
+ import { createLogger } from '@aztec/foundation/log';
4
+ import { withoutHexPrefix } from '@aztec/foundation/string';
5
+ import { BOUNDED_VEC, BYTE, BoundedVec, Option, type OracleRegistryEntry, makeEntry } from '@aztec/pxe/simulator';
6
+
7
+ import type { ForeignCallArgs, ForeignCallResult } from '../../utils/encoding.js';
8
+ import { toInputSlots } from '../txe_oracle_registry.js';
9
+ import { synthesizeDefaultFixtures } from './default_fixtures.js';
10
+
11
+ /** Name of the meta-oracle that Noir tests call to announce the next call's scenario by name. */
12
+ const SET_SCENARIO_ORACLE = 'aztec_oracle_test_set_scenario';
13
+ export const SET_SCENARIO_ENTRY = makeEntry({ params: [{ name: 'name', type: BOUNDED_VEC(BYTE) }] });
14
+
15
+ export type OracleTestCallInput = {
16
+ session_id: number;
17
+ function: string;
18
+ root_path: string;
19
+ package_name: string;
20
+ inputs: ForeignCallArgs;
21
+ };
22
+
23
+ /**
24
+ * One test case for an oracle: the expected deserialized `inputs`, the `output` to serialize back to Noir, and the
25
+ * `scenario` name (`some`/`none`) the matching Noir test announces. Single-scenario oracles leave `scenario` undefined
26
+ * and are selected automatically.
27
+ */
28
+ export interface OracleTestScenario {
29
+ scenario?: string;
30
+ inputs: Record<string, unknown>;
31
+ output: unknown;
32
+ }
33
+
34
+ /**
35
+ * Oracle resolver for serialization tests. Selects a fixture scenario, verifies that the received inputs match, and
36
+ * returns the corresponding fixture output.
37
+ *
38
+ * For oracles with a single scenario, it is selected automatically. For oracles with multiple scenarios, the Noir test
39
+ * announces the scenario by name via the `aztec_oracle_test_set_scenario` meta-oracle before calling the real oracle.
40
+ */
41
+ export class OracleTestResolver {
42
+ private readonly calledOracles = new Set<string>();
43
+ private readonly pendingScenario = new Map<number, string>();
44
+ private readonly logger: Logger;
45
+
46
+ constructor(
47
+ private readonly registry: Record<string, OracleRegistryEntry>,
48
+ private readonly fixtures: Record<string, OracleTestScenario[]>,
49
+ logger?: Logger,
50
+ ) {
51
+ this.logger = logger ?? createLogger('txe:test-resolver');
52
+ }
53
+
54
+ /**
55
+ * Builds a resolver whose fixtures are the default scenarios synthesized from the registry's type mappings. The
56
+ * matching Noir serialization tests are auto-generated by `#[auto_serialization_test]` from the same convention.
57
+ */
58
+ static fromRegistry(registry: Record<string, OracleRegistryEntry>, logger?: Logger): OracleTestResolver {
59
+ return new OracleTestResolver(registry, synthesizeDefaultFixtures(registry), logger);
60
+ }
61
+
62
+ // eslint-disable-next-line require-await
63
+ async resolve_foreign_call(callData: OracleTestCallInput): Promise<ForeignCallResult> {
64
+ const oracleName = callData.function;
65
+ this.logger.debug('Resolving oracle', { oracleName });
66
+
67
+ if (oracleName === SET_SCENARIO_ORACLE) {
68
+ return this.#handleSetScenario(callData);
69
+ }
70
+
71
+ if (!(oracleName in this.registry)) {
72
+ throw new Error(`Oracle '${oracleName}' not found in registry`);
73
+ }
74
+ const entry = this.registry[oracleName];
75
+
76
+ const scenarios = this.fixtures[oracleName];
77
+ if (!scenarios || scenarios.length === 0) {
78
+ throw new Error(`No fixture defined for oracle '${oracleName}'`);
79
+ }
80
+
81
+ const match = this.#selectScenario(callData.session_id, oracleName, scenarios);
82
+ this.#verifyInputs(callData.inputs, entry, match, oracleName);
83
+ this.calledOracles.add(oracleName);
84
+
85
+ this.logger.debug('Verified scenario for oracle', { oracleName });
86
+
87
+ const outputSlots = entry.serializeReturn(match.output);
88
+ return {
89
+ values: outputSlots.map(slot => (Array.isArray(slot) ? slot.map(withoutHexPrefix) : withoutHexPrefix(slot))),
90
+ };
91
+ }
92
+
93
+ /** Returns oracles that have a fixture defined but were never called during testing. */
94
+ getUncalledFixtures(): string[] {
95
+ return Object.keys(this.fixtures).filter(name => !this.calledOracles.has(name));
96
+ }
97
+
98
+ /** Returns oracles in the registry that have no fixture defined at all. */
99
+ getMissingFixtures(): string[] {
100
+ return Object.keys(this.registry).filter(name => !(name in this.fixtures));
101
+ }
102
+
103
+ /**
104
+ * Records the scenario name for the session's next oracle call. A test announces once per multi-scenario position
105
+ * (param or return), so successive announcements accumulate joined with `+`, matching the names the fixture
106
+ * synthesizer builds (e.g. `some+some`).
107
+ */
108
+ #handleSetScenario(callData: { session_id: number; inputs: ForeignCallArgs }): ForeignCallResult {
109
+ const name = decodeScenarioName(callData.inputs);
110
+ const existing = this.pendingScenario.get(callData.session_id);
111
+ const scenario = existing !== undefined ? `${existing}+${name}` : name;
112
+
113
+ this.logger.debug('Setting scenario for next oracle call', { sessionId: callData.session_id, scenario });
114
+ this.pendingScenario.set(callData.session_id, scenario);
115
+
116
+ return { values: [] };
117
+ }
118
+
119
+ /** Selects the fixture by the announced scenario name (for multi-scenario oracles) or returns the single fixture. */
120
+ #selectScenario(sessionId: number, oracleName: string, fixtures: OracleTestScenario[]): OracleTestScenario {
121
+ const announced = this.pendingScenario.get(sessionId);
122
+
123
+ if (announced !== undefined) {
124
+ this.pendingScenario.delete(sessionId);
125
+ const match = fixtures.find(s => s.scenario === announced);
126
+ if (!match) {
127
+ const available = fixtures.map(s => s.scenario ?? '<default>').join(', ');
128
+ throw new Error(`No scenario '${announced}' for oracle '${oracleName}'. Available: ${available}`);
129
+ }
130
+ return match;
131
+ }
132
+
133
+ if (fixtures.length === 1) {
134
+ return fixtures[0];
135
+ }
136
+
137
+ throw new Error(`Oracle '${oracleName}' has ${fixtures.length} scenarios but none was announced before the call.`);
138
+ }
139
+
140
+ /** Deserializes actual inputs and verifies they match the scenario's expected inputs. */
141
+ #verifyInputs(
142
+ rawInputs: ForeignCallArgs,
143
+ entry: OracleRegistryEntry,
144
+ fixture: OracleTestScenario,
145
+ oracleName: string,
146
+ ): void {
147
+ const expectedInputs = fixture.inputs;
148
+ if (Object.keys(expectedInputs).length === 0) {
149
+ return;
150
+ }
151
+
152
+ const scenarioLabel = fixture.scenario ? ` (scenario '${fixture.scenario}')` : '';
153
+ const named = entry.deserializeParams(toInputSlots(rawInputs));
154
+
155
+ for (const param of named) {
156
+ if (!(param.name in expectedInputs)) {
157
+ throw new Error(
158
+ `Unexpected param '${param.name}' for oracle '${oracleName}'. ` +
159
+ `Expected: ${Object.keys(expectedInputs).join(', ')}`,
160
+ );
161
+ }
162
+ if (!valuesEqual(param.value, expectedInputs[param.name])) {
163
+ throw new Error(
164
+ `Input mismatch for oracle '${oracleName}'${scenarioLabel}: ` +
165
+ `param '${param.name}' expected ${String(expectedInputs[param.name])} ` +
166
+ `but got ${String(param.value)}. ${versionBumpHint(oracleName)}`,
167
+ );
168
+ }
169
+ }
170
+ }
171
+ }
172
+
173
+ /**
174
+ * Decodes the scenario name announced by `set_oracle_test_scenario`, whose argument is a `BoundedVec<u8, N>`
175
+ */
176
+ function decodeScenarioName(inputs: ForeignCallArgs): string {
177
+ const [{ value }] = SET_SCENARIO_ENTRY.deserializeParams(toInputSlots(inputs));
178
+ return String.fromCharCode(...value.data);
179
+ }
180
+
181
+ function valuesEqual(actual: unknown, expected: unknown): boolean {
182
+ if (actual instanceof Option && expected instanceof Option) {
183
+ return actual.equals(expected, valuesEqual);
184
+ }
185
+ if (actual instanceof BoundedVec && expected instanceof BoundedVec) {
186
+ return actual.equals(expected, valuesEqual);
187
+ }
188
+ if (Array.isArray(actual) && Array.isArray(expected)) {
189
+ return actual.length === expected.length && actual.every((v, i) => valuesEqual(v, expected[i]));
190
+ }
191
+ return String(actual) === String(expected);
192
+ }
193
+
194
+ function versionBumpHint(oracle: string): string {
195
+ return oracle.startsWith('aztec_txe_') || oracle.startsWith('aztec_avm_')
196
+ ? 'If you changed this oracle, consider bumping the TXE oracle version in yarn-project/txe/src/oracle/txe_oracle_version.ts.'
197
+ : 'If you changed this oracle, consider bumping the PXE oracle version in yarn-project/pxe/src/oracle_version.ts.';
198
+ }
@@ -1,9 +1,11 @@
1
1
  import { BlockNumber } from '@aztec/foundation/branded-types';
2
2
  import { Fr } from '@aztec/foundation/curves/bn254';
3
3
  import { type Logger, createLogger } from '@aztec/foundation/log';
4
+ import type { ContractStore } from '@aztec/pxe/server';
4
5
  import { PublicDataWrite } from '@aztec/stdlib/avm';
5
6
  import { AztecAddress } from '@aztec/stdlib/aztec-address';
6
7
  import type { L2Block } from '@aztec/stdlib/block';
8
+ import type { ContractInstanceWithAddress } from '@aztec/stdlib/contract';
7
9
  import { computePublicDataTreeLeafSlot, siloNoteHash, siloNullifier } from '@aztec/stdlib/hash';
8
10
  import {
9
11
  MerkleTreeId,
@@ -29,6 +31,7 @@ export class TXEOraclePublicContext implements IAvmExecutionOracle {
29
31
  private forkedWorldTrees: MerkleTreeWriteOperations,
30
32
  private txRequestHash: Fr,
31
33
  private globalVariables: GlobalVariables,
34
+ private contractStore: ContractStore,
32
35
  ) {
33
36
  this.logger = createLogger('txe:public_context');
34
37
 
@@ -122,6 +125,63 @@ export class TXEOraclePublicContext implements IAvmExecutionOracle {
122
125
  return value;
123
126
  }
124
127
 
128
+ getContractInstanceDeployer(address: AztecAddress): Promise<{ member: Fr; exists: boolean }> {
129
+ return this.getContractInstanceMember(address, i => i.deployer.toField());
130
+ }
131
+
132
+ getContractInstanceClassId(address: AztecAddress): Promise<{ member: Fr; exists: boolean }> {
133
+ return this.getContractInstanceMember(address, i => i.currentContractClassId);
134
+ }
135
+
136
+ getContractInstanceInitializationHash(address: AztecAddress): Promise<{ member: Fr; exists: boolean }> {
137
+ return this.getContractInstanceMember(address, i => i.initializationHash);
138
+ }
139
+
140
+ getContractInstanceImmutablesHash(address: AztecAddress): Promise<{ member: Fr; exists: boolean }> {
141
+ return this.getContractInstanceMember(address, i => i.immutablesHash);
142
+ }
143
+
144
+ private async getContractInstanceMember(
145
+ address: AztecAddress,
146
+ accessor: (instance: ContractInstanceWithAddress) => Fr,
147
+ ): Promise<{ member: Fr; exists: boolean }> {
148
+ const instance = await this.contractStore.getContractInstance(address);
149
+ if (!instance) {
150
+ return { member: Fr.ZERO, exists: false };
151
+ }
152
+ return { member: accessor(instance), exists: true };
153
+ }
154
+
155
+ returndataSize(): Promise<Fr> {
156
+ throw new Error(
157
+ 'Contract calls are forbidden inside a `TestEnvironment::public_context`, use `public_call` instead',
158
+ );
159
+ }
160
+
161
+ returndataCopy(_rdOffset: number, _copySize: number): Promise<Fr[]> {
162
+ throw new Error(
163
+ 'Contract calls are forbidden inside a `TestEnvironment::public_context`, use `public_call` instead',
164
+ );
165
+ }
166
+
167
+ call(_l2Gas: number, _daGas: number, _address: AztecAddress, _argsLength: number, _args: Fr[]): Promise<void> {
168
+ throw new Error(
169
+ 'Contract calls are forbidden inside a `TestEnvironment::public_context`, use `public_call` instead',
170
+ );
171
+ }
172
+
173
+ staticCall(_l2Gas: number, _daGas: number, _address: AztecAddress, _argsLength: number, _args: Fr[]): Promise<void> {
174
+ throw new Error(
175
+ 'Contract calls are forbidden inside a `TestEnvironment::public_context`, use `public_call` instead',
176
+ );
177
+ }
178
+
179
+ successCopy(): Promise<boolean> {
180
+ throw new Error(
181
+ 'Contract calls are forbidden inside a `TestEnvironment::public_context`, use `public_call` instead',
182
+ );
183
+ }
184
+
125
185
  async close(): Promise<L2Block> {
126
186
  this.logger.debug('Exiting Public Context, building block with collected side effects', {
127
187
  blockNumber: this.globalVariables.blockNumber,