diff options
author | Joe Chen <jc@unknwon.io> | 2023-02-07 23:39:00 +0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-02-07 23:39:00 +0800 |
commit | 133b9d90441008ee175e1f8e6369e06309e1392a (patch) | |
tree | 70f29798055962f6700ba6a93b023a8328a5eff4 /internal/conf/mocks.go | |
parent | 7c453d5b3632a6bbdbd99205c518303a9e25a4e1 (diff) |
refactor(db): finish migrate methods off `user.go` (#7337)
Diffstat (limited to 'internal/conf/mocks.go')
-rw-r--r-- | internal/conf/mocks.go | 8 |
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() }) } |