add test for pgWriteDistinctTimestamp

server/pg_internal_test.go

@@ -0,0 +1,53 @@
+package server
+
+import (
+	"testing"
+
+	pilosa "github.com/molecula/featurebase/v2"
+	"github.com/molecula/featurebase/v2/pg"
+)
+
+// pg_internal_test.go tests unexported methods from server/pg.go
+
+// TestQueryResultWriter implements the QueryResultWriter interface for testing
+type TestQueryResultWriter struct {
+	Header  []pg.ColumnInfo
+	RowText []string
+	TagTag  string
+}
+
+func (t *TestQueryResultWriter) WriteHeader(headers ...pg.ColumnInfo) error {
+	for _, header := range headers {
+		t.Header = append(t.Header, header)
+	}
+	return nil
+}
+
+func (t *TestQueryResultWriter) WriteRowText(rowTexts ...string) error {
+	for _, rowText := range rowTexts {
+		t.RowText = append(t.RowText, rowText)
+	}
+	return nil
+}
+
+func (t *TestQueryResultWriter) Tag(tag string) {
+	t.TagTag = tag
+}
+
+func TestPgWriteDistinctTimestamp(t *testing.T) {
+	w := TestQueryResultWriter{}
+	expected := pilosa.DistinctTimestamp{Name: "test", Values: []string{"date1", "date2", "date3"}}
+	pgWriteDistinctTimestamp(&w, expected)
+	if w.Header[0].Name != expected.Name {
+		t.Fatalf("Header Name is wrong. got %v, want %v", w.Header[0], expected.Name)
+	}
+	if w.Header[0].Type != pg.TypeCharoid {
+		t.Fatalf("Header Type is wrong. got %v, want %v", w.Header[0].Type, pg.TypeCharoid)
+	}
+	for i, value := range w.RowText {
+		if value != expected.Values[i] {
+			t.Fatalf("Value not written properly. got %v, want %v", value, expected.Values[i])
+		}
+	}
+
+}