public void TestTableInfo(bool usenamer)
{
string filename = Path.Combine(LoadDirectory.ForLoading.FullName, "bob.csv");
var sw = new StreamWriter(filename);
sw.WriteLine("name,name2");
sw.WriteLine("Bob,Munchousain");
sw.WriteLine("Franky,Hollyw9ood");
sw.Flush();
sw.Close();
sw.Dispose();
TableInfo ti;
ColumnInfo[] cols;
Import(_table, out ti, out cols);
var attacher = new AnySeparatorFileAttacher();
attacher.Initialize(LoadDirectory, _database);
attacher.Separator = ",";
attacher.FilePattern = "bob*";
attacher.TableToLoad = ti;
INameDatabasesAndTablesDuringLoads namer = null;
if (usenamer)
{
_table.Rename("AAA");
namer = RdmpMockFactory.Mock_INameDatabasesAndTablesDuringLoads(_database, "AAA");
}
var job = new ThrowImmediatelyDataLoadJob(new HICDatabaseConfiguration(_database.Server, namer), ti);
var exitCode = attacher.Attach(job, new GracefulCancellationToken());
Assert.AreEqual(ExitCodeType.Success, exitCode);
using (var con = _database.Server.GetConnection())
{
con.Open();
var r = _database.Server.GetCommand("Select name,name2 from " + _table.GetRuntimeName(), con).ExecuteReader();
Assert.IsTrue(r.Read());
Assert.AreEqual("Bob", r["name"]);
Assert.AreEqual("Munchousain", r["name2"]);
Assert.IsTrue(r.Read());
Assert.AreEqual("Franky", r["name"]);
Assert.AreEqual("Hollyw9ood", r["name2"]);
}
attacher.LoadCompletedSoDispose(ExitCodeType.Success, new ThrowImmediatelyDataLoadEventListener());
File.Delete(filename);
}
public void TabTestWithOverrideHeaders_IncludePath(bool columnExistsInRaw)
{
string filename = Path.Combine(LoadDirectory.ForLoading.FullName, "bob.csv");
var sw = new StreamWriter(filename);
sw.WriteLine("Face\tBasher");
sw.WriteLine("Candy\tCrusher");
sw.Flush();
sw.Close();
sw.Dispose();
if (columnExistsInRaw)
{
_table.AddColumn("FilePath", new DatabaseTypeRequest(typeof(string), 500), true, 30);
}
var attacher = new AnySeparatorFileAttacher();
attacher.Initialize(LoadDirectory, _database);
attacher.Separator = "\\t";
attacher.FilePattern = "bob*";
attacher.TableName = "Bob";
attacher.ForceHeaders = "name\tname2";
attacher.AddFilenameColumnNamed = "FilePath";
if (!columnExistsInRaw)
{
var ex = Assert.Throws <FlatFileLoadException>(() => attacher.Attach(new ThrowImmediatelyDataLoadJob(), new GracefulCancellationToken()));
Assert.AreEqual("AddFilenameColumnNamed is set to 'FilePath' but the column did not exist in RAW", ex.InnerException.Message);
return;
}
var exitCode = attacher.Attach(new ThrowImmediatelyDataLoadJob(), new GracefulCancellationToken());
Assert.AreEqual(ExitCodeType.Success, exitCode);
using (var con = _database.Server.GetConnection())
{
con.Open();
var r = _database.Server.GetCommand("Select name,name2,FilePath from Bob", con).ExecuteReader();
Assert.IsTrue(r.Read());
Assert.AreEqual("Face", r["name"]);
Assert.AreEqual("Basher", r["name2"]);
Assert.AreEqual(filename, r["FilePath"]);
Assert.IsTrue(r.Read());
Assert.AreEqual("Candy", r["name"]);
Assert.AreEqual("Crusher", r["name2"]);
}
attacher.LoadCompletedSoDispose(ExitCodeType.Success, new ThrowImmediatelyDataLoadEventListener());
File.Delete(filename);
}
public void Test_ExplicitDateTimeFormat_Attachment()
{
string filename = Path.Combine(LoadDirectory.ForLoading.FullName, "bob.csv");
var sw = new StreamWriter(filename);
sw.WriteLine("name,name2");
sw.WriteLine("Bob,20011301");
sw.WriteLine("Franky,20021301");
sw.Flush();
sw.Close();
sw.Dispose();
var attacher = new AnySeparatorFileAttacher();
attacher.Initialize(LoadDirectory, _database);
attacher.Separator = ",";
attacher.FilePattern = "bob*";
attacher.TableName = "Bob";
attacher.ExplicitDateTimeFormat = "yyyyddMM";
var table = _database.ExpectTable("Bob");
table.Truncate();
Assert.IsTrue(table.Exists());
table.DiscoverColumn("name");
var name2 = table.DiscoverColumn("name2");
name2.DataType.AlterTypeTo("datetime2");
//other cases (i.e. correct separator)
attacher.Attach(new ThrowImmediatelyDataLoadJob(), new GracefulCancellationToken());
using (var con = _database.Server.GetConnection())
{
con.Open();
var r = _database.Server.GetCommand("Select * from Bob", con).ExecuteReader();
Assert.IsTrue(r.Read());
Assert.AreEqual("Bob", r["name"]);
Assert.AreEqual(new DateTime(2001, 01, 13), r["name2"]);
Assert.IsTrue(r.Read());
Assert.AreEqual("Franky", r["name"]);
Assert.AreEqual(new DateTime(2002, 01, 13), r["name2"]);
}
attacher.LoadCompletedSoDispose(ExitCodeType.Success, new ThrowImmediatelyDataLoadEventListener());
File.Delete(filename);
}
public void Test_FlatFileAttcher_IgnoreColumns()
{
string filename = Path.Combine(LoadDirectory.ForLoading.FullName, "bob.csv");
var sw = new StreamWriter(filename);
sw.WriteLine("name,name2,address");
sw.WriteLine("Bob,Munchousain,\"67, franklin\"");
sw.WriteLine("Franky,Hollyw9ood,32 dodgery");
sw.Flush();
sw.Close();
sw.Dispose();
Import(_table, out var ti, out var cols);
var attacher = new AnySeparatorFileAttacher();
attacher.Initialize(LoadDirectory, _database);
attacher.Separator = ",";
attacher.FilePattern = "bob*";
attacher.TableToLoad = ti;
attacher.IgnoreColumns = "address";
var job = new ThrowImmediatelyDataLoadJob(new HICDatabaseConfiguration(_database.Server, null), ti);
var exitCode = attacher.Attach(job, new GracefulCancellationToken());
Assert.AreEqual(ExitCodeType.Success, exitCode);
using (var con = _database.Server.GetConnection())
{
con.Open();
var r = _database.Server.GetCommand("Select name,name2 from " + _table.GetRuntimeName(), con).ExecuteReader();
Assert.IsTrue(r.Read());
Assert.AreEqual("Bob", r["name"]);
Assert.AreEqual("Munchousain", r["name2"]);
Assert.IsTrue(r.Read());
Assert.AreEqual("Franky", r["name"]);
Assert.AreEqual("Hollyw9ood", r["name2"]);
}
attacher.LoadCompletedSoDispose(ExitCodeType.Success, new ThrowImmediatelyDataLoadEventListener());
File.Delete(filename);
}
public void Test_FlatFileAttcher_AmbiguousDates(DatabaseType type, string val, string attacherCulture, string threadCulture)
{
Thread.CurrentThread.CurrentCulture = new CultureInfo(threadCulture);
string filename = Path.Combine(LoadDirectory.ForLoading.FullName, "bob.csv");
var sw = new StreamWriter(filename);
sw.WriteLine("dob");
sw.WriteLine(val);
sw.Flush();
sw.Close();
sw.Dispose();
var db = GetCleanedServer(type);
var tbl =
db.CreateTable("AmbiguousDatesTestTable",
new [] { new DatabaseColumnRequest("dob", new DatabaseTypeRequest(typeof(DateTime))) }
);
Import(tbl, out TableInfo ti, out _);
var attacher = new AnySeparatorFileAttacher();
attacher.Separator = ",";
attacher.FilePattern = "bob*";
attacher.TableName = tbl.GetRuntimeName();
attacher.Culture = new CultureInfo(attacherCulture);
attacher.Initialize(LoadDirectory, db);
var job = new ThrowImmediatelyDataLoadJob(new HICDatabaseConfiguration(_database.Server, null), ti);
var exitCode = attacher.Attach(job, new GracefulCancellationToken());
Assert.AreEqual(ExitCodeType.Success, exitCode);
attacher.LoadCompletedSoDispose(ExitCodeType.Success, new ThrowImmediatelyDataLoadEventListener());
Assert.AreEqual(new DateTime(2001, 1, 27), tbl.GetDataTable().Rows[0][0]);
File.Delete(filename);
tbl.Drop();
}
public void TabTestWithOverrideHeaders()
{
string filename = Path.Combine(LoadDirectory.ForLoading.FullName, "bob.csv");
var sw = new StreamWriter(filename);
sw.WriteLine("Face\tBasher");
sw.WriteLine("Candy\tCrusher");
sw.Flush();
sw.Close();
sw.Dispose();
var attacher = new AnySeparatorFileAttacher();
attacher.Initialize(LoadDirectory, _database);
attacher.Separator = "\\t";
attacher.FilePattern = "bob*";
attacher.TableName = "Bob";
attacher.ForceHeaders = "name\tname2";
var exitCode = attacher.Attach(new ThrowImmediatelyDataLoadJob(), new GracefulCancellationToken());
Assert.AreEqual(ExitCodeType.Success, exitCode);
using (var con = _database.Server.GetConnection())
{
con.Open();
var r = _database.Server.GetCommand("Select name,name2 from Bob", con).ExecuteReader();
Assert.IsTrue(r.Read());
Assert.AreEqual("Face", r["name"]);
Assert.AreEqual("Basher", r["name2"]);
Assert.IsTrue(r.Read());
Assert.AreEqual("Candy", r["name"]);
Assert.AreEqual("Crusher", r["name2"]);
}
attacher.LoadCompletedSoDispose(ExitCodeType.Success, new ThrowImmediatelyDataLoadEventListener());
File.Delete(filename);
}
public void Test_CSV_Attachment(string separator, bool overrideHeaders)
{
string filename = Path.Combine(LoadDirectory.ForLoading.FullName, "bob.csv");
var sw = new StreamWriter(filename);
sw.WriteLine("name,name2");
sw.WriteLine("Bob,Munchousain");
sw.WriteLine("Franky,Hollyw9ood");
sw.Flush();
sw.Close();
sw.Dispose();
string filename2 = Path.Combine(LoadDirectory.ForLoading.FullName, "bob2.csv");
var sw2 = new StreamWriter(filename2);
sw2.WriteLine("name,name2");
sw2.WriteLine("Manny2,Ok");
sw2.Flush();
sw2.Close();
sw2.Dispose();
var attacher = new AnySeparatorFileAttacher();
attacher.Initialize(LoadDirectory, _database);
attacher.Separator = separator;
attacher.FilePattern = "bob*";
attacher.TableName = "Bob";
if (overrideHeaders)
{
attacher.ForceHeaders = "name,name2";
attacher.ForceHeadersReplacesFirstLineInFile = true;
}
//Case when you are using the wrong separator
if (separator == "|")
{
var ex = Assert.Throws <FlatFileLoadException>(() => attacher.Attach(new ThrowImmediatelyDataLoadJob(), new GracefulCancellationToken()));
Assert.IsNotNull(ex.InnerException);
StringAssert.StartsWith("Your separator does not appear in the headers line of your file (bob.csv) but the separator ',' does", ex.InnerException.Message);
return;
}
//other cases (i.e. correct separator)
attacher.Attach(new ThrowImmediatelyDataLoadJob(), new GracefulCancellationToken());
var table = _database.ExpectTable("Bob");
Assert.IsTrue(table.Exists());
table.DiscoverColumn("name");
table.DiscoverColumn("name2");
using (var con = _database.Server.GetConnection())
{
con.Open();
var r = _database.Server.GetCommand("Select * from Bob", con).ExecuteReader();
Assert.IsTrue(r.Read());
Assert.AreEqual("Bob", r["name"]);
Assert.AreEqual("Munchousain", r["name2"]);
Assert.IsTrue(r.Read());
Assert.AreEqual("Franky", r["name"]);
Assert.AreEqual("Hollyw9ood", r["name2"]);
Assert.IsTrue(r.Read());
Assert.AreEqual("Manny2", r["name"]);
Assert.AreEqual("Ok", r["name2"]);
}
attacher.LoadCompletedSoDispose(ExitCodeType.Success, new ThrowImmediatelyDataLoadEventListener());
File.Delete(filename);
}