X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/900ec0e7dad7a913190fcef9016e53495de9bdf8..effd32b0f2e4d146b219b1727601b0b7f6593638:/whatsapp_import_test.go?ds=sidebyside

diff --git a/whatsapp_import_test.go b/whatsapp_import_test.go
index 5f7d21c..f93cd19 100644
--- a/whatsapp_import_test.go
+++ b/whatsapp_import_test.go
@@ -242,11 +242,13 @@ func TestWhatsAppImportPlanCommandDoesNotChangeAgentOrCallModel(t *testing.T) {
 	}})
 
 	before := rebuildTestSnapshot(t, agent.dir)
+
 	var report strings.Builder
+
 	require.NoError(t, whatsappImport(t.Context(), []string{"-n", whatsappImportSample(t)}, &report))
 	require.Equal(t, before, rebuildTestSnapshot(t, agent.dir))
 
-	for _, want := range []string{"chats: 1", "Book Club", "batches: 5", "dates: ", "cutoff: none", "bytes: ", "messages: 6", "estimated calls: 8", "owner: unknown"} {
+	for _, want := range []string{"chats: 1", "Book Club", "batches: 5", "dates: ", "cutoff: none", "bytes: ", "messages: 6", "estimated calls: 8 (about 1.3 per message, limit 1500)", "owner: unknown"} {
 		require.Contains(t, report.String(), want)
 	}
 
@@ -266,6 +268,7 @@ func TestWhatsAppImportDoesNotRebuildWhenAllDaysAreHeld(t *testing.T) {
 	agent := newRebuildTestAgent(t)
 	dir := filepath.Dir(agent.dir)
 	t.Setenv("ONECHAT_DIR", dir)
+
 	path := whatsappImportSample(t)
 	plan, err := whatsappImportPlan(t.Context(), []string{path}, nil, nil, "", time.UTC)
 	require.NoError(t, err)
@@ -274,7 +277,9 @@ func TestWhatsAppImportDoesNotRebuildWhenAllDaysAreHeld(t *testing.T) {
 	require.NoError(t, os.WriteFile(filepath.Join(dir, "config.json"), []byte("not model settings"), 0o600))
 
 	before := rebuildTestSnapshot(t, agent.dir)
+
 	var report strings.Builder
+
 	require.NoError(t, whatsappImport(t.Context(), []string{path}, &report))
 	require.Contains(t, report.String(), "batches: 0")
 	require.Contains(t, report.String(), "estimated calls: 0")
@@ -284,6 +289,7 @@ func TestWhatsAppImportDoesNotRebuildWhenAllDaysAreHeld(t *testing.T) {
 func TestWhatsAppImportReadsOnlyLocalSessionNames(t *testing.T) {
 	dir := t.TempDir()
 	whatsappTestSession(t, dir, true)
+
 	session := &WhatsApp{
 		dir: filepath.Join(dir, whatsappDirName),
 	}