Fix flaky tests

This commit is contained in:
Michal Horejsek 2020-08-25 10:20:49 +02:00
parent 2269a9edb7
commit 84147a2cb0
2 changed files with 7 additions and 3 deletions

View File

@ -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) {

View File

@ -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) {