chore: update test for custom error

unchain-tech · Oct 12, 2023 · 2239fdb · 2239fdb
1 parent a382fff
commit 2239fdb
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/docs/Polygon-Whitelist-NFT/en/section-5/lesson-2_Automated contract testing.md b/docs/Polygon-Whitelist-NFT/en/section-5/lesson-2_Automated contract testing.md
@@ -161,9 +161,9 @@ describe('Shield', function () {
         const { shield, alice } = await loadFixture(deployWhitelistFixture);
 
         // Verify that an account that is not the owner of the contract will get an error if it tries to execute the setPaused function.
-        await expect(shield.connect(alice).setPaused(true)).to.be.revertedWith(
-          'Ownable: caller is not the owner',
-        );
+        await expect(shield.connect(alice).setPaused(true))
+          .to.be.revertedWithCustomError(shield, 'OwnableUnauthorizedAccount')
+          .withArgs(alice.address);
       });
     });
     context('when set to true', function () {
@@ -280,9 +280,9 @@ describe('Shield', function () {
         const { shield, alice } = await loadFixture(deployWhitelistFixture);
 
         // Verify that an error occurs when an account that is not the owner of the contract tries to execute the withdraw function.
-        await expect(shield.connect(alice).withdraw()).to.be.revertedWith(
-          'Ownable: caller is not the owner',
-        );
+        await expect(shield.connect(alice).withdraw())
+          .to.be.revertedWithCustomError(shield, 'OwnableUnauthorizedAccount')
+          .withArgs(alice.address);
       });
     });
     context('when owner executes', function () {

diff --git a/docs/Polygon-Whitelist-NFT/ja/section-5/lesson-2_コントラクトの自動テスト.md b/docs/Polygon-Whitelist-NFT/ja/section-5/lesson-2_コントラクトの自動テスト.md
@@ -161,9 +161,9 @@ describe('Shield', function () {
         const { shield, alice } = await loadFixture(deployWhitelistFixture);
 
         // コントラクトのオーナーではないアカウントが、setPaused関数を実行しようとするとエラーとなることを確認します。
-        await expect(shield.connect(alice).setPaused(true)).to.be.revertedWith(
-          'Ownable: caller is not the owner',
-        );
+        await expect(shield.connect(alice).setPaused(true))
+          .to.be.revertedWithCustomError(shield, 'OwnableUnauthorizedAccount')
+          .withArgs(alice.address);
       });
     });
     context('when set to true', function () {
@@ -280,9 +280,9 @@ describe('Shield', function () {
         const { shield, alice } = await loadFixture(deployWhitelistFixture);
 
         // コントラクトのオーナーではないアカウントが、withdraw関数を実行しようとするとエラーとなることを確認します。
-        await expect(shield.connect(alice).withdraw()).to.be.revertedWith(
-          'Ownable: caller is not the owner',
-        );
+        await expect(shield.connect(alice).withdraw())
+          .to.be.revertedWithCustomError(shield, 'OwnableUnauthorizedAccount')
+          .withArgs(alice.address);
       });
     });
     context('when owner executes', function () {