golang scope function test

gistfile1.go

package main

import "errors"

func actionOk() (int, error) {
	return 1, nil
}

func actionErr() (int, error) {
	return 0, errors.New("err")
}

type action func() (int, error)

func scope(err error, rollback func(), cleanup func()) {
	if err == nil {
		cleanup()
	} else {
		rollback()
	}
}

func rollback(err error, f func()) {
	if err != nil {
		f()
	}
}

func cleanup(err error, f func()) {
	if err == nil {
		f()
	}
}

func testScope(first action, second action) {
	var err error
	_, err = first()
	defer scope(err,
		func() { println(" rollback: first ") },
		func() { println(" cleanup: first ") })

	_, err = second()
	defer scope(err, 
		func() { println(" rollback: second ") },
		func() { println(" cleanup: second ") })
}

func testSeparate(first action, second action) {
	var err error
	_, err = first()
	defer rollback(err, func() { println(" rollback: first ") })
	defer cleanup(err, func() { println(" cleanup: first ") })

	_, err = second()
	defer rollback(err, func() { println(" rollback: second ") })
	defer cleanup(err, func() { println(" cleanup: second ") })
}

func runTest(test func(action, action)) {
	println("Running test: OK, OK")
	test(actionOk, actionOk)
	println("Running test: OK, ERR")
	test(actionOk, actionErr)
	println("Running test: ERR, OK")
	test(actionErr, actionOk)
	println("Running test: ERR, ERR")
	test(actionErr, actionErr)
}

func main() {
	println("--------------------------")
	println("testing scope function")
	runTest(testScope)
	println("--------------------------")
	println("testing separate functions")
	runTest(testSeparate)
	println()
}