Function TestTwoPhaseCommit_WithHooks

twopc/twopc_test.go:459–551

(t *testing.T)

}

func TestTwoPhaseCommit_WithHooks(t *testing.T) {
	errorBeforePrepare := false
	beforePrepareError := errors.New("before prepare error")
	errorBeforeCommit := false
	beforeCommitError := errors.New("before commit error")
	errorBeforeRollback := false
	beforeRollbackError := errors.New("before rollback error")
	policy = AllGood

	c := twopc.NewCoordinator(twopc.NewOptionsWithCallback(
		5*time.Second,
		func(cxt context.Context) error {
			if errorBeforePrepare {
				return beforePrepareError
			}

			return nil
		}, // before prepare
		func(ctx context.Context) error {
			if errorBeforeCommit {
				return beforeCommitError
			}

			return nil
		}, // before commit
		func(ctx context.Context) error {
			if errorBeforeRollback {
				return beforeRollbackError
			}

			return nil
		}, // before rollback
		nil))

	testNodeReset()

	res, err := c.Put(nodes, &RaftWriteBatchReq{TxID: 0, Cmds: []string{"+1", "-3", "+10"}})
	if err != nil {
		t.Fatalf("error occurred: %s", err.Error())
	}

	if res.(int64) != 8 {
		t.Fatalf("TwoPC returns invalid result: %v", res)
	}

	// error before prepare
	errorBeforePrepare = true
	errorBeforeCommit = false
	errorBeforeRollback = false

	testNodeReset()

	res, err = c.Put(nodes, &RaftWriteBatchReq{TxID: 1, Cmds: []string{"+1", "-3", "+10"}})
	if err == nil {
		t.Fatal("unexpected result: returned nil while expecting an error")
	} else if err != beforePrepareError {
		t.Fatal("unexpected result: beforePrepare error is expected")
	} else {