type and bind updates
- using sql nullTypes - adding additional bind type option
This commit is contained in:
97
lazy.go
97
lazy.go
@@ -8,6 +8,7 @@ import (
|
||||
"math/rand/v2"
|
||||
"reflect"
|
||||
"regexp"
|
||||
"time"
|
||||
|
||||
"github.com/jmoiron/sqlx"
|
||||
)
|
||||
@@ -20,8 +21,15 @@ const (
|
||||
DB_TAG = "db" //tag used to parse sql fields in example struct
|
||||
LAZY_TAG = "lazy" //tag label for KEY_VALUE
|
||||
KEY_VALUE = "key" //tag value for LAZY_TAG
|
||||
//bind values
|
||||
BindQuestion Binds = 1
|
||||
BindDollar Binds = 2
|
||||
BindNamed Binds = 3
|
||||
BindAt Binds = 4
|
||||
)
|
||||
|
||||
type Binds int
|
||||
|
||||
// mock data generated based on config
|
||||
type Mock struct {
|
||||
Query string
|
||||
@@ -35,11 +43,11 @@ type Config struct {
|
||||
Example any
|
||||
Keys []any //length of keys must = RowCount, if set
|
||||
RowCount int
|
||||
BindType Binds
|
||||
}
|
||||
|
||||
// generates mock data based on configuration to be used for sqlmock
|
||||
func RandomGenerate(m Config) (*Mock, error) {
|
||||
fmt.Println("huh")
|
||||
//example struct cannot be nil and must be a struct
|
||||
if m.Example == nil {
|
||||
return nil, errors.New("example value cannot be nil")
|
||||
@@ -91,31 +99,63 @@ func RandomGenerate(m Config) (*Mock, error) {
|
||||
//generate random values
|
||||
nv := kindToRandom(field)
|
||||
if nv == nil {
|
||||
//this will catch the sql.Null* types, although not slices of them. In these statements it should be
|
||||
//50/50 chance of null vs value and then psuedo random from there
|
||||
//1. Write simple 50/50 function
|
||||
//2. move this logic to nullKindToRandom function
|
||||
//3. add bind type option to allow for more than just AT now that I'm using postgres
|
||||
//move this logic to nullKindToRandom function
|
||||
nullType := false
|
||||
switch field.Type {
|
||||
case reflect.TypeOf(sql.NullTime{}):
|
||||
//
|
||||
case reflect.TypeOf(sql.NullInt16{}):
|
||||
//
|
||||
case reflect.TypeOf(sql.NullInt32{}):
|
||||
//
|
||||
case reflect.TypeOf(sql.NullInt64{}):
|
||||
//
|
||||
case reflect.TypeOf(sql.NullBool{}):
|
||||
//
|
||||
case reflect.TypeOf(sql.NullFloat64{}):
|
||||
//
|
||||
case reflect.TypeOf(sql.NullString{}):
|
||||
//
|
||||
case reflect.TypeOf(sql.NullByte{}):
|
||||
//
|
||||
if isNull() {
|
||||
rows[y] = append(rows[y], nil)
|
||||
} else {
|
||||
rows[y] = append(rows[y], time.Now())
|
||||
}
|
||||
nullType = true
|
||||
case reflect.TypeOf(sql.NullInt16{}):
|
||||
if isNull() {
|
||||
rows[y] = append(rows[y], nil)
|
||||
} else {
|
||||
rows[y] = append(rows[y], rand.Int())
|
||||
}
|
||||
nullType = true
|
||||
case reflect.TypeOf(sql.NullInt32{}):
|
||||
if isNull() {
|
||||
rows[y] = append(rows[y], nil)
|
||||
} else {
|
||||
rows[y] = append(rows[y], rand.Int32())
|
||||
}
|
||||
nullType = true
|
||||
case reflect.TypeOf(sql.NullInt64{}):
|
||||
if isNull() {
|
||||
rows[y] = append(rows[y], nil)
|
||||
} else {
|
||||
rows[y] = append(rows[y], rand.Int64())
|
||||
}
|
||||
nullType = true
|
||||
case reflect.TypeOf(sql.NullBool{}):
|
||||
if isNull() {
|
||||
rows[y] = append(rows[y], nil)
|
||||
} else {
|
||||
rows[y] = append(rows[y], rand.Int()%2 == 0)
|
||||
}
|
||||
nullType = true
|
||||
case reflect.TypeOf(sql.NullFloat64{}):
|
||||
if isNull() {
|
||||
rows[y] = append(rows[y], nil)
|
||||
} else {
|
||||
rows[y] = append(rows[y], rand.Float64())
|
||||
}
|
||||
nullType = true
|
||||
case reflect.TypeOf(sql.NullString{}):
|
||||
if isNull() {
|
||||
rows[y] = append(rows[y], nil)
|
||||
} else {
|
||||
rows[y] = append(rows[y], fmt.Sprintf("random %d", rand.Int()))
|
||||
}
|
||||
nullType = true
|
||||
}
|
||||
if !nullType {
|
||||
return nil, fmt.Errorf("could not match type: %s", retType.Name())
|
||||
}
|
||||
}
|
||||
|
||||
rows[y] = append(rows[y], nv)
|
||||
}
|
||||
@@ -123,7 +163,7 @@ func RandomGenerate(m Config) (*Mock, error) {
|
||||
|
||||
return &Mock{
|
||||
//sql is rebound and escaped for sqlmock.ExpectQuery
|
||||
Query: sqlx.Rebind(sqlx.AT, regexp.QuoteMeta(m.Query)),
|
||||
Query: sqlx.Rebind(int(m.BindType), regexp.QuoteMeta(m.Query)),
|
||||
Columns: columns,
|
||||
Rows: rows,
|
||||
}, nil
|
||||
@@ -144,9 +184,11 @@ func kindToRandom(field reflect.StructField) any {
|
||||
case reflect.Float64:
|
||||
return rand.Float64()
|
||||
case reflect.String:
|
||||
return fmt.Sprintf("%d", rand.Int())
|
||||
return fmt.Sprintf("random %d", rand.Int())
|
||||
case reflect.Bool:
|
||||
return rand.Int()%2 == 0
|
||||
case reflect.TypeOf(time.Time{}).Kind():
|
||||
return time.Now()
|
||||
case reflect.Array:
|
||||
underlying := field.Type.Elem().Kind()
|
||||
count := reflect.ValueOf(field).Len() //fill entire length
|
||||
@@ -245,3 +287,12 @@ func kindToRandom(field reflect.StructField) any {
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func isNull() bool {
|
||||
x := rand.IntN(2)
|
||||
if x%2 == 0 {
|
||||
return true
|
||||
} else {
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user