Unit test when using Transaction

[andre-fajar-n]

I have used transaction like this

err := r.enforcer.GetAdapter().(*gormadapter.Adapter).Transaction(r.enforcer, func(i casbin.IEnforcer) error {
	allPolicies := i.GetFilteredPolicy(1, domainName)

	ok, err := i.RemovePolicies(allPolicies)
	if err != nil {
		return err
	}

	if !ok {
		err := fmt.Errorf("cannot delete this rbac config")
		return err
	}

	allGroupPolicies := enforcer.GetFilteredGroupingPolicy(2, domainName)

	ok, err := i.RemoveGroupingPolicies(allGroupPolicies)
	if err != nil {
		return err
	}

	if !ok {
		err := fmt.Errorf("cannot delete this grouping policies")
		return err
	}

	return nil
})

Then I want to write unit test of this function, but I don't know how to mock the Transaction. Any solution for my problem?

[MuZhou233]

Currently Transaction can not be mocked, but you can try this way

func YourFunc() {
	// other logic
	r.enforcer.GetAdapter().(*gormadapter.Adapter).Transaction(r.enforcer, TransactionFunc("domainName"))
	// other logic
}

func TransactionFunc(domainName string) func(i casbin.IEnforcer) error {
	return func(i casbin.IEnforcer) error {
		allPolicies := i.GetFilteredPolicy(1, domainName)

		ok, err := i.RemovePolicies(allPolicies)
		if err != nil {
			return err
		}

		if !ok {
			err := fmt.Errorf("cannot delete this rbac config")
			return err
		}

		allGroupPolicies := i.GetFilteredGroupingPolicy(2, domainName)

		ok, err := i.RemoveGroupingPolicies(allGroupPolicies)
		if err != nil {
			return err
		}

		if !ok {
			err := fmt.Errorf("cannot delete this grouping policies")
			return err
		}

		return nil
	}
}

func Test_TransactionFunc(t *testing.T) {
	// other logic
	err := TransactionFunc(domainName)(enforcer)
	// other logic
}

[hsluoyz]

@andre-fajar-n closed as resolved