mirror of
https://github.com/featurebasedb/featurebase.git
synced 2026-08-28 10:54:59 +00:00
48aef0c8a4
```bash for file in `cat diffys`; do printf '%s\n%s\n' "// Copyright 2021 Molecula Corp. All rights reserved." "$(cat $file)" >$file; done ```
50 lines
1.4 KiB
Go
50 lines
1.4 KiB
Go
// Copyright 2021 Molecula Corp. All rights reserved.
|
|
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 {
|
|
t.Header = append(t.Header, headers...)
|
|
return nil
|
|
}
|
|
|
|
func (t *TestQueryResultWriter) WriteRowText(rowTexts ...string) error {
|
|
t.RowText = append(t.RowText, rowTexts...)
|
|
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])
|
|
}
|
|
}
|
|
|
|
}
|