X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/6e7c7f3d47f086b9d59fe40b91fb071e1528684e..0bc8db71aa5d2c55b215a7a15b073bb85a1172eb:/whatsapp_state_test.go diff --git a/whatsapp_state_test.go b/whatsapp_state_test.go index 8f3d44a..2631b4d 100644 --- a/whatsapp_state_test.go +++ b/whatsapp_state_test.go @@ -47,7 +47,9 @@ func TestWhatsAppBatchesFollowTheTimesOfTheMessages(t *testing.T) { } require.NoError(t, batches.CloseAll()) + var sizes []int + for _, batch := range batches.Closed() { sizes = append(sizes, len(batch.Messages)) } @@ -119,6 +121,7 @@ func TestWhatsAppBatchesSurviveARestart(t *testing.T) { again := whatsappTestState(t, path) require.NoError(t, again.CloseAll()) + closed := again.Closed() require.Len(t, closed, 1) require.Equal(t, []string{"hi a", "hi b"}, []string{closed[0].Messages[0].Text, closed[0].Messages[1].Text})