From 7abdec4871f73aa53721ed6cbb0af489b1ab1430 Mon Sep 17 00:00:00 2001
From: Samuil Borisov <samuil.bo5@gmail.com>
Date: Wed, 24 Apr 2024 18:48:16 +0300
Subject: [PATCH] adjust test for new errors

---
 test/ValidatorSet/ValidatorSet.test.ts | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/test/ValidatorSet/ValidatorSet.test.ts b/test/ValidatorSet/ValidatorSet.test.ts
index bc8c04b8..e133b625 100644
--- a/test/ValidatorSet/ValidatorSet.test.ts
+++ b/test/ValidatorSet/ValidatorSet.test.ts
@@ -370,7 +370,7 @@ describe("ValidatorSet", function () {
           validatorSet
             .connect(this.signers.governance)
             .addToWhitelist([this.signers.validators[0].address, this.signers.validators[1].address])
-        ).to.be.revertedWith("Previously whitelisted.");
+        ).to.be.revertedWithCustomError(validatorSet, "PreviouslyWhitelisted");
       });
 
       it("should be able to remove from whitelist", async function () {
@@ -410,7 +410,7 @@ describe("ValidatorSet", function () {
 
         await expect(
           validatorSet.connect(this.signers.governance).removeFromWhitelist([this.signers.validators[3].address])
-        ).to.be.revertedWith("Must be whitelisted.");
+        ).to.be.revertedWithCustomError(validatorSet, "MustBeWhitelisted");
 
         expect((await validatorSet.validators(this.signers.validators[3].address)).status).to.be.equal(
           VALIDATOR_STATUS.None
@@ -428,7 +428,7 @@ describe("ValidatorSet", function () {
           validatorSet
             .connect(this.signers.governance)
             .addToWhitelist([this.signers.validators[0].address, this.signers.validators[1].address])
-        ).to.be.revertedWith("Previously whitelisted.");
+        ).to.be.revertedWithCustomError(validatorSet, "PreviouslyWhitelisted");
       });
     });