-
Notifications
You must be signed in to change notification settings - Fork 76
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
191d4bc
commit c61e780
Showing
5 changed files
with
4,445 additions
and
126 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,12 @@ | ||
import type { Config } from '@jest/types'; | ||
|
||
const config: Config.InitialOptions = { | ||
preset: 'ts-jest', | ||
testEnvironment: 'node', | ||
testMatch: ['**/*.test.ts'], | ||
verbose: true, | ||
testTimeout: 60000, // 30 seconds | ||
maxWorkers: 1, // Run tests sequentially | ||
}; | ||
|
||
export default config; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,150 @@ | ||
import { createAnvil, Anvil } from "@viem/anvil"; | ||
import { describe, beforeAll, afterAll, it, expect } from '@jest/globals'; | ||
import { exec } from 'child_process'; | ||
import fs from 'fs/promises'; | ||
import path from 'path'; | ||
import util from 'util'; | ||
import { ethers } from "ethers"; | ||
import * as dotenv from "dotenv"; | ||
|
||
dotenv.config(); | ||
|
||
const execAsync = util.promisify(exec); | ||
|
||
async function loadJsonFile(filePath: string): Promise<any> { | ||
try { | ||
const content = await fs.readFile(filePath, 'utf-8'); | ||
return JSON.parse(content); | ||
} catch (error) { | ||
console.error(`Error loading file ${filePath}:`, error); | ||
return null; | ||
} | ||
} | ||
|
||
async function loadDeployments(): Promise<Record<string, any>> { | ||
const coreFilePath = path.join(__dirname, '..', 'contracts', 'deployments', 'core', '31337.json'); | ||
const helloWorldFilePath = path.join(__dirname, '..', 'contracts', 'deployments', 'hello-world', '31337.json'); | ||
|
||
const [coreDeployment, helloWorldDeployment] = await Promise.all([ | ||
loadJsonFile(coreFilePath), | ||
loadJsonFile(helloWorldFilePath) | ||
]); | ||
|
||
if (!coreDeployment || !helloWorldDeployment) { | ||
console.error('Error loading deployments'); | ||
return {}; | ||
} | ||
|
||
return { | ||
core: coreDeployment, | ||
helloWorld: helloWorldDeployment | ||
}; | ||
} | ||
|
||
describe('Operator Functionality', () => { | ||
let anvil: Anvil; | ||
let deployment: Record<string, any>; | ||
let provider: ethers.JsonRpcProvider; | ||
let signer: ethers.Wallet; | ||
let delegationManager: ethers.Contract; | ||
let helloWorldServiceManager: ethers.Contract; | ||
let ecdsaRegistryContract: ethers.Contract; | ||
let avsDirectory: ethers.Contract; | ||
|
||
beforeAll(async () => { | ||
anvil = createAnvil(); | ||
await anvil.start(); | ||
await execAsync('npm run deploy:core'); | ||
await execAsync('npm run deploy:hello-world'); | ||
deployment = await loadDeployments(); | ||
|
||
provider = new ethers.JsonRpcProvider(process.env.RPC_URL); | ||
signer = new ethers.Wallet(process.env.PRIVATE_KEY!, provider); | ||
|
||
const delegationManagerABI = await loadJsonFile(path.join(__dirname, '..', 'abis', 'IDelegationManager.json')); | ||
const ecdsaRegistryABI = await loadJsonFile(path.join(__dirname, '..', 'abis', 'ECDSAStakeRegistry.json')); | ||
const helloWorldServiceManagerABI = await loadJsonFile(path.join(__dirname, '..', 'abis', 'HelloWorldServiceManager.json')); | ||
const avsDirectoryABI = await loadJsonFile(path.join(__dirname, '..', 'abis', 'IAVSDirectory.json')); | ||
|
||
delegationManager = new ethers.Contract(deployment.core.addresses.delegation, delegationManagerABI, signer); | ||
helloWorldServiceManager = new ethers.Contract(deployment.helloWorld.addresses.helloWorldServiceManager, helloWorldServiceManagerABI, signer); | ||
ecdsaRegistryContract = new ethers.Contract(deployment.helloWorld.addresses.stakeRegistry, ecdsaRegistryABI, signer); | ||
avsDirectory = new ethers.Contract(deployment.core.addresses.avsDirectory, avsDirectoryABI, signer); | ||
}); | ||
|
||
it('should register as an operator', async () => { | ||
const tx = await delegationManager.registerAsOperator({ | ||
__deprecated_earningsReceiver: await signer.getAddress(), | ||
delegationApprover: "0x0000000000000000000000000000000000000000", | ||
stakerOptOutWindowBlocks: 0 | ||
}, ""); | ||
await tx.wait(); | ||
|
||
const isOperator = await delegationManager.isOperator(signer.address); | ||
expect(isOperator).toBe(true); | ||
}); | ||
|
||
it('should register operator to AVS', async () => { | ||
const salt = ethers.hexlify(ethers.randomBytes(32)); | ||
const expiry = Math.floor(Date.now() / 1000) + 3600; | ||
|
||
const operatorDigestHash = await avsDirectory.calculateOperatorAVSRegistrationDigestHash( | ||
signer.address, | ||
await helloWorldServiceManager.getAddress(), | ||
salt, | ||
expiry | ||
); | ||
|
||
const operatorSigningKey = new ethers.SigningKey(process.env.PRIVATE_KEY!); | ||
const operatorSignedDigestHash = operatorSigningKey.sign(operatorDigestHash); | ||
const operatorSignature = ethers.Signature.from(operatorSignedDigestHash).serialized; | ||
|
||
const tx = await ecdsaRegistryContract.registerOperatorWithSignature( | ||
{ | ||
signature: operatorSignature, | ||
salt: salt, | ||
expiry: expiry | ||
}, | ||
signer.address | ||
); | ||
await tx.wait(); | ||
|
||
const isRegistered = await ecdsaRegistryContract.operatorRegistered(signer.address); | ||
expect(isRegistered).toBe(true); | ||
}); | ||
|
||
it('should create a new task', async () => { | ||
const taskName = "Steven"; | ||
|
||
const tx = await helloWorldServiceManager.createNewTask(taskName); | ||
await tx.wait(); | ||
}); | ||
|
||
it('should sign and respond to a task', async () => { | ||
const taskIndex = 0; | ||
const taskCreatedBlock = await provider.getBlockNumber(); | ||
const taskName = "Steven"; | ||
const message = `Hello, ${taskName}`; | ||
const messageHash = ethers.solidityPackedKeccak256(["string"], [message]); | ||
const messageBytes = ethers.getBytes(messageHash); | ||
const signature = await signer.signMessage(messageBytes); | ||
|
||
const operators = [await signer.getAddress()]; | ||
const signatures = [signature]; | ||
const signedTask = ethers.AbiCoder.defaultAbiCoder().encode( | ||
["address[]", "bytes[]", "uint32"], | ||
[operators, signatures, ethers.toBigInt(taskCreatedBlock)] | ||
); | ||
|
||
const tx = await helloWorldServiceManager.respondToTask( | ||
{ name: taskName, taskCreatedBlock: taskCreatedBlock }, | ||
taskIndex, | ||
signedTask | ||
); | ||
await tx.wait(); | ||
}); | ||
|
||
afterAll(async () => { | ||
await anvil.stop(); | ||
}); | ||
}); |
Oops, something went wrong.