fix system test

splitice · Aug 8, 2023 · 8986c94 · 8986c94
1 parent 8c414ab
commit 8986c94
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/IPTables.Net.Tests/IpTableSystemTests.cs b/IPTables.Net.Tests/IpTableSystemTests.cs
@@ -1,4 +1,5 @@
-using IPTables.Net.Iptables;
+using IPTables.Net.Exceptions;
+using IPTables.Net.Iptables;
 using IPTables.Net.Iptables.Adapter;
 using NUnit.Framework;
 using System;
@@ -36,7 +37,7 @@ public void TestGetRules()
                 }
 
                 // Invalid table cause exception
-                Assert.Throws<ArgumentException>(() => _system.GetRules("_invalidTableName", IP_VERSION));
+                Assert.Throws<IpTablesNetException>(() => _system.GetRules("_invalidTableName", IP_VERSION));
             }
         }