88 "testing"
99
1010 "code.gitea.io/gitea/models/auth"
11+ "code.gitea.io/gitea/modules/test"
1112 "code.gitea.io/gitea/services/auth/source/ldap"
1213
1314 "github.com/stretchr/testify/assert"
@@ -16,9 +17,7 @@ import (
1617
1718func TestAddLdapBindDn (t * testing.T ) {
1819 // Mock cli functions to do not exit on error
19- osExiter := cli .OsExiter
20- defer func () { cli .OsExiter = osExiter }()
21- cli .OsExiter = func (code int ) {}
20+ defer test .MockVariableValue (& cli .OsExiter , func (code int ) {})()
2221
2322 // Test cases
2423 cases := []struct {
@@ -256,9 +255,7 @@ func TestAddLdapBindDn(t *testing.T) {
256255
257256func TestAddLdapSimpleAuth (t * testing.T ) {
258257 // Mock cli functions to do not exit on error
259- osExiter := cli .OsExiter
260- defer func () { cli .OsExiter = osExiter }()
261- cli .OsExiter = func (code int ) {}
258+ defer test .MockVariableValue (& cli .OsExiter , func (code int ) {})()
262259
263260 // Test cases
264261 cases := []struct {
@@ -487,9 +484,7 @@ func TestAddLdapSimpleAuth(t *testing.T) {
487484
488485func TestUpdateLdapBindDn (t * testing.T ) {
489486 // Mock cli functions to do not exit on error
490- osExiter := cli .OsExiter
491- defer func () { cli .OsExiter = osExiter }()
492- cli .OsExiter = func (code int ) {}
487+ defer test .MockVariableValue (& cli .OsExiter , func (code int ) {})()
493488
494489 // Test cases
495490 cases := []struct {
@@ -964,9 +959,7 @@ func TestUpdateLdapBindDn(t *testing.T) {
964959
965960func TestUpdateLdapSimpleAuth (t * testing.T ) {
966961 // Mock cli functions to do not exit on error
967- osExiter := cli .OsExiter
968- defer func () { cli .OsExiter = osExiter }()
969- cli .OsExiter = func (code int ) {}
962+ defer test .MockVariableValue (& cli .OsExiter , func (code int ) {})()
970963
971964 // Test cases
972965 cases := []struct {
0 commit comments