func (s *txnSuite) TestAfterHooks(c *gc.C) {
changeFuncs := []func(){
func() { s.insertDoc(c, "1", "Foo") },
}
defer txntesting.SetAfterHooks(c, s.txnRunner, changeFuncs...).Check()
maxAttempt := 0
buildTxn := func(attempt int) ([]txn.Op, error) {
maxAttempt = attempt
ops := []txn.Op{{
C: s.collection.Name,
Id: "1",
Assert: bson.D{{"name", "Foo"}},
Update: bson.D{{"$set", bson.D{{"name", "Bar"}}}},
}}
return ops, nil
}
err := s.txnRunner.Run(buildTxn)
c.Assert(err, gc.IsNil)
var found simpleDoc
err = s.collection.FindId("1").One(&found)
c.Assert(err, gc.IsNil)
c.Assert(maxAttempt, gc.Equals, 1)
doc := simpleDoc{"1", "Bar"}
c.Assert(found, gc.DeepEquals, doc)
}
func SetAfterHooks(c *gc.C, st *State, fs ...func()) txntesting.TransactionChecker {
runner := jujutxn.NewRunner(jujutxn.RunnerParams{Database: st.db})
st.transactionRunner = runner
return txntesting.SetAfterHooks(c, runner, fs...)
}
func SetAfterHooks(c *gc.C, st *State, fs ...func()) txntesting.TransactionChecker {
return txntesting.SetAfterHooks(c, newRunnerForHooks(st), fs...)
}
func SetAfterHooks(c *gc.C, st *State, fs ...func()) txntesting.TransactionChecker {
return txntesting.SetAfterHooks(c, st.transactionRunner, fs...)
}