# HG changeset patch
# User IBBoard <dev@ibboard.co.uk>
# Date 1299959647 0
# Node ID 5addcb8f7766b0b309729c0ba81f39b06a5a00fc
# Parent  dd384f35a96f024a523bc4e65f002450980ba699
Re #324: Add saving of Race and System data to files
 * Add unit tests for multiple files in one zip

diff -r dd384f35a96f -r 5addcb8f7766 API/Savers/IWarFoundryFileSaverTests.cs
--- a/API/Savers/IWarFoundryFileSaverTests.cs	Wed Mar 09 20:55:35 2011 +0000
+++ b/API/Savers/IWarFoundryFileSaverTests.cs	Sat Mar 12 19:54:07 2011 +0000
@@ -143,6 +143,54 @@
 				}
 			}
 		}
+
+		[Test()]
+		public void TestSaverCreatesTwoEntriesForOneSystemOneArmy()
+		{
+			string tempFile = Path.GetTempFileName();
+			try
+			{
+				GetSaver().Save(tempFile, new MockGameSystem(), MockArmy.GetMockArmy());
+				ZipFile file = new ZipFile(tempFile);
+				Assert.That(file.Count, Is.EqualTo(2));
+				file.Close();
+			}
+			finally
+			{
+				if (File.Exists(tempFile))
+				{
+					File.Delete(tempFile);
+				}
+			}
+		}
+
+		[Test()]
+		public void TestSaverCreatesTwoEntriesWithCorrectContentForOneSystemOneArmy()
+		{
+			//This doesn't make much sense (system and army in one file) but at the time of writing then Race isn't implemented
+			string tempFile = Path.GetTempFileName();
+			try
+			{
+				MockGameSystem system = new MockGameSystem();
+				MockArmy army = MockArmy.GetMockArmy();
+				GetSaver().Save(tempFile, system, army);
+				ZipFile file = new ZipFile(tempFile);
+				ZipEntry zipEntry = file.GetEntry(GetEntryName(system));
+				Stream stream = file.GetInputStream(zipEntry);
+				Assert.That(StreamUtil.ToBytes(stream), Is.EqualTo(GetGameSystemContentBytes()));
+				zipEntry = file.GetEntry(GetEntryName(army));
+				stream = file.GetInputStream(zipEntry);
+				Assert.That(StreamUtil.ToBytes(stream), Is.EqualTo(GetArmyContentBytes()));
+				file.Close();
+			}
+			finally
+			{
+				if (File.Exists(tempFile))
+				{
+					File.Delete(tempFile);
+				}
+			}
+		}
 		
 		[Test()]
 		public void TestFileExtensionsAreCorrectForType()