sbv2-solana/javascript/solana.js/test/aggregator.spec.ts

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`);
}
});
});