private ImportStatement createInsertStatement(SourceDataRow sourceDataRow)
{
TableMappingOrderer tableMappingOrderer = new TableMappingOrderer(config.TableMappings);
TableMapping[] orderedTableMappings = tableMappingOrderer.OrderTableMappings();
StatementSetupPart statementSetupPart = new StatementSetupPart(config);
string setupPart = string.Format("{0}\n{1}", statementSetupPart.GetDatabasePart(), statementSetupPart.GetWarningsPart());
StatementTransactionPart statementTransactionPart = new StatementTransactionPart(config);
string transactionStartPart = statementTransactionPart.GetTransactionStartPart();
string transactionEndPart = statementTransactionPart.GetTransactionEndPart();
List<string> statementTableVariableParts = new List<string>();
List<string> statmentBodyParts = new List<string>();
foreach (TableMapping tableMapping in orderedTableMappings)
{
StatementTableMappingPart statementTableMappingPart = new StatementTableMappingPart(tableMapping, sourceDataRow);
statementTableVariableParts.Add(statementTableMappingPart.GetTableVariablePart());
statmentBodyParts.Add(statementTableMappingPart.GetStatementBodyPart());
}
string importStatementStructure = "{0}\n{1}\n{2}\n{3}\n{4}";
string importStatement = string.Format(importStatementStructure, setupPart, transactionStartPart,
string.Join("\n", statementTableVariableParts), string.Join("\n", statmentBodyParts), transactionEndPart);
return new ImportStatement(importStatement, sourceDataRow.RowReference);
}
public void TransactionEndPartTest()
{
ImportConfiguration config = getTestImportConfig();
StatementTransactionPart part = new StatementTransactionPart(config);
config.ErrorHandling.ImportAsTransaction = true;
string[] parts = part.GetTransactionEndPart()
.Split(new string[] { "\n", "\t" }, StringSplitOptions.None).Where(s => s.Length > 0).ToArray();
Assert.AreEqual(12, parts.Length);
Assert.AreEqual("COMMIT TRANSACTION [ImportTransaction]", parts[0]);
Assert.AreEqual("END TRY", parts[1]);
Assert.AreEqual("BEGIN CATCH", parts[2]);
Assert.AreEqual("ROLLBACK TRANSACTION [ImportTransaction]", parts[3]);
Assert.AreEqual("DECLARE @ErrorMessage NVARCHAR(MAX)", parts[4]);
Assert.AreEqual("DECLARE @ErrorSeverity INT", parts[5]);
Assert.AreEqual("DECLARE @ErrorState INT", parts[6]);
Assert.AreEqual("SET @ErrorMessage = ERROR_MESSAGE()", parts[7]);
Assert.AreEqual("SET @ErrorSeverity = ERROR_SEVERITY()", parts[8]);
Assert.AreEqual("SET @ErrorState = ERROR_STATE()", parts[9]);
Assert.AreEqual("RAISERROR (@ErrorMessage, @ErrorSeverity, @ErrorState)", parts[10]);
Assert.AreEqual("END CATCH", parts[11]);
config.ErrorHandling.ImportAsTransaction = false;
Assert.AreEqual("", part.GetTransactionEndPart());
}
public void TransactionStartPartTest()
{
ImportConfiguration config = getTestImportConfig();
StatementTransactionPart part = new StatementTransactionPart(config);
config.ErrorHandling.ImportAsTransaction = true;
Assert.AreEqual("BEGIN TRANSACTION [ImportTransaction]\nBEGIN TRY\n", part.GetTransactionStartPart());
config.ErrorHandling.ImportAsTransaction = false;
Assert.AreEqual("", part.GetTransactionStartPart());
}
public void StatementCreatorOrderTest()
{
DBTable table = new DBTable("dbo", "TestTable");
DBColumn col1 = new DBColumn(table, "TestCol1", true, DBDatatype.integer);
DBColumn col2 = new DBColumn(table, "TestCol2", false, DBDatatype.nvarchar);
DBColumn col3 = new DBColumn(table, "TestCol3", false, DBDatatype.integer);
table.Columns = new List <DBColumn>()
{
col1, col2, col3
};
Database db = new Database("TestDB", new List <DBTable>()
{
table
});
TableMapping sourceTablemapping = new TableMapping(new DBTable("dbo", "TestTable2"), TableMappingImportType.Insert, null);
ColumnMapping colMap1 = new TableColumnMapping(sourceTablemapping, col1, col1, ColumnUse.Where);
ColumnMapping colMap2 = new LiteralColumnMapping("2", LiteralType.String, col2, ColumnUse.Where);
ColumnMapping colMap3 = new LiteralColumnMapping("3", LiteralType.String, col2, ColumnUse.Set);
TableMapping tableMapping = new TableMapping(table, TableMappingImportType.Update, new ColumnMapping[] { colMap1, colMap2, colMap3 });
ErrorHandling errorHandling = new ErrorHandling();
ImportConfiguration config = new ImportConfiguration(new TableMapping[] { tableMapping }, null, "TestDB", errorHandling);
SourceDataEntry[] entries = new SourceDataEntry[] { SourceDataEntry.CreateDataEntry("", DataType.String, "") };
SourceDataRow[] rows = new SourceDataRow[] { new SourceDataRow(entries, "0") };
SourceDataTable dt = new SourceDataTable(rows, new string[] { "" });
SQLServerStatementCreator statementCreator = new SQLServerStatementCreator(config, dt);
ImportStatement statement = statementCreator.CreateStatement(0);
ImportStatement[] statements = statementCreator.CreateStatements();
Assert.AreEqual(1, statements.Length);
Assert.AreEqual(statement.RowReference, statements[0].RowReference);
Assert.AreEqual(statement.SqlStatement, statements[0].SqlStatement);
string[] lines = statement.SqlStatement
.Split(new string[] { "\n" }, StringSplitOptions.None).Where(s => s.Length > 0).ToArray();
StatementSetupPart setupPart = new StatementSetupPart(config);
Assert.AreEqual(setupPart.GetDatabasePart(), lines[0]);
Assert.AreEqual(setupPart.GetWarningsPart(), lines[1]);
StatementTransactionPart transPart = new StatementTransactionPart(config);
string[] transStartPart = transPart.GetTransactionStartPart()
.Split(new string[] { "\n" }, StringSplitOptions.None).Where(s => s.Length > 0).ToArray();;
Assert.AreEqual(2, transStartPart.Length);
Assert.AreEqual(transStartPart[0], lines[2]);
Assert.AreEqual(transStartPart[1], lines[3]);
StatementTableMappingPart tmParts = new StatementTableMappingPart(tableMapping, dt.GetDataRow(0));
string variablePart = tmParts.GetTableVariablePart().Replace("\n", "");
Assert.AreEqual(variablePart, lines[4]);
string[] bodyParts = tmParts.GetStatementBodyPart()
.Split(new string[] { "\n" }, StringSplitOptions.None).Where(s => s.Length > 0).ToArray();
Assert.AreEqual(4, bodyParts.Length);
Assert.AreEqual(bodyParts[0], lines[5]);
Assert.AreEqual(bodyParts[1], lines[6]);
Assert.AreEqual(bodyParts[2], lines[7]);
Assert.AreEqual(bodyParts[3], lines[8]);
string[] transEndPart = transPart.GetTransactionEndPart()
.Split(new string[] { "\n" }, StringSplitOptions.None).Where(s => s.Length > 0).ToArray();
Assert.AreEqual(12, transEndPart.Length);
Assert.AreEqual(transEndPart[0], lines[9]);
Assert.AreEqual(transEndPart[1], lines[10]);
Assert.AreEqual(transEndPart[2], lines[11]);
Assert.AreEqual(transEndPart[3], lines[12]);
Assert.AreEqual(transEndPart[4], lines[13]);
Assert.AreEqual(transEndPart[5], lines[14]);
Assert.AreEqual(transEndPart[6], lines[15]);
Assert.AreEqual(transEndPart[7], lines[16]);
Assert.AreEqual(transEndPart[8], lines[17]);
Assert.AreEqual(transEndPart[9], lines[18]);
Assert.AreEqual(transEndPart[10], lines[19]);
Assert.AreEqual(transEndPart[11], lines[20]);
}
public void TransactionStartPartTest()
{
ImportConfiguration config = getTestImportConfig();
StatementTransactionPart part = new StatementTransactionPart(config);
config.ErrorHandling.ImportAsTransaction = true;
Assert.AreEqual("BEGIN TRANSACTION [ImportTransaction]\nBEGIN TRY\n", part.GetTransactionStartPart());
config.ErrorHandling.ImportAsTransaction = false;
Assert.AreEqual("", part.GetTransactionStartPart());
}
public void TransactionEndPartTest()
{
ImportConfiguration config = getTestImportConfig();
StatementTransactionPart part = new StatementTransactionPart(config);
config.ErrorHandling.ImportAsTransaction = true;
string[] parts = part.GetTransactionEndPart()
.Split(new string[] { "\n", "\t" }, StringSplitOptions.None).Where(s => s.Length > 0).ToArray();
Assert.AreEqual(12, parts.Length);
Assert.AreEqual("COMMIT TRANSACTION [ImportTransaction]", parts[0]);
Assert.AreEqual("END TRY", parts[1]);
Assert.AreEqual("BEGIN CATCH", parts[2]);
Assert.AreEqual("ROLLBACK TRANSACTION [ImportTransaction]", parts[3]);
Assert.AreEqual("DECLARE @ErrorMessage NVARCHAR(MAX)", parts[4]);
Assert.AreEqual("DECLARE @ErrorSeverity INT", parts[5]);
Assert.AreEqual("DECLARE @ErrorState INT", parts[6]);
Assert.AreEqual("SET @ErrorMessage = ERROR_MESSAGE()", parts[7]);
Assert.AreEqual("SET @ErrorSeverity = ERROR_SEVERITY()", parts[8]);
Assert.AreEqual("SET @ErrorState = ERROR_STATE()", parts[9]);
Assert.AreEqual("RAISERROR (@ErrorMessage, @ErrorSeverity, @ErrorState)", parts[10]);
Assert.AreEqual("END CATCH", parts[11]);
config.ErrorHandling.ImportAsTransaction = false;
Assert.AreEqual("", part.GetTransactionEndPart());
}
public void StatementCreatorOrderTest()
{
DBTable table = new DBTable("dbo", "TestTable");
DBColumn col1 = new DBColumn(table, "TestCol1", true, DBDatatype.integer);
DBColumn col2 = new DBColumn(table, "TestCol2", false, DBDatatype.nvarchar);
DBColumn col3 = new DBColumn(table, "TestCol3", false, DBDatatype.integer);
table.Columns = new List<DBColumn>() { col1, col2, col3 };
Database db = new Database("TestDB", new List<DBTable>() { table });
TableMapping sourceTablemapping = new TableMapping(new DBTable("dbo", "TestTable2"), TableMappingImportType.Insert, null);
ColumnMapping colMap1 = new TableColumnMapping(sourceTablemapping, col1, col1, ColumnUse.Where);
ColumnMapping colMap2 = new LiteralColumnMapping("2", LiteralType.String, col2, ColumnUse.Where);
ColumnMapping colMap3 = new LiteralColumnMapping("3", LiteralType.String, col2, ColumnUse.Set);
TableMapping tableMapping = new TableMapping(table, TableMappingImportType.Update, new ColumnMapping[] { colMap1, colMap2, colMap3 });
ErrorHandling errorHandling = new ErrorHandling();
ImportConfiguration config = new ImportConfiguration(new TableMapping[] { tableMapping }, null, "TestDB", errorHandling);
SourceDataEntry[] entries = new SourceDataEntry[] {SourceDataEntry.CreateDataEntry("", DataType.String, "") };
SourceDataRow[] rows = new SourceDataRow[] { new SourceDataRow(entries, "0") };
SourceDataTable dt = new SourceDataTable(rows, new string[] { "" });
SQLServerStatementCreator statementCreator = new SQLServerStatementCreator(config, dt);
ImportStatement statement = statementCreator.CreateStatement(0);
ImportStatement[] statements = statementCreator.CreateStatements();
Assert.AreEqual(1, statements.Length);
Assert.AreEqual(statement.RowReference, statements[0].RowReference);
Assert.AreEqual(statement.SqlStatement, statements[0].SqlStatement);
string[] lines = statement.SqlStatement
.Split(new string[] { "\n" }, StringSplitOptions.None).Where(s => s.Length > 0).ToArray();
StatementSetupPart setupPart = new StatementSetupPart(config);
Assert.AreEqual(setupPart.GetDatabasePart(), lines[0]);
Assert.AreEqual(setupPart.GetWarningsPart(), lines[1]);
StatementTransactionPart transPart = new StatementTransactionPart(config);
string[] transStartPart = transPart.GetTransactionStartPart()
.Split(new string[] { "\n" }, StringSplitOptions.None).Where(s => s.Length > 0).ToArray(); ;
Assert.AreEqual(2, transStartPart.Length);
Assert.AreEqual(transStartPart[0], lines[2]);
Assert.AreEqual(transStartPart[1], lines[3]);
StatementTableMappingPart tmParts = new StatementTableMappingPart(tableMapping, dt.GetDataRow(0));
string variablePart = tmParts.GetTableVariablePart().Replace("\n", "");
Assert.AreEqual(variablePart, lines[4]);
string[] bodyParts = tmParts.GetStatementBodyPart()
.Split(new string[] { "\n" }, StringSplitOptions.None).Where(s => s.Length > 0).ToArray();
Assert.AreEqual(4, bodyParts.Length);
Assert.AreEqual(bodyParts[0], lines[5]);
Assert.AreEqual(bodyParts[1], lines[6]);
Assert.AreEqual(bodyParts[2], lines[7]);
Assert.AreEqual(bodyParts[3], lines[8]);
string[] transEndPart = transPart.GetTransactionEndPart()
.Split(new string[] { "\n" }, StringSplitOptions.None).Where(s => s.Length > 0).ToArray();
Assert.AreEqual(12, transEndPart.Length);
Assert.AreEqual(transEndPart[0], lines[9]);
Assert.AreEqual(transEndPart[1], lines[10]);
Assert.AreEqual(transEndPart[2], lines[11]);
Assert.AreEqual(transEndPart[3], lines[12]);
Assert.AreEqual(transEndPart[4], lines[13]);
Assert.AreEqual(transEndPart[5], lines[14]);
Assert.AreEqual(transEndPart[6], lines[15]);
Assert.AreEqual(transEndPart[7], lines[16]);
Assert.AreEqual(transEndPart[8], lines[17]);
Assert.AreEqual(transEndPart[9], lines[18]);
Assert.AreEqual(transEndPart[10], lines[19]);
Assert.AreEqual(transEndPart[11], lines[20]);
}