321 lines
9.1 KiB
TypeScript
321 lines
9.1 KiB
TypeScript
/* eslint-disable no-unused-vars */
|
|
import 'mocha';
|
|
|
|
import * as sbv2 from '../src';
|
|
import { setupTest, TestContext } from './utilts';
|
|
import { Keypair } from '@solana/web3.js';
|
|
import {
|
|
AggregatorAccount,
|
|
JobAccount,
|
|
LeaseAccount,
|
|
PermissionAccount,
|
|
QueueAccount,
|
|
} from '../src';
|
|
import { OracleJob } from '@switchboard-xyz/common';
|
|
import { assert } from 'console';
|
|
import { PermitOracleQueueUsage } from '../src/generated/types/SwitchboardPermission';
|
|
|
|
describe('Aggregator Tests', () => {
|
|
let ctx: TestContext;
|
|
|
|
const queueAuthority = Keypair.generate();
|
|
let queueAccount: QueueAccount;
|
|
|
|
let jobAccount: JobAccount;
|
|
|
|
let fundedAggregator: AggregatorAccount;
|
|
|
|
before(async () => {
|
|
ctx = await setupTest();
|
|
|
|
const [oracleQueue] = await sbv2.QueueAccount.create(ctx.program, {
|
|
name: 'aggregator-queue',
|
|
metadata: '',
|
|
authority: queueAuthority.publicKey,
|
|
queueSize: 1,
|
|
reward: 0,
|
|
minStake: 0,
|
|
oracleTimeout: 86400,
|
|
slashingEnabled: false,
|
|
unpermissionedFeeds: true,
|
|
unpermissionedVrf: true,
|
|
enableBufferRelayers: false,
|
|
});
|
|
|
|
queueAccount = oracleQueue;
|
|
|
|
await ctx.program.mint.getOrCreateAssociatedUser(ctx.program.walletPubkey);
|
|
|
|
// add a single oracle for open round calls
|
|
await queueAccount.createOracle({
|
|
name: 'oracle-1',
|
|
});
|
|
|
|
const [jobAccount1] = await JobAccount.create(ctx.program, {
|
|
data: OracleJob.encodeDelimited(
|
|
OracleJob.fromObject({
|
|
tasks: [
|
|
{
|
|
valueTask: {
|
|
value: 1,
|
|
},
|
|
},
|
|
],
|
|
})
|
|
).finish(),
|
|
name: 'Job1',
|
|
});
|
|
jobAccount = jobAccount1;
|
|
});
|
|
|
|
it("Adds job, updates it's weight, then removes it from aggregator", async () => {
|
|
const aggregatorKeypair = Keypair.generate();
|
|
const aggregatorAuthority = Keypair.generate();
|
|
|
|
const [aggregatorAccount] = await AggregatorAccount.create(ctx.program, {
|
|
queueAccount,
|
|
queueAuthority: queueAuthority.publicKey,
|
|
authority: aggregatorAuthority.publicKey,
|
|
batchSize: 1,
|
|
minRequiredOracleResults: 1,
|
|
minRequiredJobResults: 1,
|
|
minUpdateDelaySeconds: 60,
|
|
keypair: aggregatorKeypair,
|
|
});
|
|
await aggregatorAccount.loadData();
|
|
|
|
const oracleJob = OracleJob.fromObject({
|
|
tasks: [
|
|
{
|
|
valueTask: {
|
|
value: 1,
|
|
},
|
|
},
|
|
],
|
|
});
|
|
|
|
const [jobAccount] = await JobAccount.create(ctx.program, {
|
|
data: OracleJob.encodeDelimited(oracleJob).finish(),
|
|
name: 'Job1',
|
|
});
|
|
|
|
await aggregatorAccount.addJob({
|
|
job: jobAccount,
|
|
weight: 1,
|
|
authority: aggregatorAuthority,
|
|
});
|
|
|
|
const postAddJobAggregatorState = await aggregatorAccount.loadData();
|
|
const jobIdx = postAddJobAggregatorState.jobPubkeysData.findIndex(pubkey =>
|
|
pubkey.equals(jobAccount.publicKey)
|
|
);
|
|
if (jobIdx === -1) {
|
|
throw new Error(`Failed to add job to aggregator`);
|
|
}
|
|
|
|
await aggregatorAccount.updateJobWeight({
|
|
job: jobAccount,
|
|
jobIdx: jobIdx,
|
|
weight: 2,
|
|
authority: aggregatorAuthority,
|
|
});
|
|
const postUpdateWeightAggregatorState = await aggregatorAccount.loadData();
|
|
if (postUpdateWeightAggregatorState.jobWeights[0] !== 2) {
|
|
throw new Error(`Failed to update job weight in aggregator`);
|
|
}
|
|
|
|
await aggregatorAccount.removeJob({
|
|
job: jobAccount,
|
|
jobIdx: jobIdx,
|
|
authority: aggregatorAuthority,
|
|
});
|
|
const postRemoveJobAggregatorState = await aggregatorAccount.loadData();
|
|
const jobIdx1 = postRemoveJobAggregatorState.jobPubkeysData.findIndex(
|
|
pubkey => pubkey.equals(jobAccount.publicKey)
|
|
);
|
|
if (jobIdx1 !== -1) {
|
|
throw new Error(`Failed to remove job from aggregator`);
|
|
}
|
|
});
|
|
|
|
it('Creates and funds aggregator', async () => {
|
|
const [aggregatorAccount] = await queueAccount.createFeed({
|
|
queueAuthority: queueAuthority,
|
|
batchSize: 1,
|
|
minRequiredOracleResults: 1,
|
|
minRequiredJobResults: 1,
|
|
minUpdateDelaySeconds: 60,
|
|
fundAmount: 2.5,
|
|
enable: true,
|
|
jobs: [
|
|
{ pubkey: jobAccount.publicKey },
|
|
{
|
|
weight: 2,
|
|
data: OracleJob.encodeDelimited(
|
|
OracleJob.fromObject({
|
|
tasks: [
|
|
{
|
|
valueTask: {
|
|
value: 1,
|
|
},
|
|
},
|
|
],
|
|
})
|
|
).finish(),
|
|
},
|
|
],
|
|
});
|
|
|
|
const aggregator = await aggregatorAccount.loadData();
|
|
|
|
if (aggregator.jobPubkeysSize !== 2) {
|
|
throw new Error(`Aggregator failed to add the correct number of jobs`);
|
|
}
|
|
|
|
if (aggregator.jobWeights[0] !== 1 || aggregator.jobWeights[1] !== 2) {
|
|
throw new Error(`Aggregator set the incorrect job weights`);
|
|
}
|
|
|
|
const [leaseAccount] = LeaseAccount.fromSeed(
|
|
ctx.program,
|
|
queueAccount.publicKey,
|
|
aggregatorAccount.publicKey
|
|
);
|
|
const leaseBalance = await leaseAccount.getBalance();
|
|
if (leaseBalance !== 2.5) {
|
|
throw new Error(
|
|
`Lease balance has incorrect funds, expected 2.5 wSOL, received ${leaseBalance}`
|
|
);
|
|
}
|
|
|
|
fundedAggregator = aggregatorAccount;
|
|
});
|
|
|
|
it('Withdraws funds from an aggregator lease', async () => {
|
|
if (!fundedAggregator) {
|
|
throw new Error(`Aggregator does not exist`);
|
|
}
|
|
const aggregatorAccount = fundedAggregator;
|
|
|
|
const userTokenAddress = ctx.program.mint.getAssociatedAddress(
|
|
ctx.payer.publicKey
|
|
);
|
|
|
|
let initialUserTokenBalance = await ctx.program.mint.getBalance(
|
|
userTokenAddress
|
|
);
|
|
|
|
if (initialUserTokenBalance === null || initialUserTokenBalance <= 0) {
|
|
const [user, userInit] =
|
|
await ctx.program.mint.getOrCreateWrappedUserInstructions(
|
|
ctx.payer.publicKey,
|
|
{ fundUpTo: 0.1 }
|
|
);
|
|
|
|
if (userInit && userInit.ixns.length > 0) {
|
|
await ctx.program.signAndSend(userInit);
|
|
}
|
|
|
|
initialUserTokenBalance =
|
|
(await ctx.program.mint.getBalance(userTokenAddress)) ?? 0;
|
|
}
|
|
|
|
const [leaseAccount] = LeaseAccount.fromSeed(
|
|
ctx.program,
|
|
queueAccount.publicKey,
|
|
aggregatorAccount.publicKey
|
|
);
|
|
const leaseBalance = await leaseAccount.getBalance();
|
|
|
|
const expectedFinalBalance = leaseBalance - 1;
|
|
|
|
await leaseAccount.withdraw({
|
|
amount: 1,
|
|
unwrap: true,
|
|
withdrawWallet: userTokenAddress,
|
|
});
|
|
|
|
const finalBalance = await leaseAccount.getBalance();
|
|
if (expectedFinalBalance !== finalBalance) {
|
|
throw new Error(
|
|
`Lease balance has incorrect funds, expected ${expectedFinalBalance} wSOL, received ${finalBalance}`
|
|
);
|
|
}
|
|
|
|
const finalUserBalance = await ctx.program.mint.getBalance(
|
|
userTokenAddress
|
|
);
|
|
if (!finalUserBalance) {
|
|
throw new Error(`Users wrapped account was closed`);
|
|
}
|
|
|
|
const finalUserTokenBalance =
|
|
(await ctx.program.mint.getBalance(userTokenAddress)) ?? 0;
|
|
if (initialUserTokenBalance !== finalUserTokenBalance) {
|
|
throw new Error(
|
|
`User token balance has incorrect funds, expected ${initialUserTokenBalance} wSOL, received ${finalUserTokenBalance}`
|
|
);
|
|
}
|
|
});
|
|
|
|
it("Adds job, updates it's config, then removes it from aggregator", async () => {
|
|
const aggregatorKeypair = Keypair.generate();
|
|
const aggregatorAuthority = Keypair.generate();
|
|
|
|
const [aggregatorAccount] = await AggregatorAccount.create(ctx.program, {
|
|
queueAccount,
|
|
queueAuthority: queueAuthority.publicKey,
|
|
authority: aggregatorAuthority.publicKey,
|
|
batchSize: 1,
|
|
minRequiredOracleResults: 1,
|
|
minRequiredJobResults: 1,
|
|
minUpdateDelaySeconds: 60,
|
|
keypair: aggregatorKeypair,
|
|
});
|
|
await aggregatorAccount.loadData();
|
|
|
|
const oracleJob = OracleJob.fromObject({
|
|
tasks: [{ valueTask: { value: 1 } }],
|
|
});
|
|
|
|
const [jobAccount] = await JobAccount.create(ctx.program, {
|
|
data: OracleJob.encodeDelimited(oracleJob).finish(),
|
|
name: 'Job1',
|
|
});
|
|
|
|
await aggregatorAccount.addJob({
|
|
job: jobAccount,
|
|
weight: 1,
|
|
authority: aggregatorAuthority,
|
|
});
|
|
|
|
const postAddJobAggregatorState = await aggregatorAccount.loadData();
|
|
const jobIdx = postAddJobAggregatorState.jobPubkeysData.findIndex(pubkey =>
|
|
pubkey.equals(jobAccount.publicKey)
|
|
);
|
|
if (jobIdx === -1) {
|
|
throw new Error(`Failed to add job to aggregator`);
|
|
}
|
|
|
|
const badSetConfigSignature = await aggregatorAccount
|
|
.setConfigInstruction(aggregatorAuthority.publicKey, { minJobResults: 2 })
|
|
.catch(() => undefined);
|
|
// If badSetConfigSignature isn't undefined, a (bad) transaction was built and sent.
|
|
if (badSetConfigSignature) {
|
|
throw new Error(
|
|
'Aggregator should not let minJobResults increase above numJobs'
|
|
);
|
|
}
|
|
|
|
await aggregatorAccount.setConfig({
|
|
authority: aggregatorAuthority,
|
|
minUpdateDelaySeconds: 300,
|
|
force: true, // Bypass validation rules.
|
|
});
|
|
const postUpdateAggregatorState = await aggregatorAccount.loadData();
|
|
if (postUpdateAggregatorState.minUpdateDelaySeconds !== 300) {
|
|
throw new Error(`Failed to setConfig on aggregator`);
|
|
}
|
|
});
|
|
});
|