Function TestRegisterAndUnregisterRunCommand_Bash

completion/bash_test.go:71–116

(t *testing.T)

}

func TestRegisterAndUnregisterRunCommand_Bash(t *testing.T) {
	home := t.TempDir()
	b := NewBashCompletion(home, &cobra.Command{Use: "celer"})

	// prepare a .bashrc without the register line
	bashrc := filepath.Join(home, ".bashrc")
	if err := os.WriteFile(bashrc, []byte("export LANG=C\n"), 0644); err != nil {
		t.Fatalf("failed to write .bashrc: %v", err)
	}

	// Register should append the registerBinary line
	if err := b.registerRunCommand(); err != nil {
		t.Fatalf("registerRunCommand failed: %v", err)
	}
	data, err := os.ReadFile(bashrc)
	if err != nil {
		t.Fatalf("failed to read .bashrc: %v", err)
	}
	if !strings.Contains(string(data), b.registerBinary) {
		t.Fatalf("expected .bashrc to contain register line, got: %s", string(data))
	}

	// Calling register again should not duplicate the line
	if err := b.registerRunCommand(); err != nil {
		t.Fatalf("second registerRunCommand failed: %v", err)
	}
	data, _ = os.ReadFile(bashrc)
	if count := strings.Count(string(data), b.registerBinary); count != 1 {
		t.Fatalf("expected register line to appear once, got %d", count)
	}

	// Now test unregister: create a .bashrc that has the register line plus other lines
	if err := os.WriteFile(bashrc, []byte("line1\n"+b.registerBinary+"\nline2\n"), 0644); err != nil {
		t.Fatalf("failed to write .bashrc for unregister test: %v", err)
	}
	if err := b.unregisterRunCommand(); err != nil {
		t.Fatalf("unregisterRunCommand failed: %v", err)
	}
	data, _ = os.ReadFile(bashrc)
	if strings.Contains(string(data), b.registerBinary) {
		t.Fatalf("expected register line to be removed, still present in: %s", string(data))
	}
	if !strings.Contains(string(data), "line1") || !strings.Contains(string(data), "line2") {
		t.Fatalf("expected other lines to be preserved, got: %s", string(data))
	}
}