aboutsummaryrefslogtreecommitdiff
path: root/internal/conf/mocks.go
diff options
context:
space:
mode:
authorJoe Chen <jc@unknwon.io>2023-02-07 23:39:00 +0800
committerGitHub <noreply@github.com>2023-02-07 23:39:00 +0800
commit133b9d90441008ee175e1f8e6369e06309e1392a (patch)
tree70f29798055962f6700ba6a93b023a8328a5eff4 /internal/conf/mocks.go
parent7c453d5b3632a6bbdbd99205c518303a9e25a4e1 (diff)
refactor(db): finish migrate methods off `user.go` (#7337)
Diffstat (limited to 'internal/conf/mocks.go')
-rw-r--r--internal/conf/mocks.go8
1 files changed, 8 insertions, 0 deletions
diff --git a/internal/conf/mocks.go b/internal/conf/mocks.go
index 51d1fb14..efdd3aeb 100644
--- a/internal/conf/mocks.go
+++ b/internal/conf/mocks.go
@@ -37,11 +37,15 @@ func SetMockServer(t *testing.T, opts ServerOpts) {
})
}
+var mockSSH sync.Mutex
+
func SetMockSSH(t *testing.T, opts SSHOpts) {
+ mockSSH.Lock()
before := SSH
SSH = opts
t.Cleanup(func() {
SSH = before
+ mockSSH.Unlock()
})
}
@@ -65,10 +69,14 @@ func SetMockUI(t *testing.T, opts UIOpts) {
})
}
+var mockPicture sync.Mutex
+
func SetMockPicture(t *testing.T, opts PictureOpts) {
+ mockPicture.Lock()
before := Picture
Picture = opts
t.Cleanup(func() {
Picture = before
+ mockPicture.Unlock()
})
}