relationalmodel_test.go

package gorma_test

import (
	"fmt"
	"testing"

	"github.com/goadesign/goa/design"
	"github.com/goadesign/gorma"
	"github.com/goadesign/gorma/dsl"
)

func TestModelContext(t *testing.T) {
	sg := &gorma.RelationalModelDefinition{
		UserTypeDefinition: &design.UserTypeDefinition{
			AttributeDefinition: &design.AttributeDefinition{},
		},
	}

	sg.Type = design.String
	sg.ModelName = "SG"

	c := sg.Context()
	exp := fmt.Sprintf("RelationalModel %#v", sg.Name())
	if c != exp {
		t.Errorf("Expected %s, got %s", exp, c)
	}

	sg.ModelName = ""

	c = sg.Context()
	exp = "unnamed RelationalModel"
	if c != exp {
		t.Errorf("Expected %s, got %s", exp, c)
	}
}

func TestModelDSL(t *testing.T) {
	sg := &gorma.RelationalModelDefinition{}
	f := func() {
		return
	}
	sg.DefinitionDSL = f
	c := sg.DSL()
	if c == nil {
		t.Errorf("Expected %T, got nil", f)
	}

}

func TestPKAttributesSingle(t *testing.T) {
	sg := &gorma.RelationalModelDefinition{}
	sg.RelationalFields = make(map[string]*gorma.RelationalFieldDefinition)
	f := makePK("id")
	sg.PrimaryKeys = append(sg.PrimaryKeys, f)
	sg.RelationalFields[f.FieldName] = f

	pka := sg.PKAttributes()

	if pka != "id int" {
		t.Errorf("Expected %s, got %s", "id int", pka)
	}

}
func TestPKAttributesMultiple(t *testing.T) {
	sg := &gorma.RelationalModelDefinition{}
	sg.RelationalFields = make(map[string]*gorma.RelationalFieldDefinition)
	f := makePK("Field1")
	sg.PrimaryKeys = append(sg.PrimaryKeys, f)
	sg.RelationalFields[f.FieldName] = f

	f2 := makePK("Field2")
	sg.RelationalFields[f2.FieldName] = f2
	sg.PrimaryKeys = append(sg.PrimaryKeys, f2)

	pka := sg.PKAttributes()

	if pka != "field1 int,field2 int" {
		t.Errorf("Expected %s, got %s", "field1 int,field2 int", pka)
	}

}
func makePK(name string) *gorma.RelationalFieldDefinition {

	f := &gorma.RelationalFieldDefinition{}
	f.FieldName = dsl.SanitizeFieldName(name)
	f.DatabaseFieldName = dsl.SanitizeDBFieldName(f.FieldName)
	f.Datatype = gorma.Integer
	f.PrimaryKey = true
	return f

}
func TestPKWhereSingle(t *testing.T) {
	sg := &gorma.RelationalModelDefinition{}
	sg.RelationalFields = make(map[string]*gorma.RelationalFieldDefinition)
	f := &gorma.RelationalFieldDefinition{}
	f.FieldName = dsl.SanitizeFieldName("ID")
	f.DatabaseFieldName = dsl.SanitizeDBFieldName(f.FieldName)
	f.Datatype = gorma.Integer
	f.PrimaryKey = true

	sg.RelationalFields[f.FieldName] = f
	sg.PrimaryKeys = append(sg.PrimaryKeys, f)

	pkw := sg.PKWhere()

	if pkw != "id = ?" {
		t.Errorf("Expected %s, got %s", "id = ?", pkw)
	}

}

func TestPKWhereMultiple(t *testing.T) {
	sg := &gorma.RelationalModelDefinition{}
	sg.RelationalFields = make(map[string]*gorma.RelationalFieldDefinition)
	f := makePK("Field1")
	sg.RelationalFields[f.FieldName] = f
	sg.PrimaryKeys = append(sg.PrimaryKeys, f)

	f2 := makePK("Field2")
	sg.RelationalFields[f2.FieldName] = f2
	sg.PrimaryKeys = append(sg.PrimaryKeys, f2)

	pkw := sg.PKWhere()

	if pkw != "field1 = ? and field2 = ?" {
		t.Errorf("Expected %s, got %s", "field1 = ? and field2 = ?", pkw)
	}

}

func TestPKWhereFieldsSingle(t *testing.T) {
	sg := &gorma.RelationalModelDefinition{}
	sg.RelationalFields = make(map[string]*gorma.RelationalFieldDefinition)
	f := &gorma.RelationalFieldDefinition{}
	f.FieldName = dsl.SanitizeFieldName("ID")
	f.DatabaseFieldName = dsl.SanitizeDBFieldName(f.FieldName)
	f.Datatype = gorma.Integer
	f.PrimaryKey = true

	sg.RelationalFields[f.FieldName] = f
	sg.PrimaryKeys = append(sg.PrimaryKeys, f)

	pkw := sg.PKWhereFields()

	if pkw != "id" {
		t.Errorf("Expected %s, got %s", "id", pkw)
	}

}

func TestPKWhereFieldsMultiple(t *testing.T) {
	sg := &gorma.RelationalModelDefinition{}
	sg.RelationalFields = make(map[string]*gorma.RelationalFieldDefinition)
	f := makePK("Field1")
	sg.RelationalFields[f.FieldName] = f
	sg.PrimaryKeys = append(sg.PrimaryKeys, f)

	f2 := makePK("Field2")
	sg.RelationalFields[f2.FieldName] = f2
	sg.PrimaryKeys = append(sg.PrimaryKeys, f2)

	pkw := sg.PKWhereFields()

	if pkw != "field1,field2" {
		t.Errorf("Expected %s, got %s", "field1,field2", pkw)
	}

}

func TestPKUpdateFieldsSingle(t *testing.T) {
	sg := &gorma.RelationalModelDefinition{}
	sg.RelationalFields = make(map[string]*gorma.RelationalFieldDefinition)
	f := &gorma.RelationalFieldDefinition{}
	f.FieldName = dsl.SanitizeFieldName("ID")
	f.DatabaseFieldName = dsl.SanitizeDBFieldName(f.FieldName)
	f.Datatype = gorma.Integer
	f.PrimaryKey = true

	sg.RelationalFields[f.FieldName] = f
	sg.PrimaryKeys = append(sg.PrimaryKeys, f)

	pkw := sg.PKUpdateFields("model")

	if pkw != "model.ID" {
		t.Errorf("Expected %s, got %s", "model.ID", pkw)
	}

}

func TestPKUpdateFieldsMultiple(t *testing.T) {
	sg := &gorma.RelationalModelDefinition{}
	sg.RelationalFields = make(map[string]*gorma.RelationalFieldDefinition)
	f := makePK("Field1")
	sg.RelationalFields[f.FieldName] = f
	sg.PrimaryKeys = append(sg.PrimaryKeys, f)

	f2 := makePK("Field2")
	sg.RelationalFields[f2.FieldName] = f2
	sg.PrimaryKeys = append(sg.PrimaryKeys, f2)

	pkw := sg.PKUpdateFields("model")

	if pkw != "model.Field1,model.Field2" {
		t.Errorf("Expected %s, got %s", "model.Field1,model.Field2", pkw)
	}

}

func TestTableName(t *testing.T) {
	sg := &gorma.RelationalModelDefinition{}
	sg.ModelName = "status"
	if sg.TableName() != "statuses" {
		t.Errorf("Expected %s, got %s", "statuses", sg.TableName())
	}
	sg.ModelName = "study"
	if sg.TableName() != "studies" {
		t.Errorf("Expected %s, got %s", "studies", sg.TableName())
	}
	sg.ModelName = "user"
	if sg.TableName() != "users" {
		t.Errorf("Expected %s, got %s", "users", sg.TableName())
	}
}