From 699de5a7e64c0714826d33b37d3400f446f3af0f Mon Sep 17 00:00:00 2001
From: DanBribi <80496947+Dbribs@users.noreply.github.com>
Date: Wed, 15 Dec 2021 16:44:21 -0800
Subject: [PATCH] Fixed test cases (#18)

---
 .../ArchivingTests.cs                                     | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/Source Code/TeamHobby.HobbyProjectGenerator.ArchiveTests/ArchivingTests.cs b/Source Code/TeamHobby.HobbyProjectGenerator.ArchiveTests/ArchivingTests.cs
index 9775996..3d80763 100644
--- a/Source Code/TeamHobby.HobbyProjectGenerator.ArchiveTests/ArchivingTests.cs	
+++ b/Source Code/TeamHobby.HobbyProjectGenerator.ArchiveTests/ArchivingTests.cs	
@@ -222,19 +222,21 @@ namespace TeamHobby.HobbyProjectGenerator.ArchiveTests
                 "('2021-09-03 23:00:00', 'Info', 'Business', 'log in', 'log in successfully');");
 
                 OdbcDataReader odbcObj = (OdbcDataReader)sqlDS.ReadData("SELECT * FROM log WHERE DATEDIFF(CURRENT_TIMESTAMP, log.LtimeStamp) > 30;");
-
+                
                 int count = 0;
                 while (odbcObj.Read()) { ++count; }
                 Console.WriteLine("Number of entries > 30 days old: " + count);
+                sqlDS.GetConnection().Close();
 
                 Console.WriteLine("\nRemoving entries from archive...");
                 sqlDS.RemoveEntries();
 
-                odbcObj = (OdbcDataReader)sqlDS.ReadData("SELECT * FROM log WHERE DATEDIFF(CURRENT_TIMESTAMP, log.LtimeStamp) > 30;");
-
+                odbcObj = (OdbcDataReader)sqlDS.ReadData("SELECT * FROM log WHERE DATEDIFF(CURRENT_TIMESTAMP, log.LtimeStamp) > 30;"); 
+                                
                 count = 0;
                 while (odbcObj.Read()) { ++count; }
                 Console.WriteLine("Number of entries > 30 days old: " + count);
+                sqlDS.GetConnection().Close();
 
                 // Assert
                 bool expectedVal = true;