diff options
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() }) } |