Function TestWriteOutput

cmd/gosqlx/cmd/stdin_utils_test.go:189–248

(t *testing.T)

}

func TestWriteOutput(t *testing.T) {
	// Use platform-appropriate temp directory
	tmpFile := filepath.Join(os.TempDir(), "test_output.sql")

	tests := []struct {
		name       string
		content    []byte
		outputFile string
		wantErr    bool
		cleanup    func()
	}{
		{
			name:       "write to stdout",
			content:    []byte("SELECT * FROM users"),
			outputFile: "",
			wantErr:    false,
		},
		{
			name:       "write to file",
			content:    []byte("SELECT * FROM users"),
			outputFile: tmpFile,
			wantErr:    false,
			cleanup: func() {
				os.Remove(tmpFile)
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if tt.cleanup != nil {
				defer tt.cleanup()
			}

			var buf bytes.Buffer
			err := WriteOutput(tt.content, tt.outputFile, &buf)
			if (err != nil) != tt.wantErr {
				t.Errorf("WriteOutput() error = %v, wantErr %v", err, tt.wantErr)
				return
			}

			// If writing to stdout, verify content
			if tt.outputFile == "" {
				if !bytes.Equal(buf.Bytes(), tt.content) {
					t.Errorf("WriteOutput() stdout content mismatch")
				}
			} else {
				// If writing to file, verify file exists and content
				content, err := os.ReadFile(tt.outputFile)
				if err != nil {
					t.Errorf("Failed to read output file: %v", err)
					return
				}
				if !bytes.Equal(content, tt.content) {
					t.Errorf("WriteOutput() file content mismatch")
				}
			}
		})