// AddColumn adds a ConstraintNode from the nodes package to the AST to apply to a column.
func (self *AlterManager) AddConstraint(columns []interface{}, kind sql.Constraint, options ...interface{}) *AlterManager {
for _, column := range columns {
if _, ok := column.(string); ok {
column = nodes.UnqualifiedColumn(column)
}
}
var node interface{}
switch kind {
case sql.NotNull:
node = nodes.NotNull(columns, options...)
case sql.Unique:
node = nodes.Unique(columns, options...)
case sql.PrimaryKey:
node = nodes.PrimaryKey(columns, options...)
case sql.ForeignKey:
node = nodes.ForeignKey(columns, options...)
case sql.Check:
node = nodes.Check(columns, options...)
case sql.Default:
node = nodes.Default(columns, options...)
default:
node = nodes.Constraint(columns, options...)
}
self.Tree.Constraints = append(self.Tree.Constraints, node)
return self
}
func TestVisitForeignKey(t *testing.T) {
fkey := nodes.ForeignKey([]interface{}{nodes.UnqualifiedColumn("column")})
fkey.Options = append(fkey.Options, nodes.Relation("table"))
expected := `ADD FOREIGN KEY("column") REFERENCES "table"`
if got, _ := sql.Accept(fkey); expected != got {
t.Errorf("TestVisitForeignKey was expected to return %s, got %s", expected, got)
}
}