mirror of
https://github.com/featurebasedb/featurebase.git
synced 2026-10-15 07:18:04 +00:00
4c53f86e82
i used this script, a little clunky but it got the job done
```bash
for file in `find . -type f -print | grep '\.go'`; do
sed '1,/^\/\/ limitations under the License.$/d' $file > $file.tmp;
result=`cat $file.tmp`
if [[ result != "" ]]; then
gofmt $file.tmp &> /dev/null;
if [[ $? == 0 ]]; then
mv $file.tmp $file && gofmt -w $file;
else
rm $file.tmp;
fi
else
rm $file.tmp;
fi
done
```
49 lines
1.3 KiB
Go
49 lines
1.3 KiB
Go
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])
|
|
}
|
|
}
|
|
|
|
}
|