Added tests

planetarium · Jan 2, 2024 · 2197636 · 2197636
1 parent ef390d4
commit 2197636
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/Libplanet.Tests/Action/AccountTest.cs b/Libplanet.Tests/Action/AccountTest.cs
@@ -143,6 +143,23 @@ public virtual void States()
             Assert.Equal("z", (Text)b.GetState(_addr[0]));
         }
 
+        [Fact]
+        public void RemoveState()
+        {
+            IAccount a = _initAccount.SetState(_addr[0], (Text)"A");
+            a = a.SetState(_addr[1], (Text)"B");
+            Assert.Equal((Text)"A", a.GetState(_addr[0]));
+            Assert.Equal((Text)"B", a.GetState(_addr[1]));
+
+            a = a.RemoveState(_addr[0]);
+            Assert.Null(a.GetState(_addr[0]));
+            Assert.Equal((Text)"B", a.GetState(_addr[1]));
+
+            a = a.RemoveState(_addr[1]);
+            Assert.Null(a.GetState(_addr[0]));
+            Assert.Null(a.GetState(_addr[1]));
+        }
+
         [Fact]
         public virtual void FungibleAssets()
         {