Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Chore/rebase master #115

Merged
merged 107 commits into from
Dec 18, 2023
Merged
Show file tree
Hide file tree
Changes from 106 commits
Commits
Show all changes
107 commits
Select commit Hold shift + click to select a range
5edcc78
fix: remove husky command that only needs to be run once
Nov 8, 2023
b0dd616
Merge pull request #59 from Layr-Labs/chore/fix-contributing-doc
wadealexc Nov 8, 2023
cda67ff
update nonSignerForQuorumIndex outside of loop
gpsanant Nov 16, 2023
7c8a7c9
Merge branch 'master' into fix-loop-reset-nonSignerForQuorumIndex
ChaoticWalrus Nov 16, 2023
d5d520e
Merge pull request #65 from Layr-Labs/fix-loop-reset-nonSignerForQuor…
gpsanant Nov 16, 2023
119085d
chore: add MacOS indexing file to gitignore
ChaoticWalrus Nov 16, 2023
92aefd2
chore: correct addresses for M2 Goerli deployment
ChaoticWalrus Nov 16, 2023
bee1c9f
Merge pull request #67 from Layr-Labs/repo-cleanup
ChaoticWalrus Nov 16, 2023
2eee795
chore: test out relative import
Dec 4, 2023
e610c6a
chore: all relative imports
Dec 4, 2023
f8a0cf8
chore: fix remaining relative imports
Dec 4, 2023
889bfad
fix: update missed import
Dec 4, 2023
af9ddea
fix: additional missed imports
Dec 4, 2023
88b200d
Merge pull request #84 from Layr-Labs/feat/relative-imports
stevennevins Dec 4, 2023
0f1a70b
Update README.md to avoid confusing phrasing (#75)
bowenli86 Dec 15, 2023
08c29fb
style: use for loop indices when the index is just an index
wadealexc Oct 24, 2023
40b343a
refactor: Use a signed delta value in StakeRegistry to remove tons of…
wadealexc Oct 24, 2023
66e2ddc
style: use over , and simplify
wadealexc Oct 24, 2023
057da67
feat: only update total history for nonzero delta, and dont push upda…
wadealexc Oct 24, 2023
3125c79
style: use uint256 in registry coordinator
wadealexc Oct 25, 2023
c496d5e
test: fix broken tests for StakeRegistry changes
wadealexc Oct 25, 2023
a8d0e20
refactor: wip refactor to move "createQuorum" to the registry coordin…
wadealexc Oct 26, 2023
acdd56a
style: use for loop indices when the index is just an index
wadealexc Oct 24, 2023
32978ea
refactor: Use a signed delta value in StakeRegistry to remove tons of…
wadealexc Oct 24, 2023
7bbdf37
style: use over , and simplify
wadealexc Oct 24, 2023
4242029
feat: only update total history for nonzero delta, and dont push upda…
wadealexc Oct 24, 2023
7b691a1
style: use uint256 in registry coordinator
wadealexc Oct 25, 2023
c2d5959
test: fix broken tests for StakeRegistry changes
wadealexc Oct 25, 2023
e6a18f1
fix: fix compilation issues and tests
wadealexc Oct 30, 2023
732747a
refactor: wip refactor for index and blspubkey registries to simplify…
wadealexc Oct 31, 2023
b922421
style: pull out common logic to a helper method
wadealexc Nov 1, 2023
ea9aa1a
style: pulled additional logic out into a helper method
wadealexc Nov 1, 2023
b33bfc3
refactor: simplify registry coord state variable names and clean logi…
wadealexc Nov 1, 2023
c86303e
fix: enforce invariant that existing indice have nonzero length histo…
wadealexc Nov 1, 2023
4174295
style: shorten state variable and function naming in registry coordin…
wadealexc Nov 2, 2023
be8f444
style: shorten state variable and function naming in stake registry
wadealexc Nov 2, 2023
e1ef733
style: remove unused index registry function
wadealexc Nov 2, 2023
2d42360
refactor: pk compendium stores operator pubkeys and can look them up
wadealexc Nov 2, 2023
e2eb6d6
style: shorten state variable and function names to be more consisten…
wadealexc Nov 2, 2023
3f8458f
style: removed redundant check and swapped param order to be consiste…
wadealexc Nov 6, 2023
1d7b201
style: remove redundant checks from pubkey compendium
wadealexc Nov 6, 2023
a54109e
fix: replace borked state variable
wadealexc Nov 7, 2023
78155a4
refactor: move updateStakes to registry coordinator
wadealexc Nov 7, 2023
a9ab3f9
fix: fixes several compiler issues
wadealexc Nov 7, 2023
2ae9bad
fix: stack too deep issue addressed
wadealexc Nov 8, 2023
6fba84e
fix: actually commit natspec changes
wadealexc Nov 8, 2023
258b517
style: stakeregistry naming is more consistent with other contracts
wadealexc Nov 8, 2023
041cc55
style: rename upper bound in bitmap utils, and rename Operator to Ope…
wadealexc Nov 8, 2023
408ea4e
fix: operator has minimum if weight is equal to minimum
wadealexc Nov 8, 2023
614c433
test: update tests to support refactor
wadealexc Nov 8, 2023
48cfdc3
style: reword comments and remove duplicate check
wadealexc Nov 8, 2023
c5569cb
style: remove unneeded parenthesis and fix _registerOperator naming
wadealexc Nov 8, 2023
8c0ee3f
style: rename OperatorStakeUpdate struct to StakeUpdate, and swap eve…
wadealexc Nov 9, 2023
1a2024b
cleanup imports (#78)
0x0aa0 Nov 29, 2023
4a16d34
chore: merge master into m2-mainnet (#79)
wadealexc Nov 30, 2023
eadba0e
build: point eigenlayer gitmodule to m2-mainnet (#80)
8sunyuan Dec 1, 2023
648f8d7
ci: add rule to run CI on m2-mainnet branch
Dec 4, 2023
94d4029
chore: remove old tree file (#83)
stevennevins Dec 4, 2023
49ab307
chore: rename several contracts and fix interfaces and tests (#95)
wadealexc Dec 7, 2023
52c1ef5
chore: copy BLSPublicKeyCompendium storage and functions into BLSApkR…
ChaoticWalrus Dec 7, 2023
8c8c0ab
chore: delete BLSPublicKeyCompendium and associated interface
ChaoticWalrus Dec 7, 2023
df3f14a
feat: create harness for BLSApkRegistry and use it in tests
ChaoticWalrus Dec 11, 2023
6d1a256
chore: delete unused mock file
ChaoticWalrus Dec 11, 2023
f5443e8
chore: cleanup unused file and reduce mutability of test utils functions
ChaoticWalrus Dec 11, 2023
e0b2cc9
feat: make BLSApkRegistry. registerBLSPublicKey only callable by Regi…
ChaoticWalrus Dec 11, 2023
efa343e
chore: remove `RegistryCoordinator.registerOperator` method without p…
ChaoticWalrus Dec 12, 2023
c54b1ed
feat: create struct for pubkey registration params
ChaoticWalrus Dec 12, 2023
5b55366
fix: address stack-too-deep issue in RegistryCoordinator
ChaoticWalrus Dec 12, 2023
2ac5c90
chore: delete unused index
ChaoticWalrus Dec 12, 2023
f9d020a
feat: optimization for fetching operatorId
ChaoticWalrus Dec 12, 2023
96311bb
chore: shorten variable name that is quite clear from context
ChaoticWalrus Dec 12, 2023
266d426
feat: use EIP712 for the pubkeyRegistrationMessageHash
ChaoticWalrus Dec 12, 2023
19b23f8
chore: add NatSpec to getter function
ChaoticWalrus Dec 12, 2023
1ddc973
Feat(M2-Mainnet): StakeRegistry pull updates per quorum (#62)
8sunyuan Dec 8, 2023
af96516
feat: issue templates (#87)
stevennevins Dec 11, 2023
29ac14f
fix typo BLSPublicKeyCompendium.md
tudorpintea999 Dec 2, 2023
f45df7e
fix typo BLSSignatureChecker.md
tudorpintea999 Dec 2, 2023
7ad4768
fix typo StakeRegistry.md
tudorpintea999 Dec 2, 2023
02555d6
fix typo README.md
tudorpintea999 Dec 2, 2023
bbea0db
chore: remove ServiceManagerBase and add RegistryCoordinator owner (#98)
wadealexc Dec 12, 2023
cce5430
Feat: Add AVS/Operator Registration Support in RegistryCoordinator (#99)
ypatil12 Dec 12, 2023
25bbad9
chore: fix merge artifacts / broken calls
ChaoticWalrus Dec 13, 2023
1598e0b
chore: re-implement stack-too-deep fix
ChaoticWalrus Dec 13, 2023
57c9a22
chore: de-duplicate code into an internal function
ChaoticWalrus Dec 13, 2023
3e1ac84
fix: actually use the `operator` input to `_getOrCreateOperatorId`
ChaoticWalrus Dec 13, 2023
bb228bb
feat: add optimizer runs count to foundry config
ChaoticWalrus Dec 13, 2023
7b8e7c8
chore: remove redundant check and return data
ChaoticWalrus Dec 13, 2023
055ddec
chore: fewer memory operations(?)
ChaoticWalrus Dec 13, 2023
4b026fd
fix: reduce optimizer runs to meet contract code size limits
ChaoticWalrus Dec 13, 2023
e40e59c
chore: rename file to reflect it only being used in tests
ChaoticWalrus Dec 13, 2023
e53a400
chore: delete unused (memory) variable
ChaoticWalrus Dec 13, 2023
851849a
fix: have the harness import Test so it gets ignored in build sizes
wadealexc Dec 13, 2023
7a6e4a8
fix: remove circular dependency (#108)
wadealexc Dec 13, 2023
cb63e21
fix: address multiple issues in BLSSignatureChecker, optimize, and cl…
wadealexc Dec 14, 2023
497f758
docs: add wip docs and clean main README (#96)
wadealexc Dec 14, 2023
23185d8
chore: add eigenda goerli deployment (#111)
wadealexc Dec 14, 2023
c570c42
feat: concentrate eigenlayer avs interactions into service manager (#…
ChaoticWalrus Dec 14, 2023
fd1e1e8
fix: correctly set bit in stake registry (#112)
wadealexc Dec 14, 2023
a7f4b98
feat: add restakeable strategies to service manager for indexing (#113)
ypatil12 Dec 15, 2023
51171fc
chore: resolve imports
Dec 18, 2023
a13e3ba
Merge branch 'm2-mainnet' into chore/rebase-master
stevennevins Dec 18, 2023
cd7600e
fix: remove eigenlayer deployer
Dec 18, 2023
f272bd4
chore: additional relative imports
Dec 18, 2023
7cfa95a
chore: remove whitespace diff
Dec 18, 2023
0839f94
chore: clean up remappings
Dec 18, 2023
3d61d23
chore: remove whitespace diff
Dec 18, 2023
3999198
chore: stragglers
Dec 18, 2023
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions src/BLSApkRegistry.sol
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
// SPDX-License-Identifier: BUSL-1.1
pragma solidity =0.8.12;

import {BLSApkRegistryStorage} from "src/BLSApkRegistryStorage.sol";
import {BLSApkRegistryStorage} from "./BLSApkRegistryStorage.sol";

import {IRegistryCoordinator} from "src/interfaces/IRegistryCoordinator.sol";
import {IRegistryCoordinator} from "./interfaces/IRegistryCoordinator.sol";

import {BN254} from "src/libraries/BN254.sol";
import {BN254} from "./libraries/BN254.sol";

contract BLSApkRegistry is BLSApkRegistryStorage {
using BN254 for BN254.G1Point;
Expand Down
6 changes: 3 additions & 3 deletions src/BLSApkRegistryStorage.sol
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
// SPDX-License-Identifier: BUSL-1.1
pragma solidity =0.8.12;

import {IBLSApkRegistry} from "src/interfaces/IBLSApkRegistry.sol";
import {IRegistryCoordinator} from "src/interfaces/IRegistryCoordinator.sol";
import {IBLSApkRegistry} from "./interfaces/IBLSApkRegistry.sol";
import {IRegistryCoordinator} from "./interfaces/IRegistryCoordinator.sol";

import {Initializable} from "@openzeppelin-upgrades/contracts/proxy/utils/Initializable.sol";

import {BN254} from "src/libraries/BN254.sol";
import {BN254} from "./libraries/BN254.sol";

abstract contract BLSApkRegistryStorage is Initializable, IBLSApkRegistry {
/// @notice the hash of the zero pubkey aka BN254.G1Point(0,0)
Expand Down
12 changes: 6 additions & 6 deletions src/BLSSignatureChecker.sol
Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
// SPDX-License-Identifier: BUSL-1.1
pragma solidity =0.8.12;

import {IBLSSignatureChecker} from "src/interfaces/IBLSSignatureChecker.sol";
import {IRegistryCoordinator} from "src/interfaces/IRegistryCoordinator.sol";
import {IBLSApkRegistry} from "src/interfaces/IBLSApkRegistry.sol";
import {IStakeRegistry, IDelegationManager} from "src/interfaces/IStakeRegistry.sol";
import {IBLSSignatureChecker} from "./interfaces/IBLSSignatureChecker.sol";
import {IRegistryCoordinator} from "./interfaces/IRegistryCoordinator.sol";
import {IBLSApkRegistry} from "./interfaces/IBLSApkRegistry.sol";
import {IStakeRegistry, IDelegationManager} from "./interfaces/IStakeRegistry.sol";

import {BitmapUtils} from "src/libraries/BitmapUtils.sol";
import {BN254} from "src/libraries/BN254.sol";
import {BitmapUtils} from "./libraries/BitmapUtils.sol";
import {BN254} from "./libraries/BN254.sol";

/**
* @title Used for checking BLS aggregate signatures from the operators of a `BLSRegistry`.
Expand Down
4 changes: 2 additions & 2 deletions src/IndexRegistry.sol
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
// SPDX-License-Identifier: BUSL-1.1
pragma solidity =0.8.12;

import {IndexRegistryStorage} from "src/IndexRegistryStorage.sol";
import {IRegistryCoordinator} from "src/interfaces/IRegistryCoordinator.sol";
import {IndexRegistryStorage} from "./IndexRegistryStorage.sol";
import {IRegistryCoordinator} from "./interfaces/IRegistryCoordinator.sol";

/**
* @title A `Registry` that keeps track of an ordered list of operators for each quorum
Expand Down
4 changes: 2 additions & 2 deletions src/IndexRegistryStorage.sol
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,8 @@ pragma solidity =0.8.12;

import {Initializable} from "@openzeppelin-upgrades/contracts/proxy/utils/Initializable.sol";

import {IRegistryCoordinator} from "src/interfaces/IRegistryCoordinator.sol";
import {IIndexRegistry} from "src/interfaces/IIndexRegistry.sol";
import {IRegistryCoordinator} from "./interfaces/IRegistryCoordinator.sol";
import {IIndexRegistry} from "./interfaces/IIndexRegistry.sol";

/**
* @title Storage variables for the `IndexRegistry` contract.
Expand Down
10 changes: 5 additions & 5 deletions src/OperatorStateRetriever.sol
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
// SPDX-License-Identifier: BUSL-1.1
pragma solidity =0.8.12;

import {IRegistryCoordinator} from "src/interfaces/IRegistryCoordinator.sol";
import {IBLSApkRegistry} from "src/interfaces/IBLSApkRegistry.sol";
import {IStakeRegistry} from "src/interfaces/IStakeRegistry.sol";
import {IIndexRegistry} from "src/interfaces/IIndexRegistry.sol";
import {IRegistryCoordinator} from "./interfaces/IRegistryCoordinator.sol";
import {IBLSApkRegistry} from "./interfaces/IBLSApkRegistry.sol";
import {IStakeRegistry} from "./interfaces/IStakeRegistry.sol";
import {IIndexRegistry} from "./interfaces/IIndexRegistry.sol";

import {BitmapUtils} from "src/libraries/BitmapUtils.sol";
import {BitmapUtils} from "./libraries/BitmapUtils.sol";

/**
* @title OperatorStateRetriever with view functions that allow to retrieve the state of an AVSs registry system.
Expand Down
18 changes: 9 additions & 9 deletions src/RegistryCoordinator.sol
Original file line number Diff line number Diff line change
Expand Up @@ -9,16 +9,16 @@ import {EIP1271SignatureUtils} from "eigenlayer-contracts/src/contracts/librarie
import {IPauserRegistry} from "eigenlayer-contracts/src/contracts/interfaces/IPauserRegistry.sol";
import {Pausable} from "eigenlayer-contracts/src/contracts/permissions/Pausable.sol";

import {IRegistryCoordinator} from "src/interfaces/IRegistryCoordinator.sol";
import {IRegistryCoordinator} from "./interfaces/IRegistryCoordinator.sol";
import {ISignatureUtils} from "eigenlayer-contracts/src/contracts/interfaces/ISignatureUtils.sol";
import {IBLSApkRegistry} from "src/interfaces/IBLSApkRegistry.sol";
import {ISocketUpdater} from "src/interfaces/ISocketUpdater.sol";
import {IStakeRegistry} from "src/interfaces/IStakeRegistry.sol";
import {IIndexRegistry} from "src/interfaces/IIndexRegistry.sol";
import {IServiceManager} from "src/interfaces/IServiceManager.sol";

import {BitmapUtils} from "src/libraries/BitmapUtils.sol";
import {BN254} from "src/libraries/BN254.sol";
import {IBLSApkRegistry} from "./interfaces/IBLSApkRegistry.sol";
import {ISocketUpdater} from "./interfaces/ISocketUpdater.sol";
import {IStakeRegistry} from "./interfaces/IStakeRegistry.sol";
import {IIndexRegistry} from "./interfaces/IIndexRegistry.sol";
import {IServiceManager} from "./interfaces/IServiceManager.sol";

import {BitmapUtils} from "./libraries/BitmapUtils.sol";
import {BN254} from "./libraries/BN254.sol";

/**
* @title A `RegistryCoordinator` that has three registries:
Expand Down
8 changes: 4 additions & 4 deletions src/ServiceManagerBase.sol
Original file line number Diff line number Diff line change
Expand Up @@ -3,13 +3,13 @@ pragma solidity =0.8.12;

import {OwnableUpgradeable} from "@openzeppelin-upgrades/contracts/access/OwnableUpgradeable.sol";

import {BitmapUtils} from "src/libraries/BitmapUtils.sol";
import {BitmapUtils} from "./libraries/BitmapUtils.sol";
import {ISignatureUtils} from "eigenlayer-contracts/src/contracts/interfaces/ISignatureUtils.sol";
import {IDelegationManager} from "eigenlayer-contracts/src/contracts/interfaces/IDelegationManager.sol";

import {IServiceManager} from "src/interfaces/IServiceManager.sol";
import {IRegistryCoordinator} from "src/interfaces/IRegistryCoordinator.sol";
import {IStakeRegistry} from "src/interfaces/IStakeRegistry.sol";
import {IServiceManager} from "./interfaces/IServiceManager.sol";
import {IRegistryCoordinator} from "./interfaces/IRegistryCoordinator.sol";
import {IStakeRegistry} from "./interfaces/IStakeRegistry.sol";

/**
* @title Minimal implementation of a ServiceManager-type contract.
Expand Down
8 changes: 4 additions & 4 deletions src/StakeRegistry.sol
Original file line number Diff line number Diff line change
Expand Up @@ -3,12 +3,12 @@ pragma solidity =0.8.12;

import {IDelegationManager} from "eigenlayer-contracts/src/contracts/interfaces/IDelegationManager.sol";

import {StakeRegistryStorage} from "src/StakeRegistryStorage.sol";
import {StakeRegistryStorage} from "./StakeRegistryStorage.sol";

import {IRegistryCoordinator} from "src/interfaces/IRegistryCoordinator.sol";
import {IStakeRegistry} from "src/interfaces/IStakeRegistry.sol";
import {IRegistryCoordinator} from "./interfaces/IRegistryCoordinator.sol";
import {IStakeRegistry} from "./interfaces/IStakeRegistry.sol";

import {BitmapUtils} from "src/libraries/BitmapUtils.sol";
import {BitmapUtils} from "./libraries/BitmapUtils.sol";

/**
* @title A `Registry` that keeps track of stakes of operators for up to 256 quorums.
Expand Down
4 changes: 2 additions & 2 deletions src/StakeRegistryStorage.sol
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,8 @@ pragma solidity =0.8.12;
import {IDelegationManager} from "eigenlayer-contracts/src/contracts/interfaces/IDelegationManager.sol";
import {IStrategyManager} from "eigenlayer-contracts/src/contracts/interfaces/IStrategyManager.sol";

import {IRegistryCoordinator} from "src/interfaces/IRegistryCoordinator.sol";
import {IStakeRegistry} from "src/interfaces/IStakeRegistry.sol";
import {IRegistryCoordinator} from "./interfaces/IRegistryCoordinator.sol";
import {IStakeRegistry} from "./interfaces/IStakeRegistry.sol";

/**
* @title Storage variables for the `StakeRegistry` contract.
Expand Down
4 changes: 2 additions & 2 deletions src/interfaces/IBLSApkRegistry.sol
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
// SPDX-License-Identifier: BUSL-1.1
pragma solidity =0.8.12;

import {IRegistry} from "src/interfaces/IRegistry.sol";
import {IRegistry} from "./IRegistry.sol";

import {BN254} from "src/libraries/BN254.sol";
import {BN254} from "../libraries/BN254.sol";

/**
* @title Minimal interface for a registry that keeps track of aggregate operator public keys for among many quorums.
Expand Down
8 changes: 4 additions & 4 deletions src/interfaces/IBLSSignatureChecker.sol
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
// SPDX-License-Identifier: BUSL-1.1
pragma solidity =0.8.12;

import {IRegistryCoordinator} from "src/interfaces/IRegistryCoordinator.sol";
import {IBLSApkRegistry} from "src/interfaces/IBLSApkRegistry.sol";
import {IStakeRegistry, IDelegationManager} from "src/interfaces/IStakeRegistry.sol";
import {IRegistryCoordinator} from "./IRegistryCoordinator.sol";
import {IBLSApkRegistry} from "./IBLSApkRegistry.sol";
import {IStakeRegistry, IDelegationManager} from "./IStakeRegistry.sol";

import {BN254} from "src/libraries/BN254.sol";
import {BN254} from "../libraries/BN254.sol";

/**
* @title Used for checking BLS aggregate signatures from the operators of a EigenLayer AVS with the RegistryCoordinator/BLSApkRegistry/StakeRegistry architechture.
Expand Down
8 changes: 4 additions & 4 deletions src/interfaces/IRegistryCoordinator.sol
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
// SPDX-License-Identifier: BUSL-1.1
pragma solidity =0.8.12;

import {IBLSApkRegistry} from "src/interfaces/IBLSApkRegistry.sol";
import {IStakeRegistry} from "src/interfaces/IStakeRegistry.sol";
import {IIndexRegistry} from "src/interfaces/IIndexRegistry.sol";
import {BN254} from "src/libraries/BN254.sol";
import {IBLSApkRegistry} from "./IBLSApkRegistry.sol";
import {IStakeRegistry} from "./IStakeRegistry.sol";
import {IIndexRegistry} from "./IIndexRegistry.sol";
import {BN254} from "../libraries/BN254.sol";

/**
* @title Interface for a contract that coordinates between various registries for an AVS.
Expand Down
6 changes: 3 additions & 3 deletions test/ffi/BLSPubKeyCompendiumFFI.t.sol
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
// SPDX-License-Identifier: BUSL-1.1
pragma solidity =0.8.12;

import "src/BLSApkRegistry.sol";
import "test/ffi/util/G2Operations.sol";
import {IBLSApkRegistry} from "src/interfaces/IBLSApkRegistry.sol";
import "../../src/BLSApkRegistry.sol";
import "../ffi/util/G2Operations.sol";
import {IBLSApkRegistry} from "../../src/interfaces/IBLSApkRegistry.sol";

contract BLSApkRegistryFFITests is G2Operations {
using BN254 for BN254.G1Point;
Expand Down
12 changes: 6 additions & 6 deletions test/ffi/BLSSignatureCheckerFFI.t.sol
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
// SPDX-License-Identifier: BUSL-1.1
pragma solidity =0.8.12;

import {G2Operations} from "test/ffi/util/G2Operations.sol";
import {MockAVSDeployer} from "test/utils/MockAVSDeployer.sol";
import {BLSSignatureChecker} from "src/BLSSignatureChecker.sol";
import {OperatorStateRetriever} from "src/OperatorStateRetriever.sol";
import {BN254} from "src/libraries/BN254.sol";
import {BitmapUtils} from "src/libraries/BitmapUtils.sol";
import {G2Operations} from "../ffi/util/G2Operations.sol";
import {MockAVSDeployer} from "../utils/MockAVSDeployer.sol";
import {BLSSignatureChecker} from "../../src/BLSSignatureChecker.sol";
import {OperatorStateRetriever} from "../../src/OperatorStateRetriever.sol";
import {BN254} from "../../src/libraries/BN254.sol";
import {BitmapUtils} from "../../src/libraries/BitmapUtils.sol";


contract BLSSignatureCheckerFFITests is MockAVSDeployer, G2Operations {
Expand Down
2 changes: 1 addition & 1 deletion test/ffi/util/G2Operations.sol
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ pragma solidity =0.8.12;

import "forge-std/Test.sol";
import "openzeppelin-contracts/contracts/utils/Strings.sol";
import "src/libraries/BN254.sol";
import "../../../src/libraries/BN254.sol";

contract G2Operations is Test {
using Strings for uint256;
Expand Down
2 changes: 1 addition & 1 deletion test/harnesses/BLSApkRegistryHarness.sol
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
// SPDX-License-Identifier: BUSL-1.1
pragma solidity =0.8.12;

import "src/BLSApkRegistry.sol";
import "../../src/BLSApkRegistry.sol";

// wrapper around the BLSApkRegistry contract that exposes internal functionality, for unit testing _other functionality_.
contract BLSApkRegistryHarness is BLSApkRegistry {
Expand Down
2 changes: 1 addition & 1 deletion test/harnesses/BitmapUtilsWrapper.sol
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
// SPDX-License-Identifier: BUSL-1.1
pragma solidity =0.8.12;

import "src/libraries/BitmapUtils.sol";
import "../../src/libraries/BitmapUtils.sol";

// wrapper around the BitmapUtils library that exposes the internal functions
contract BitmapUtilsWrapper {
Expand Down
2 changes: 1 addition & 1 deletion test/harnesses/RegistryCoordinatorHarness.t.sol
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
// SPDX-License-Identifier: BUSL-1.1
pragma solidity =0.8.12;

import "src/RegistryCoordinator.sol";
import "../../src/RegistryCoordinator.sol";

import "forge-std/Test.sol";

Expand Down
2 changes: 1 addition & 1 deletion test/harnesses/StakeRegistryHarness.sol
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
// SPDX-License-Identifier: BUSL-1.1
pragma solidity =0.8.12;

import "src/StakeRegistry.sol";
import "../../src/StakeRegistry.sol";

// wrapper around the StakeRegistry contract that exposes the internal functions for unit testing.
contract StakeRegistryHarness is StakeRegistry {
Expand Down
2 changes: 1 addition & 1 deletion test/mocks/RegistryCoordinatorMock.sol
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
pragma solidity =0.8.12;


import "src/interfaces/IRegistryCoordinator.sol";
import "../../src/interfaces/IRegistryCoordinator.sol";


contract RegistryCoordinatorMock is IRegistryCoordinator {
Expand Down
4 changes: 2 additions & 2 deletions test/mocks/StakeRegistryMock.sol
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
// SPDX-License-Identifier: BUSL-1.1
pragma solidity =0.8.12;

import "src/interfaces/IStakeRegistry.sol";
import "src/interfaces/IRegistryCoordinator.sol";
import "../../src/interfaces/IStakeRegistry.sol";
import "../../src/interfaces/IRegistryCoordinator.sol";

/**
* @title Interface for a `Registry` that keeps track of stakes of operators for up to 256 quorums.
Expand Down
4 changes: 2 additions & 2 deletions test/unit/BLSSignatureCheckerUnit.t.sol
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
// SPDX-License-Identifier: BUSL-1.1
pragma solidity =0.8.12;

import "src/BLSSignatureChecker.sol";
import "test/utils/BLSMockAVSDeployer.sol";
import "../../src/BLSSignatureChecker.sol";
import "../utils/BLSMockAVSDeployer.sol";

contract BLSSignatureCheckerUnitTests is BLSMockAVSDeployer {
using BN254 for BN254.G1Point;
Expand Down
8 changes: 4 additions & 4 deletions test/unit/IndexRegistryUnit.t.sol
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
//SPDX-License-Identifier: BUSL-1.1
pragma solidity ^0.8.12;

import "src/interfaces/IIndexRegistry.sol";
import "src/IndexRegistry.sol";
import "test/mocks/RegistryCoordinatorMock.sol";
import "test/harnesses/BitmapUtilsWrapper.sol";
import "../../src/interfaces/IIndexRegistry.sol";
import "../../src/IndexRegistry.sol";
import "../mocks/RegistryCoordinatorMock.sol";
import "../harnesses/BitmapUtilsWrapper.sol";

import "forge-std/Test.sol";

Expand Down
3 changes: 2 additions & 1 deletion test/unit/RegistryCoordinatorUnit.t.sol
Original file line number Diff line number Diff line change
Expand Up @@ -354,11 +354,13 @@ contract RegistryCoordinatorUnit is MockAVSDeployer {
stakeRegistry.setOperatorWeight(uint8(quorumNumbers[0]), defaultOperator, defaultStake);
cheats.prank(defaultOperator);
cheats.roll(registrationBlockNumber);

registryCoordinator.registerOperator(quorumNumbers, defaultSocket, pubkeyRegistrationParams, emptySig);

cheats.prank(defaultOperator);
cheats.roll(nextRegistrationBlockNumber);
cheats.expectRevert("RegistryCoordinator._registerOperator: operator already registered for some quorums being registered for");

registryCoordinator.registerOperator(quorumNumbers, defaultSocket, pubkeyRegistrationParams, emptySig);
}

Expand Down Expand Up @@ -597,7 +599,6 @@ contract RegistryCoordinatorUnit is MockAVSDeployer {

// re-register the operator
registryCoordinator.registerOperator(quorumNumbers, defaultSocket, pubkeyRegistrationParams, emptySig);

// check success of registration
uint256 quorumBitmap = BitmapUtils.orderedBytesArrayToBitmap(quorumNumbers);
assertEq(registryCoordinator.getOperatorId(defaultOperator), defaultOperatorId, "1");
Expand Down
18 changes: 9 additions & 9 deletions test/unit/StakeRegistryUnit.t.sol
Original file line number Diff line number Diff line change
Expand Up @@ -8,21 +8,21 @@ import {Slasher} from "eigenlayer-contracts/src/contracts/core/Slasher.sol";
import {PauserRegistry} from "eigenlayer-contracts/src/contracts/permissions/PauserRegistry.sol";
import {ISlasher} from "eigenlayer-contracts/src/contracts/interfaces/ISlasher.sol";
import {IStrategy} from "eigenlayer-contracts/src/contracts/interfaces/IStrategy.sol";
import {IStakeRegistry} from "src/interfaces/IStakeRegistry.sol";
import {IIndexRegistry} from "src/interfaces/IIndexRegistry.sol";
import {IRegistryCoordinator} from "src/interfaces/IRegistryCoordinator.sol";
import {IBLSApkRegistry} from "src/interfaces/IBLSApkRegistry.sol";
import {IServiceManager} from "src/interfaces/IServiceManager.sol";
import {IStakeRegistry} from "../../src/interfaces/IStakeRegistry.sol";
import {IIndexRegistry} from "../../src/interfaces/IIndexRegistry.sol";
import {IRegistryCoordinator} from "../../src/interfaces/IRegistryCoordinator.sol";
import {IBLSApkRegistry} from "../../src/interfaces/IBLSApkRegistry.sol";
import {IServiceManager} from "../../src/interfaces/IServiceManager.sol";

import {BitmapUtils} from "src/libraries/BitmapUtils.sol";
import {BitmapUtils} from "../../src/libraries/BitmapUtils.sol";

import {StrategyManagerMock} from "eigenlayer-contracts/src/test/mocks/StrategyManagerMock.sol";
import {EigenPodManagerMock} from "eigenlayer-contracts/src/test/mocks/EigenPodManagerMock.sol";
import {DelegationManagerMock} from "eigenlayer-contracts/src/test/mocks/DelegationManagerMock.sol";

import {StakeRegistryHarness} from "test/harnesses/StakeRegistryHarness.sol";
import {StakeRegistry} from "src/StakeRegistry.sol";
import {RegistryCoordinatorHarness} from "test/harnesses/RegistryCoordinatorHarness.t.sol";
import {StakeRegistryHarness} from "../harnesses/StakeRegistryHarness.sol";
import {StakeRegistry} from "../../src/StakeRegistry.sol";
import {RegistryCoordinatorHarness} from "../harnesses/RegistryCoordinatorHarness.t.sol";

import "forge-std/Test.sol";

Expand Down
7 changes: 4 additions & 3 deletions test/unit/VoteWeigherBaseUnit.t.sol
Original file line number Diff line number Diff line change
Expand Up @@ -9,10 +9,11 @@ import {IStrategyManager} from "eigenlayer-contracts/src/contracts/interfaces/IS
import {IStrategy} from "eigenlayer-contracts/src/contracts/interfaces/IStrategy.sol";
import {IEigenPodManager} from "eigenlayer-contracts/src/contracts/interfaces/IEigenPodManager.sol";
import {ISlasher} from "eigenlayer-contracts/src/contracts/interfaces/ISlasher.sol";
import {IStakeRegistry} from "src/interfaces/IStakeRegistry.sol";
import {StakeRegistry} from "src/StakeRegistry.sol";
import {IServiceManager} from "../../src/interfaces/IServiceManager.sol";
import {StakeRegistry, IStakeRegistry} from "../../src/StakeRegistry.sol";

import {RegistryCoordinatorMock} from "../mocks/RegistryCoordinatorMock.sol";

import {RegistryCoordinatorMock} from "test/mocks/RegistryCoordinatorMock.sol";
import {OwnableMock} from "eigenlayer-contracts/src/test/mocks/OwnableMock.sol";
import {DelegationManagerMock} from "eigenlayer-contracts/src/test/mocks/DelegationManagerMock.sol";

Expand Down
Loading