]> git.giacomolli.uk - onechat/blobdiff - whatsapp_import_test.go
Check spacing between Go statement blocks
[onechat] / whatsapp_import_test.go
index 5f7d21c656a4aa4d65771bae93217e811e4c47a2..d735f4c8b9aad51a72ac147d326c93b5e51d8e49 100644 (file)
@@ -242,7 +242,9 @@ func TestWhatsAppImportPlanCommandDoesNotChangeAgentOrCallModel(t *testing.T) {
        }})
 
        before := rebuildTestSnapshot(t, agent.dir)
        }})
 
        before := rebuildTestSnapshot(t, agent.dir)
+
        var report strings.Builder
        var report strings.Builder
+
        require.NoError(t, whatsappImport(t.Context(), []string{"-n", whatsappImportSample(t)}, &report))
        require.Equal(t, before, rebuildTestSnapshot(t, agent.dir))
 
        require.NoError(t, whatsappImport(t.Context(), []string{"-n", whatsappImportSample(t)}, &report))
        require.Equal(t, before, rebuildTestSnapshot(t, agent.dir))
 
@@ -266,6 +268,7 @@ func TestWhatsAppImportDoesNotRebuildWhenAllDaysAreHeld(t *testing.T) {
        agent := newRebuildTestAgent(t)
        dir := filepath.Dir(agent.dir)
        t.Setenv("ONECHAT_DIR", dir)
        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)
        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)
        require.NoError(t, os.WriteFile(filepath.Join(dir, "config.json"), []byte("not model settings"), 0o600))
 
        before := rebuildTestSnapshot(t, agent.dir)
+
        var report strings.Builder
        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")
        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)
 func TestWhatsAppImportReadsOnlyLocalSessionNames(t *testing.T) {
        dir := t.TempDir()
        whatsappTestSession(t, dir, true)
+
        session := &WhatsApp{
                dir: filepath.Join(dir, whatsappDirName),
        }
        session := &WhatsApp{
                dir: filepath.Join(dir, whatsappDirName),
        }