|
1 | 1 | using Microsoft.VisualStudio.TestTools.UnitTesting; |
2 | 2 | using QueryDB.Exceptions; |
3 | 3 | using System; |
| 4 | +using System.Collections.Generic; |
4 | 5 | using System.Linq; |
5 | 6 |
|
6 | 7 | namespace QueryDB.Core.Tests |
@@ -452,6 +453,139 @@ public void Test_MySQL_ExecuteCommand_DCL_Queries() |
452 | 453 |
|
453 | 454 | #endregion |
454 | 455 |
|
| 456 | + #region Execute Transaction Tests - << bool ExecuteTransaction(List<string> sqlStatements) >> |
| 457 | + |
| 458 | + [TestMethod] |
| 459 | + [TestCategory(DB_TESTS), TestCategory(MYSQL_TESTS)] |
| 460 | + public void Test_MySQL_ExecuteTransaction_DDL_Multiple_Queries() |
| 461 | + { |
| 462 | + var createTableSql = Queries.MySQLQueries.TestDB.DDL.Create_Table; |
| 463 | + var alterTableSql = Queries.MySQLQueries.TestDB.DDL.Alter_Table; |
| 464 | + var truncateTableSql = Queries.MySQLQueries.TestDB.DDL.Truncate_Table; |
| 465 | + var renameTableSql = Queries.MySQLQueries.TestDB.DDL.Rename_Table; |
| 466 | + var dropTableSql = Queries.MySQLQueries.TestDB.DDL.Drop_Table; |
| 467 | + var dDLExecutionCheckSql = Queries.MySQLQueries.TestDB.DDL.DDL_Execute_check; |
| 468 | + |
| 469 | + // Create, Alter & Truncate |
| 470 | + var statements = new List<string> |
| 471 | + { |
| 472 | + createTableSql, |
| 473 | + alterTableSql, |
| 474 | + truncateTableSql |
| 475 | + }; |
| 476 | + var dbContext = new DBContext(DB.MySQL, MySQLConnectionString); |
| 477 | + var result = dbContext.ExecuteTransaction(statements); |
| 478 | + Assert.AreEqual(true, result); |
| 479 | + |
| 480 | + var tableCount = dbContext |
| 481 | + .FetchData(string.Format(dDLExecutionCheckSql, "mysql", "Employee")); |
| 482 | + Assert.AreEqual("1", tableCount[0].ReferenceData["Table_Count"]); |
| 483 | + |
| 484 | + // Rename & Drop |
| 485 | + statements = new List<string> |
| 486 | + { |
| 487 | + renameTableSql, |
| 488 | + dropTableSql |
| 489 | + }; |
| 490 | + result = dbContext.ExecuteTransaction(statements); |
| 491 | + Assert.AreEqual(true, result); |
| 492 | + |
| 493 | + tableCount = dbContext |
| 494 | + .FetchData(string.Format(dDLExecutionCheckSql, "mysql", "Employees")); |
| 495 | + Assert.AreEqual("0", tableCount[0].ReferenceData["Table_Count"]); |
| 496 | + } |
| 497 | + |
| 498 | + [TestMethod] |
| 499 | + [TestCategory(DB_TESTS), TestCategory(MYSQL_TESTS)] |
| 500 | + public void Test_MySQL_ExecuteTransaction_DML_Multiple_Queries() |
| 501 | + { |
| 502 | + var insertSql = Queries.MySQLQueries.TestDB.DML.InsertSql; |
| 503 | + var updateSql = Queries.MySQLQueries.TestDB.DML.UpdateSql; |
| 504 | + var deleteSql = Queries.MySQLQueries.TestDB.DML.DeleteSql; |
| 505 | + var verifyDMLExecution = Queries.MySQLQueries.TestDB.DML.VerifyDMLExecution; |
| 506 | + |
| 507 | + var statements = new List<string> |
| 508 | + { |
| 509 | + insertSql, |
| 510 | + updateSql |
| 511 | + }; |
| 512 | + var dbContext = new DBContext(DB.MySQL, MySQLConnectionString); |
| 513 | + |
| 514 | + // Insert & Update |
| 515 | + var result = dbContext.ExecuteTransaction(statements); |
| 516 | + Assert.AreEqual(true, result); |
| 517 | + var data = dbContext.FetchData(verifyDMLExecution); |
| 518 | + Assert.IsTrue(data.Count == 1); |
| 519 | + var agent = data.FirstOrDefault(); |
| 520 | + Assert.AreEqual("A020", agent.ReferenceData["Agent_Code"]); |
| 521 | + Assert.AreEqual("John", agent.ReferenceData["Agent_Name"]); |
| 522 | + Assert.AreEqual("Wick", agent.ReferenceData["Working_Area"]); |
| 523 | + Assert.AreEqual("0.15", agent.ReferenceData["Commission"]); |
| 524 | + Assert.AreEqual("010-44536178", agent.ReferenceData["Phone_No"]); |
| 525 | + Assert.AreEqual("", agent.ReferenceData["Country"]); |
| 526 | + |
| 527 | + // Delete |
| 528 | + statements = new List<string> |
| 529 | + { |
| 530 | + deleteSql |
| 531 | + }; |
| 532 | + result = dbContext.ExecuteTransaction(statements); |
| 533 | + Assert.AreEqual(true, result); |
| 534 | + data = dbContext.FetchData(verifyDMLExecution); |
| 535 | + Assert.IsTrue(data.Count == 0); |
| 536 | + } |
| 537 | + |
| 538 | + [TestMethod] |
| 539 | + [TestCategory(DB_TESTS), TestCategory(MYSQL_TESTS)] |
| 540 | + public void Test_MySQL_ExecuteTransaction_Incomplete_Transaction_Rollback_On_Error() |
| 541 | + { |
| 542 | + var insertSql = Queries.MySQLQueries.TestDB.DML.InsertSql; |
| 543 | + var updateSql = Queries.MySQLQueries.TestDB.DML.UpdateSql; |
| 544 | + var updateErrorSql = "UPDATE"; |
| 545 | + var verifyDMLExecution = Queries.MySQLQueries.TestDB.DML.VerifyDMLExecution; |
| 546 | + |
| 547 | + var statements = new List<string> |
| 548 | + { |
| 549 | + insertSql, |
| 550 | + updateSql, |
| 551 | + updateErrorSql |
| 552 | + }; |
| 553 | + var dbContext = new DBContext(DB.MySQL, MySQLConnectionString); |
| 554 | + |
| 555 | + // Insert & Update |
| 556 | + var result = dbContext.ExecuteTransaction(statements); |
| 557 | + Assert.AreEqual(false, result); |
| 558 | + var data = dbContext.FetchData(verifyDMLExecution); |
| 559 | + Assert.IsTrue(data.Count == 0); |
| 560 | + } |
| 561 | + |
| 562 | + [TestMethod] |
| 563 | + [TestCategory(DB_TESTS), TestCategory(MYSQL_TESTS)] |
| 564 | + public void Test_MySQL_ExecuteTransaction_DML_Unsupported_SELECT_Queries() |
| 565 | + { |
| 566 | + var selectSql = Queries.MySQLQueries.TestDB.DML.SelectSql; |
| 567 | + |
| 568 | + // Select |
| 569 | + try |
| 570 | + { |
| 571 | + var statements = new List<string> |
| 572 | + { |
| 573 | + selectSql |
| 574 | + }; |
| 575 | + var dbContext = new DBContext(DB.MySQL, MySQLConnectionString); |
| 576 | + var result = dbContext.ExecuteTransaction(statements); |
| 577 | + Assert.Fail("No Exception"); |
| 578 | + } |
| 579 | + catch (QueryDBException ex) |
| 580 | + { |
| 581 | + Assert.AreEqual("SELECT queries are not supported here.", ex.Message); |
| 582 | + Assert.AreEqual("UnsupportedCommand", ex.ErrorType); |
| 583 | + Assert.AreEqual("'ExecuteTransaction' doesn't support SELECT queries.", ex.AdditionalInfo); |
| 584 | + } |
| 585 | + } |
| 586 | + |
| 587 | + #endregion |
| 588 | + |
455 | 589 | #endregion |
456 | 590 |
|
457 | 591 | } |
|
0 commit comments