diff options
Diffstat (limited to 'ride/store_test.go')
| -rw-r--r-- | ride/store_test.go | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/ride/store_test.go b/ride/store_test.go index 877f4e5..89c4830 100644 --- a/ride/store_test.go +++ b/ride/store_test.go @@ -197,6 +197,20 @@ func TestBackfill(t *testing.T) { assert.Equal(t, 1, count) } +func TestBackfillReportsProgress(t *testing.T) { + db := newTestDB(t) + require.NoError(t, Save(db, sampleRide(t))) + + var calls [][2]int + count, err := Backfill(db, func(current, total int) { + calls = append(calls, [2]int{current, total}) + }) + require.NoError(t, err) + assert.Equal(t, 1, count) + require.Len(t, calls, 1) + assert.Equal(t, [2]int{1, 1}, calls[0]) +} + func TestBackfillSkipsInvalidRide(t *testing.T) { db := newTestDB(t) _, err := db.Exec(` |