diff --git a/internal/store/event_loop_test.go b/internal/store/event_loop_test.go index 9692363b..51751af8 100644 --- a/internal/store/event_loop_test.go +++ b/internal/store/event_loop_test.go @@ -24,6 +24,7 @@ import ( "github.com/ProtonMail/proton-bridge/pkg/pmapi" "github.com/golang/mock/gomock" + "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" ) @@ -102,10 +103,13 @@ func TestEventLoopUpdateMessageFromLoop(t *testing.T) { // Event loop runs in goroutine and will be stopped by deferred mock clearing. go m.store.eventLoop.start() - require.Eventually(t, func() bool { - msg, err := m.store.getMessageFromDB("msg1") + var err error + assert.Eventually(t, func() bool { + var msg *pmapi.Message + msg, err = m.store.getMessageFromDB("msg1") return err == nil && msg.Subject == newSubject }, time.Second, 10*time.Millisecond) + require.NoError(t, err) } func TestEventLoopUpdateMessage(t *testing.T) { diff --git a/pkg/parallel/parallel_test.go b/pkg/parallel/parallel_test.go index 8fc999de..2fdc8d70 100644 --- a/pkg/parallel/parallel_test.go +++ b/pkg/parallel/parallel_test.go @@ -32,7 +32,7 @@ var ( testInput = []interface{}{1, 2, 3, 4, 5, 6, 7, 8, 9, 10} wantOutput = []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10} testProcessSleep = 100 // ms - runParallelTimeOverhead = 100 // ms + runParallelTimeOverhead = 150 // ms ) func TestParallel(t *testing.T) {