From 133b9d90441008ee175e1f8e6369e06309e1392a Mon Sep 17 00:00:00 2001 From: Joe Chen Date: Tue, 7 Feb 2023 23:39:00 +0800 Subject: refactor(db): finish migrate methods off `user.go` (#7337) --- internal/conf/mocks.go | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'internal/conf/mocks.go') 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() }) } -- cgit v1.2.3