source/csla10/CSLA.Test/Rollback.vb

@@ -0,0 +1,111 @@
+<TestFixture()> _
+Public Class Rollback
+
+  <Test()> _
+  Public Sub NoFail()
+    Session.Clear()
+    Dim root As RollbackRoot
+    root = root.NewRoot()
+
+    root.BeginEdit()
+    root.Data = "saved"
+    Assert.AreEqual("saved", root.Data, "data is 'saved'")
+    Assert.AreEqual(False, root.Fail, "fail is false")
+    Assert.AreEqual(True, root.IsDirty, "isdirty is true")
+    Assert.AreEqual(True, root.IsValid, "isvalid is true")
+    Assert.AreEqual(True, root.IsNew, "isnew is true")
+
+    Session.Clear()
+    Dim tmp As RollbackRoot = DirectCast(root.Clone, RollbackRoot)
+    Try
+      root.ApplyEdit()
+      root = DirectCast(root.Save, RollbackRoot)
+
+    Catch
+      root = tmp
+      Assert.Fail("exception occurred")
+    End Try
+
+    Assert.IsNotNull(root, "obj is not null")
+    Assert.AreEqual("Inserted", Session("Root"), "obj was inserted")
+    Assert.AreEqual("saved", root.Data, "data is 'saved'")
+    Assert.AreEqual(False, root.IsNew, "isnew is false")
+    Assert.AreEqual(False, root.IsDeleted, "isdeleted is false")
+    Assert.AreEqual(False, root.IsDirty, "isdirty is false")
+
+  End Sub
+
+  <Test()> _
+  Public Sub YesFail()
+    Session.Clear()
+    Dim root As RollbackRoot
+    root = root.NewRoot()
+
+    root.BeginEdit()
+    root.Data = "saved"
+    root.Fail = True
+    Assert.AreEqual("saved", root.Data, "data is 'saved'")
+    Assert.AreEqual(True, root.Fail, "fail is true")
+    Assert.AreEqual(True, root.IsDirty, "isdirty is true")
+    Assert.AreEqual(True, root.IsValid, "isvalid is true")
+    Assert.AreEqual(True, root.IsNew, "isnew is true")
+
+    Session.Clear()
+    Dim tmp As RollbackRoot = DirectCast(root.Clone, RollbackRoot)
+    Try
+      root.ApplyEdit()
+      root = DirectCast(root.Save, RollbackRoot)
+      Assert.Fail("exception didn't occur")
+
+    Catch
+      root = tmp
+    End Try
+
+    Assert.IsNotNull(root, "obj is not null")
+    Assert.AreEqual("Inserted", Session("Root"), "obj was inserted")
+    Assert.AreEqual("saved", root.Data, "data is 'saved'")
+    Assert.AreEqual(True, root.IsNew, "isnew is true")
+    Assert.AreEqual(False, root.IsDeleted, "isdeleted is false")
+    Assert.AreEqual(True, root.IsDirty, "isdirty is true")
+
+  End Sub
+
+  <Test()> _
+  Public Sub YesFailCancel()
+    Session.Clear()
+    Dim root As RollbackRoot
+    root = root.NewRoot()
+    Assert.AreEqual(True, root.IsDirty, "isdirty is true")
+    Assert.AreEqual("<new>", root.Data, "data is '<new>'")
+
+    root.BeginEdit()
+    root.Data = "saved"
+    root.Fail = True
+    Assert.AreEqual("saved", root.Data, "data is 'saved'")
+    Assert.AreEqual(True, root.Fail, "fail is true")
+    Assert.AreEqual(True, root.IsDirty, "isdirty is true")
+    Assert.AreEqual(True, root.IsValid, "isvalid is true")
+    Assert.AreEqual(True, root.IsNew, "isnew is true")
+
+    Session.Clear()
+    Dim tmp As RollbackRoot = DirectCast(root.Clone, RollbackRoot)
+    Try
+      root.ApplyEdit()
+      root = DirectCast(root.Save, RollbackRoot)
+      Assert.Fail("exception didn't occur")
+
+    Catch
+      root = tmp
+      root.CancelEdit()
+    End Try
+
+    Assert.IsNotNull(root, "obj is not null")
+    Assert.AreEqual("Inserted", Session("Root"), "obj was inserted")
+    Assert.AreEqual("<new>", root.Data, "data is '<new>'")
+    Assert.AreEqual(True, root.IsNew, "isnew is true")
+    Assert.AreEqual(False, root.IsDeleted, "isdeleted is false")
+    Assert.AreEqual(True, root.IsDirty, "isdirty is true")
+
+  End Sub
+
+End Class