public void Generate_can_output_drop_table_statement()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var generatedMigration
= codeGenerator.Generate(
"Migration",
new[] { new DropTableOperation("Customers") },
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Imports Microsoft.VisualBasic
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
DropTable(""Customers"")
End Sub
Public Overrides Sub Down()
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_can_output_move_procedure_statement()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var generatedMigration
= codeGenerator.Generate(
"Migration",
new[] { new MoveProcedureOperation("Insert_Customers", "foo") },
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Imports Microsoft.VisualBasic
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
MoveStoredProcedure(name := ""Insert_Customers"", newSchema := ""foo"")
End Sub
Public Overrides Sub Down()
MoveStoredProcedure(name := ""foo.Insert_Customers"", newSchema := Nothing)
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_can_output_drop_column()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var dropColumnOperation = new DropColumnOperation("Customers", "Foo");
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[] { dropColumnOperation },
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
DropColumn(""Customers"", ""Foo"")
End Sub
Public Overrides Sub Down()
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_can_output_alter_procedure_operations()
{
var alterProcedureOperation
= new AlterProcedureOperation("Foo", "SELECT ShinyHead\r\nFROM Pilkingtons");
alterProcedureOperation.Parameters.Add(
new ParameterModel(PrimitiveTypeKind.String)
{
Name = "P'",
DefaultValue = "Bar",
IsOutParameter = true
});
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[]
{
alterProcedureOperation
},
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Imports Microsoft.VisualBasic
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
AlterStoredProcedure(
""Foo"",
Function(p) New With
{
.P = p.String(name := ""P'"", defaultValue := ""Bar"", outParameter := True)
},
body :=
""SELECT ShinyHead"" & vbCrLf & _
""FROM Pilkingtons""
)
End Sub
Public Overrides Sub Down()
Throw New NotSupportedException(""" + Strings.ScaffoldSprocInDownNotSupported + @""")
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_can_output_create_procedure_operations()
{
var createProcedureOperation
= new CreateProcedureOperation("Foo", "SELECT ShinyHead\r\nFROM Pilkingtons");
createProcedureOperation.Parameters.Add(
new ParameterModel(PrimitiveTypeKind.String)
{
Name = "P'",
DefaultValue = "Bar",
IsOutParameter = true
});
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[]
{
createProcedureOperation
},
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Imports Microsoft.VisualBasic
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
CreateStoredProcedure(
""Foo"",
Function(p) New With
{
.P = p.String(name := ""P'"", defaultValue := ""Bar"", outParameter := True)
},
body :=
""SELECT ShinyHead"" & vbCrLf & _
""FROM Pilkingtons""
)
End Sub
Public Overrides Sub Down()
DropStoredProcedure(""Foo"")
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_can_process_null_source_model()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var generatedMigration = codeGenerator.Generate(
"Migration",
new MigrationOperation[] { },
null,
"Target",
"Foo",
"Bar");
Assert.Equal(
@"' <auto-generated />
Imports System.CodeDom.Compiler
Imports System.Data.Entity.Migrations
Imports System.Data.Entity.Migrations.Infrastructure
Imports System.Resources
Namespace Foo
<GeneratedCode(""EntityFramework.Migrations"", """ + typeof(DbContext).Assembly().GetInformationalVersion() + @""")>
Public NotInheritable Partial Class Bar
Implements IMigrationMetadata
Private ReadOnly Resources As New ResourceManager(GetType(Bar))
Private ReadOnly Property IMigrationMetadata_Id() As String Implements IMigrationMetadata.Id
Get
Return ""Migration""
End Get
End Property
Private ReadOnly Property IMigrationMetadata_Source() As String Implements IMigrationMetadata.Source
Get
Return Nothing
End Get
End Property
Private ReadOnly Property IMigrationMetadata_Target() As String Implements IMigrationMetadata.Target
Get
Return Resources.GetString(""Target"")
End Get
End Property
End Class
End Namespace
",
generatedMigration.DesignerCode);
Assert.Equal(1, generatedMigration.Resources.Count);
Assert.Equal("Target", generatedMigration.Resources["Target"]);
}
public void Generate_can_output_timestamp_column()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var createTableOperation = new CreateTableOperation("Customers");
var column = new ColumnModel(PrimitiveTypeKind.Binary)
{
Name = "Version",
IsTimestamp = true
};
createTableOperation.Columns.Add(column);
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[] { createTableOperation },
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Imports Microsoft.VisualBasic
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
CreateTable(
""Customers"",
Function(c) New With
{
.Version = c.Binary(timestamp := True)
})
End Sub
Public Overrides Sub Down()
DropTable(""Customers"")
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_should_scrub_class_name()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var generatedMigration = codeGenerator.Generate(
"Migration",
new MigrationOperation[] { },
"Source",
"Target",
"Foo",
"1$%^&DFDSH");
Assert.True(generatedMigration.UserCode.Contains("Class _1DFDSH"));
generatedMigration = codeGenerator.Generate(
"Migration",
new MigrationOperation[] { },
"Source",
"Target",
"Foo",
"While");
Assert.True(generatedMigration.UserCode.Contains("Class _While"));
}
public void Generate_can_output_composite_add_foreign_key()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var addForeignKeyOperation
= new AddForeignKeyOperation
{
DependentTable = "Orders",
PrincipalTable = "Customers"
};
addForeignKeyOperation.DependentColumns.Add("CustomerId1");
addForeignKeyOperation.DependentColumns.Add("CustomerId2");
addForeignKeyOperation.PrincipalColumns.Add("Id1");
addForeignKeyOperation.PrincipalColumns.Add("Id2");
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[] { addForeignKeyOperation },
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Imports Microsoft.VisualBasic
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
AddForeignKey(""Orders"", New String() { ""CustomerId1"", ""CustomerId2"" }, ""Customers"", New String() { ""Id1"", ""Id2"" })
End Sub
Public Overrides Sub Down()
DropForeignKey(""Orders"", New String() { ""CustomerId1"", ""CustomerId2"" }, ""Customers"")
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_can_output_drop_procedure_operations()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[]
{
new DropProcedureOperation("Foo"),
new DropTableOperation("Bar", new CreateTableOperation("Bar"))
},
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Imports Microsoft.VisualBasic
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
DropStoredProcedure(""Foo"")
DropTable(""Bar"")
End Sub
Public Overrides Sub Down()
CreateTable(
""Bar"",
Function(c) New With
{
})
Throw New NotSupportedException(""" + Strings.ScaffoldSprocInDownNotSupported + @""")
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_can_output_add_primary_key_with_explicit_name()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var addPrimaryKeyOperation
= new AddPrimaryKeyOperation
{
Table = "T",
Name = "PK"
};
addPrimaryKeyOperation.Columns.Add("c1");
addPrimaryKeyOperation.Columns.Add("c2");
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[] { addPrimaryKeyOperation },
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Imports Microsoft.VisualBasic
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
AddPrimaryKey(""T"", New String() { ""c1"", ""c2"" }, name := ""PK"")
End Sub
Public Overrides Sub Down()
DropPrimaryKey(""T"", name := ""PK"")
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_can_output_drop_primary_key_with_explicit_name()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var dropPrimaryKeyOperation
= new DropPrimaryKeyOperation
{
Table = "T",
Name = "PK"
};
dropPrimaryKeyOperation.Columns.Add("c1");
dropPrimaryKeyOperation.Columns.Add("c2");
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[] { dropPrimaryKeyOperation },
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
DropPrimaryKey(""T"", ""PK"")
End Sub
Public Overrides Sub Down()
AddPrimaryKey(""T"", New String() { ""c1"", ""c2"" }, name := ""PK"")
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_can_output_rename_procedure_operation()
{
var renameProcedureOperation
= new RenameProcedureOperation("Foo", "Bar");
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[]
{
renameProcedureOperation
},
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Imports Microsoft.VisualBasic
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
RenameStoredProcedure(name := ""Foo"", newName := ""Bar"")
End Sub
Public Overrides Sub Down()
RenameStoredProcedure(name := ""Bar"", newName := ""Foo"")
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_should_not_produce_lines_that_are_too_long_for_the_compiler()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[] { },
new string('a', 10000),
"Target",
"Foo",
"Bar");
using (var stringReader = new StringReader(generatedMigration.DesignerCode))
{
string line;
while ((line = stringReader.ReadLine()) != null)
{
Assert.True(line.Length <= 1100);
}
}
}
public void Generate_should_not_produce_lines_that_are_too_long_for_the_compiler()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[] { },
new string('a', 10000),
"Target",
"Foo",
"Bar");
using (var stringReader = new StringReader(generatedMigration.DesignerCode))
{
string line;
while ((line = stringReader.ReadLine()) != null)
{
Assert.True(line.Length <= 1100);
}
}
}
public void Generate_can_process_null_source_model()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var generatedMigration = codeGenerator.Generate(
"Migration",
new MigrationOperation[] { },
null,
"Target",
"Foo",
"Bar");
Assert.Equal(
@"' <auto-generated />
Imports System.CodeDom.Compiler
Imports System.Data.Entity.Migrations
Imports System.Data.Entity.Migrations.Infrastructure
Imports System.Resources
Namespace Foo
<GeneratedCode(""EntityFramework.Migrations"", """ + typeof(DbContext).Assembly.GetInformationalVersion() + @""")>
Public NotInheritable Partial Class Bar
Implements IMigrationMetadata
Private ReadOnly Resources As New ResourceManager(GetType(Bar))
Private ReadOnly Property IMigrationMetadata_Id() As String Implements IMigrationMetadata.Id
Get
Return ""Migration""
End Get
End Property
Private ReadOnly Property IMigrationMetadata_Source() As String Implements IMigrationMetadata.Source
Get
Return Nothing
End Get
End Property
Private ReadOnly Property IMigrationMetadata_Target() As String Implements IMigrationMetadata.Target
Get
Return Resources.GetString(""Target"")
End Get
End Property
End Class
End Namespace
",
generatedMigration.DesignerCode);
Assert.Equal(1, generatedMigration.Resources.Count);
Assert.Equal("Target", generatedMigration.Resources["Target"]);
}
public void Generate_can_output_drop_procedure_operations()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[]
{
new DropProcedureOperation("Foo"),
new DropTableOperation("Bar", new CreateTableOperation("Bar"))
},
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Imports Microsoft.VisualBasic
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
DropStoredProcedure(""Foo"")
DropTable(""Bar"")
End Sub
Public Overrides Sub Down()
CreateTable(
""Bar"",
Function(c) New With
{
})
Throw New NotSupportedException(""Scaffolding create procedure operations is not supported in down methods."")
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_can_output_drop_column()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var dropColumnOperation = new DropColumnOperation("Customers", "Foo");
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[] { dropColumnOperation },
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Imports Microsoft.VisualBasic
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
DropColumn(""Customers"", ""Foo"")
End Sub
Public Overrides Sub Down()
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Can_generate_AlterTableAnnotations_with_annotations()
{
var operation = new AlterTableOperation(
"Customers",
new Dictionary<string, AnnotationValues>
{
{ "AT1", new AnnotationValues(null, "VT1") },
{
CollationAttribute.AnnotationName,
new AnnotationValues(
new CollationAttribute("At a reasonable volume..."),
new CollationAttribute("While I'm collating..."))
},
{ "AT2", new AnnotationValues(null, "VT2") }
});
var idColumn = new ColumnModel(PrimitiveTypeKind.Int32)
{
Name = "I.d",
IsNullable = true,
IsIdentity = true,
Annotations =
new Dictionary<string, AnnotationValues>
{
{ "A1", new AnnotationValues(null, "V1") },
{ "A2", new AnnotationValues(null, "V2") }
}
};
operation.Columns.Add(idColumn);
operation.Columns.Add(
new ColumnModel(PrimitiveTypeKind.String)
{
Name = "Name",
IsNullable = false,
Annotations =
new Dictionary<string, AnnotationValues>
{
{
CollationAttribute.AnnotationName,
new AnnotationValues(
new CollationAttribute("At a reasonable volume..."),
new CollationAttribute("While I'm collating..."))
}
}
});
var operations = new[] { operation };
var generator = new VisualBasicMigrationCodeGenerator();
generator.AnnotationGenerators[CollationAttribute.AnnotationName] = () => new CollationCSharpCodeGenerator();
var generatedMigration = generator.Generate("Migration", operations, "Source", "Target", "MyNamespace", "MyMigration");
Assert.Equal(
@"Imports System
Imports System.Collections.Generic
Imports System.Data.Entity.Infrastructure.Annotations
Imports System.Data.Entity.Migrations
Imports System.Data.Entity.TestHelpers
Imports Microsoft.VisualBasic
Namespace MyNamespace
Public Partial Class MyMigration
Inherits DbMigration
Public Overrides Sub Up()
AlterTableAnnotations(
""Customers"",
Function(c) New With
{
.Id = c.Int(name := ""I.d"", identity := True,
annotations := New Dictionary(Of String, AnnotationValues)() From _
{
{
""A1"",
New AnnotationValues(oldValue := Nothing, newValue := ""V1"")
},
{
""A2"",
New AnnotationValues(oldValue := Nothing, newValue := ""V2"")
}
}),
.Name = c.String(nullable := False,
annotations := New Dictionary(Of String, AnnotationValues)() From _
{
{
""Collation"",
New AnnotationValues(oldValue := new CollationAttribute(""At a reasonable volume...""), newValue := new CollationAttribute(""While I'm collating...""))
}
})
},
annotations := New Dictionary(Of String, AnnotationValues)() From _
{
{
""AT1"",
New AnnotationValues(oldValue := Nothing, newValue := ""VT1"")
},
{
""AT2"",
New AnnotationValues(oldValue := Nothing, newValue := ""VT2"")
},
{
""Collation"",
New AnnotationValues(oldValue := new CollationAttribute(""At a reasonable volume...""), newValue := new CollationAttribute(""While I'm collating...""))
}
})
End Sub
Public Overrides Sub Down()
AlterTableAnnotations(
""Customers"",
Function(c) New With
{
.Id = c.Int(name := ""I.d"", identity := True,
annotations := New Dictionary(Of String, AnnotationValues)() From _
{
{
""A1"",
New AnnotationValues(oldValue := Nothing, newValue := ""V1"")
},
{
""A2"",
New AnnotationValues(oldValue := Nothing, newValue := ""V2"")
}
}),
.Name = c.String(nullable := False,
annotations := New Dictionary(Of String, AnnotationValues)() From _
{
{
""Collation"",
New AnnotationValues(oldValue := new CollationAttribute(""At a reasonable volume...""), newValue := new CollationAttribute(""While I'm collating...""))
}
})
},
annotations := New Dictionary(Of String, AnnotationValues)() From _
{
{
""AT1"",
New AnnotationValues(oldValue := ""VT1"", newValue := Nothing)
},
{
""AT2"",
New AnnotationValues(oldValue := ""VT2"", newValue := Nothing)
},
{
""Collation"",
New AnnotationValues(oldValue := new CollationAttribute(""While I'm collating...""), newValue := new CollationAttribute(""At a reasonable volume...""))
}
})
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_can_output_rename_procedure_operation()
{
var renameProcedureOperation
= new RenameProcedureOperation("Foo", "Bar");
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[]
{
renameProcedureOperation
},
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Imports Microsoft.VisualBasic
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
RenameStoredProcedure(name := ""Foo"", newName := ""Bar"")
End Sub
Public Overrides Sub Down()
RenameStoredProcedure(name := ""Bar"", newName := ""Foo"")
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void GenerateAnnotation_checks_arguments()
{
var generator = new VisualBasicMigrationCodeGenerator();
Assert.Equal(
"alterTableOperation",
Assert.Throws<ArgumentNullException>(() => generator.Generate(null, new IndentedTextWriter(new Mock<TextWriter>().Object))).ParamName);
Assert.Equal(
"writer",
Assert.Throws<ArgumentNullException>(() => generator.Generate(new AlterTableOperation("N", null), null)).ParamName);
}
public void Can_generate_AddColumn_with_custom_annotation_code_gen()
{
var operations = new[]
{
new AddColumnOperation(
"MyTable",
new ColumnModel(PrimitiveTypeKind.String)
{
Name = "MyColumn",
IsFixedLength = true,
Annotations =
new Dictionary<string, AnnotationValues>
{
{
CollationAttribute.AnnotationName,
new AnnotationValues(
new CollationAttribute("At a reasonable volume..."),
new CollationAttribute("While I'm collating..."))
}
}
},
false),
};
var generator = new VisualBasicMigrationCodeGenerator();
generator.AnnotationGenerators[CollationAttribute.AnnotationName] = () => new CollationCSharpCodeGenerator();
var generatedMigration = generator.Generate("Migration", operations, "Source", "Target", "MyNamespace", "MyMigration");
Assert.Equal(
@"Imports System
Imports System.Collections.Generic
Imports System.Data.Entity.Infrastructure.Annotations
Imports System.Data.Entity.Migrations
Imports System.Data.Entity.TestHelpers
Imports Microsoft.VisualBasic
Namespace MyNamespace
Public Partial Class MyMigration
Inherits DbMigration
Public Overrides Sub Up()
AddColumn(""MyTable"", ""MyColumn"", Function(c) c.String(fixedLength := true,
annotations := New Dictionary(Of String, AnnotationValues)() From _
{
{
""Collation"",
New AnnotationValues(oldValue := new CollationAttribute(""At a reasonable volume...""), newValue := new CollationAttribute(""While I'm collating...""))
}
}))
End Sub
Public Overrides Sub Down()
DropColumn(""MyTable"", ""MyColumn"",
removedAnnotations := New Dictionary(Of String, Object)() From _
{
{ ""Collation"", new CollationAttribute(""While I'm collating..."") }
})
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_should_scrub_class_name()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var generatedMigration = codeGenerator.Generate(
"Migration",
new MigrationOperation[] { },
"Source",
"Target",
"Foo",
"1$%^&DFDSH");
Assert.True(generatedMigration.UserCode.Contains("Class _1DFDSH"));
generatedMigration = codeGenerator.Generate(
"Migration",
new MigrationOperation[] { },
"Source",
"Target",
"Foo",
"While");
Assert.True(generatedMigration.UserCode.Contains("Class _While"));
}
public void Generate_can_output_create_table_statement()
{
var createTableOperation = new CreateTableOperation("Customers");
var idColumn = new ColumnModel(PrimitiveTypeKind.Int32)
{
Name = "Id",
IsNullable = true,
IsIdentity = true
};
createTableOperation.Columns.Add(idColumn);
createTableOperation.Columns.Add(
new ColumnModel(PrimitiveTypeKind.String)
{
Name = "Name",
IsNullable = false
});
createTableOperation.PrimaryKey = new AddPrimaryKeyOperation
{
Name = "MyPK"
};
createTableOperation.PrimaryKey.Columns.Add(idColumn.Name);
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var addForeignKeyOperation = new AddForeignKeyOperation
{
DependentTable = "Customers",
PrincipalTable = "Blogs",
CascadeDelete = true
};
addForeignKeyOperation.DependentColumns.Add("Blog_Id");
addForeignKeyOperation.PrincipalColumns.Add("Id");
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[]
{
createTableOperation,
addForeignKeyOperation,
addForeignKeyOperation.CreateCreateIndexOperation()
},
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
CreateTable(
""Customers"",
Function(c) New With
{
.Id = c.Int(identity := True),
.Name = c.String(nullable := False)
}) _
.PrimaryKey(Function(t) t.Id, name := ""MyPK"") _
.ForeignKey(""Blogs"", Function(t) t.Blog_Id, cascadeDelete := True) _
.Index(Function(t) t.Blog_Id)
End Sub
Public Overrides Sub Down()
DropIndex(""Customers"", New String() { ""Blog_Id"" })
DropForeignKey(""Customers"", ""Blog_Id"", ""Blogs"")
DropTable(""Customers"")
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
Assert.Equal(
@"' <auto-generated />
Imports System.Data.Entity.Migrations
Imports System.Data.Entity.Migrations.Infrastructure
Imports System.Resources
Namespace Foo
Public NotInheritable Partial Class Bar
Implements IMigrationMetadata
Private ReadOnly Resources As New ResourceManager(GetType(Bar))
Private ReadOnly Property IMigrationMetadata_Id() As String Implements IMigrationMetadata.Id
Get
Return ""Migration""
End Get
End Property
Private ReadOnly Property IMigrationMetadata_Source() As String Implements IMigrationMetadata.Source
Get
Return Resources.GetString(""Source"")
End Get
End Property
Private ReadOnly Property IMigrationMetadata_Target() As String Implements IMigrationMetadata.Target
Get
Return Resources.GetString(""Target"")
End Get
End Property
End Class
End Namespace
",
generatedMigration.DesignerCode);
Assert.Equal("vb", generatedMigration.Language);
Assert.Equal(2, generatedMigration.Resources.Count);
Assert.Equal("Source", generatedMigration.Resources["Source"]);
Assert.Equal("Target", generatedMigration.Resources["Target"]);
}
public void Generate_can_output_timestamp_column()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var createTableOperation = new CreateTableOperation("Customers");
var column = new ColumnModel(PrimitiveTypeKind.Binary)
{
Name = "Version",
IsTimestamp = true
};
createTableOperation.Columns.Add(column);
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[] { createTableOperation },
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
CreateTable(
""Customers"",
Function(c) New With
{
.Version = c.Binary(timestamp := True)
})
End Sub
Public Overrides Sub Down()
DropTable(""Customers"")
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_can_output_add_primary_key_with_non_clustered_index()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var addPrimaryKeyOperation
= new AddPrimaryKeyOperation
{
Table = "T",
Name = "PK",
IsClustered = false
};
addPrimaryKeyOperation.Columns.Add("c1");
addPrimaryKeyOperation.Columns.Add("c2");
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[] { addPrimaryKeyOperation },
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Imports Microsoft.VisualBasic
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
AddPrimaryKey(""T"", New String() { ""c1"", ""c2"" }, name := ""PK"", clustered := False)
End Sub
Public Overrides Sub Down()
DropPrimaryKey(""T"", name := ""PK"")
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_can_output_composite_add_foreign_key()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var addForeignKeyOperation
= new AddForeignKeyOperation
{
DependentTable = "Orders",
PrincipalTable = "Customers"
};
addForeignKeyOperation.DependentColumns.Add("CustomerId1");
addForeignKeyOperation.DependentColumns.Add("CustomerId2");
addForeignKeyOperation.PrincipalColumns.Add("Id1");
addForeignKeyOperation.PrincipalColumns.Add("Id2");
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[] { addForeignKeyOperation },
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
AddForeignKey(""Orders"", New String() { ""CustomerId1"", ""CustomerId2"" }, ""Customers"", New String() { ""Id1"", ""Id2"" })
End Sub
Public Overrides Sub Down()
DropForeignKey(""Orders"", New String() { ""CustomerId1"", ""CustomerId2"" }, ""Customers"")
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_can_output_create_table_statement()
{
var createTableOperation = new CreateTableOperation("Customers");
var idColumn = new ColumnModel(PrimitiveTypeKind.Int32)
{
Name = "I.d",
IsNullable = true,
IsIdentity = true
};
createTableOperation.Columns.Add(idColumn);
createTableOperation.Columns.Add(
new ColumnModel(PrimitiveTypeKind.String)
{
Name = "Name",
IsNullable = false
});
createTableOperation.PrimaryKey = new AddPrimaryKeyOperation
{
Name = "MyPK"
};
createTableOperation.PrimaryKey.Columns.Add(idColumn.Name);
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var addForeignKeyOperation = new AddForeignKeyOperation
{
DependentTable = "Customers",
PrincipalTable = "Blogs",
CascadeDelete = true
};
addForeignKeyOperation.DependentColumns.Add("Blog.Id");
addForeignKeyOperation.PrincipalColumns.Add("Id");
var generatedMigration
= codeGenerator.Generate(
"Migration",
new MigrationOperation[]
{
createTableOperation,
addForeignKeyOperation,
addForeignKeyOperation.CreateCreateIndexOperation()
},
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Imports Microsoft.VisualBasic
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
CreateTable(
""Customers"",
Function(c) New With
{
.Id = c.Int(name := ""I.d"", identity := True),
.Name = c.String(nullable := False)
}) _
.PrimaryKey(Function(t) t.Id, name := ""MyPK"") _
.ForeignKey(""Blogs"", Function(t) t.BlogId, cascadeDelete := True) _
.Index(Function(t) t.BlogId)
End Sub
Public Overrides Sub Down()
DropIndex(""Customers"", New String() { ""Blog.Id"" })
DropForeignKey(""Customers"", ""Blog.Id"", ""Blogs"")
DropTable(""Customers"")
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
Assert.Equal(
@"' <auto-generated />
Imports System.CodeDom.Compiler
Imports System.Data.Entity.Migrations
Imports System.Data.Entity.Migrations.Infrastructure
Imports System.Resources
Namespace Foo
<GeneratedCode(""EntityFramework.Migrations"", """ + typeof(DbContext).Assembly().GetInformationalVersion() + @""")>
Public NotInheritable Partial Class Bar
Implements IMigrationMetadata
Private ReadOnly Resources As New ResourceManager(GetType(Bar))
Private ReadOnly Property IMigrationMetadata_Id() As String Implements IMigrationMetadata.Id
Get
Return ""Migration""
End Get
End Property
Private ReadOnly Property IMigrationMetadata_Source() As String Implements IMigrationMetadata.Source
Get
Return Resources.GetString(""Source"")
End Get
End Property
Private ReadOnly Property IMigrationMetadata_Target() As String Implements IMigrationMetadata.Target
Get
Return Resources.GetString(""Target"")
End Get
End Property
End Class
End Namespace
",
generatedMigration.DesignerCode);
Assert.Equal("vb", generatedMigration.Language);
Assert.Equal(2, generatedMigration.Resources.Count);
Assert.Equal("Source", generatedMigration.Resources["Source"]);
Assert.Equal("Target", generatedMigration.Resources["Target"]);
}
public void Generate_can_process_null_namespace()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var generatedMigration = codeGenerator.Generate(
"Migration",
new MigrationOperation[0],
null,
"Target",
null,
"Bar");
Assert.Equal(
@"' <auto-generated />
Imports System.Data.Entity.Migrations
Imports System.Data.Entity.Migrations.Infrastructure
Imports System.Resources
Public NotInheritable Partial Class Bar
Implements IMigrationMetadata
Private ReadOnly Resources As New ResourceManager(GetType(Bar))
Private ReadOnly Property IMigrationMetadata_Id() As String Implements IMigrationMetadata.Id
Get
Return ""Migration""
End Get
End Property
Private ReadOnly Property IMigrationMetadata_Source() As String Implements IMigrationMetadata.Source
Get
Return Nothing
End Get
End Property
Private ReadOnly Property IMigrationMetadata_Target() As String Implements IMigrationMetadata.Target
Get
Return Resources.GetString(""Target"")
End Get
End Property
End Class
",
generatedMigration.DesignerCode);
Assert.Equal(1, generatedMigration.Resources.Count);
Assert.Equal("Target", generatedMigration.Resources["Target"]);
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
End Sub
Public Overrides Sub Down()
End Sub
End Class
",
generatedMigration.UserCode);
}
public void Can_generate_AlterColumn_for_added_removed_and_changed_annotations()
{
var operations = new[]
{
new AlterColumnOperation(
"MyTable",
new ColumnModel(PrimitiveTypeKind.Int32)
{
Name = "MyColumn",
IsFixedLength = true,
Annotations =
new Dictionary<string, AnnotationValues>
{
{ "A2", new AnnotationValues(null, "V2") },
{ "A3", new AnnotationValues(null, "V3") },
{ "A1", new AnnotationValues(null, "V1") },
{ "A8", new AnnotationValues("V8A", "V8B") },
{ "A7", new AnnotationValues("V7A", "V7B") },
{ "A9", new AnnotationValues("V9A", "V9B") },
{ "A5", new AnnotationValues("V5", null) },
{ "A4", new AnnotationValues("V4", null) },
{ "A6", new AnnotationValues("V6", null) }
}
},
false),
};
var generator = new VisualBasicMigrationCodeGenerator();
var generatedMigration = generator.Generate("Migration", operations, "Source", "Target", "MyNamespace", "MyMigration");
Assert.Equal(
@"Imports System
Imports System.Collections.Generic
Imports System.Data.Entity.Infrastructure.Annotations
Imports System.Data.Entity.Migrations
Imports Microsoft.VisualBasic
Namespace MyNamespace
Public Partial Class MyMigration
Inherits DbMigration
Public Overrides Sub Up()
AlterColumn(""MyTable"", ""MyColumn"", Function(c) c.Int(fixedLength := true,
annotations := New Dictionary(Of String, AnnotationValues)() From _
{
{
""A1"",
New AnnotationValues(oldValue := Nothing, newValue := ""V1"")
},
{
""A2"",
New AnnotationValues(oldValue := Nothing, newValue := ""V2"")
},
{
""A3"",
New AnnotationValues(oldValue := Nothing, newValue := ""V3"")
},
{
""A4"",
New AnnotationValues(oldValue := ""V4"", newValue := Nothing)
},
{
""A5"",
New AnnotationValues(oldValue := ""V5"", newValue := Nothing)
},
{
""A6"",
New AnnotationValues(oldValue := ""V6"", newValue := Nothing)
},
{
""A7"",
New AnnotationValues(oldValue := ""V7A"", newValue := ""V7B"")
},
{
""A8"",
New AnnotationValues(oldValue := ""V8A"", newValue := ""V8B"")
},
{
""A9"",
New AnnotationValues(oldValue := ""V9A"", newValue := ""V9B"")
}
}))
End Sub
Public Overrides Sub Down()
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_can_output_drop_table_statement()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var generatedMigration
= codeGenerator.Generate(
"Migration",
new[] { new DropTableOperation("Customers") },
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
DropTable(""Customers"")
End Sub
Public Overrides Sub Down()
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_can_output_move_procedure_statement()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var generatedMigration
= codeGenerator.Generate(
"Migration",
new[] { new MoveProcedureOperation("Insert_Customers", "foo") },
"Source",
"Target",
"Foo",
"Bar");
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Imports Microsoft.VisualBasic
Namespace Foo
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
MoveStoredProcedure(name := ""Insert_Customers"", newSchema := ""foo"")
End Sub
Public Overrides Sub Down()
MoveStoredProcedure(name := ""foo.Insert_Customers"", newSchema := Nothing)
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}
public void Generate_can_process_null_namespace()
{
var codeGenerator = new VisualBasicMigrationCodeGenerator();
var generatedMigration = codeGenerator.Generate(
"Migration",
new MigrationOperation[0],
null,
"Target",
null,
"Bar");
Assert.Equal(
@"' <auto-generated />
Imports System.Data.Entity.Migrations
Imports System.Data.Entity.Migrations.Infrastructure
Imports System.Resources
Public NotInheritable Partial Class Bar
Implements IMigrationMetadata
Private ReadOnly Resources As New ResourceManager(GetType(Bar))
Private ReadOnly Property IMigrationMetadata_Id() As String Implements IMigrationMetadata.Id
Get
Return ""Migration""
End Get
End Property
Private ReadOnly Property IMigrationMetadata_Source() As String Implements IMigrationMetadata.Source
Get
Return Nothing
End Get
End Property
Private ReadOnly Property IMigrationMetadata_Target() As String Implements IMigrationMetadata.Target
Get
Return Resources.GetString(""Target"")
End Get
End Property
End Class
",
generatedMigration.DesignerCode);
Assert.Equal(1, generatedMigration.Resources.Count);
Assert.Equal("Target", generatedMigration.Resources["Target"]);
Assert.Equal(
@"Imports System
Imports System.Data.Entity.Migrations
Public Partial Class Bar
Inherits DbMigration
Public Overrides Sub Up()
End Sub
Public Overrides Sub Down()
End Sub
End Class
",
generatedMigration.UserCode);
}
public void Can_generate_AddColumn_with_annotations()
{
var operations = new[]
{
new AddColumnOperation(
"MyTable",
new ColumnModel(PrimitiveTypeKind.String)
{
Name = "MyColumn",
IsFixedLength = true,
Annotations =
new Dictionary<string, AnnotationValues>
{
{ "A3", new AnnotationValues(null, "V3") },
{ "A1", new AnnotationValues(null, "V1") },
}
},
false),
};
var generator = new VisualBasicMigrationCodeGenerator();
var generatedMigration = generator.Generate("Migration", operations, "Source", "Target", "MyNamespace", "MyMigration");
Assert.Equal(
@"Imports System
Imports System.Collections.Generic
Imports System.Data.Entity.Infrastructure.Annotations
Imports System.Data.Entity.Migrations
Imports Microsoft.VisualBasic
Namespace MyNamespace
Public Partial Class MyMigration
Inherits DbMigration
Public Overrides Sub Up()
AddColumn(""MyTable"", ""MyColumn"", Function(c) c.String(fixedLength := true,
annotations := New Dictionary(Of String, AnnotationValues)() From _
{
{
""A1"",
New AnnotationValues(oldValue := Nothing, newValue := ""V1"")
},
{
""A3"",
New AnnotationValues(oldValue := Nothing, newValue := ""V3"")
}
}))
End Sub
Public Overrides Sub Down()
DropColumn(""MyTable"", ""MyColumn"",
removedAnnotations := New Dictionary(Of String, Object)() From _
{
{ ""A1"", ""V1"" },
{ ""A3"", ""V3"" }
})
End Sub
End Class
End Namespace
",
generatedMigration.UserCode);
}