sq/testh/fixt/fixt.go
2024-01-29 15:30:26 -07:00

119 lines
3.3 KiB
Go

// Package fixt contains common test fixture values.
// It exists as its own package separate from testh mostly
// for semantic convenience. Broadly speaking, for any one
// database column, we want to test three values: the zero value,
// a non-zero value, and NULL (note that for some columns, e.g.
// primary key columns, NULL does not make sense). Thus for each
// value there is a ValueZ (zero) and Value (non-zero) item.
package fixt
import (
"time"
"github.com/shopspring/decimal"
"github.com/neilotoole/sq/libsq/core/kind"
)
// These consts are test fixtures for various data types.
const (
// TimestampUnixNano1989 is 1989-11-09T15:17:59.1234567Z.
TimestampUnixNano1989 = 626627879123456700
Text string = "seven"
TextZ string = ""
Int int64 = 7
IntZ int64 = 0
Float float64 = 7.7
FloatZ float64 = 0
Bool bool = true
BoolZ bool = false
BitString string = "1001"
BitStringZ string = "0"
TimeOfDay string = "07:07:07"
TimeOfDayZ string = "00:00:00"
JSON string = `{"val": 7}`
JSONZ string = "{}"
EnumAlfa string = "alfa"
EnumBravo string = "bravo"
UUID string = "77777777-7777-7777-7777-777777777777"
UUIDZ string = "00000000-0000-0000-0000-000000000000"
)
// These vars are text fixtures for various data types.
var (
Decimal = decimal.New(7777, -2)
DecimalZ = decimal.New(0, 0)
Money = decimal.New(7777, -2)
MoneyZ = decimal.New(0, 0)
Bytes = []byte("seven")
BytesZ = []byte("")
Datetime = mustParseTime(time.RFC3339, "2017-07-07T07:07:07-00:00").UTC()
DatetimeZ = mustParseTime(time.RFC3339, "1989-11-09T00:00:00-00:00").UTC()
Date = mustParseTime(time.DateOnly, "2017-07-07").UTC()
DateZ = mustParseTime(time.DateOnly, "1989-11-09").UTC()
)
func mustParseTime(layout, value string) time.Time {
t, err := time.ParseInLocation(layout, value, time.UTC)
if err != nil {
panic(err)
}
return t
}
// ColNamePerKind returns a slice of column names, one column name for
// each kind (excepting kind.Unknown and kind.Null, unless withNull
// or withUnknown are set). If isIntBool is
// true, kind.Int is returned for "col_bool", otherwise kind.Bool.
func ColNamePerKind(isIntBool, withNull, withUnknown bool) (colNames []string, kinds []kind.Kind) {
colNames = []string{
"col_int", "col_float", "col_decimal", "col_bool", "col_text", "col_datetime", "col_date",
"col_time", "col_bytes",
}
kinds = []kind.Kind{
kind.Int, kind.Float, kind.Decimal, kind.Bool, kind.Text, kind.Datetime, kind.Date, kind.Time,
kind.Bytes,
}
if isIntBool {
kinds[3] = kind.Int
}
if withNull {
colNames = append(colNames, "col_null")
kinds = append(kinds, kind.Null)
}
if withUnknown {
colNames = append(colNames, "col_unknown")
kinds = append(kinds, kind.Unknown)
}
return colNames, kinds
}
// The gopher.gif image used for testing bytes.
const (
GopherFilename = "gopher.gif"
GopherPath = "testh/fixt/testdata/gopher.gif"
GopherSize = 1788 // filesize in bytes
// BlobDBPath is the path to blob.db, which contains the gopher image
// in the "blobs" table.
BlobDBPath = "drivers/sqlite3/testdata/blob.db"
)
var (
LosAngeles = mustLoadLocation("America/Los_Angeles")
Denver = mustLoadLocation("America/Denver")
)
func mustLoadLocation(name string) *time.Location {
loc, err := time.LoadLocation(name)
if err != nil {
panic(err)
}
return loc
}