diff API/Savers/IWarFoundryFileSaverTests.cs @ 99:dd384f35a96f

Re #324: Add saving of Race and System data to files * Start to migrate Army saving to same structure * Add first test * Give MockArmy a specific ID
author IBBoard <dev@ibboard.co.uk>
date Wed, 09 Mar 2011 20:55:35 +0000
parents 49c5e24decb9
children 5addcb8f7766
line wrap: on
line diff
--- a/API/Savers/IWarFoundryFileSaverTests.cs	Wed Mar 09 20:45:13 2011 +0000
+++ b/API/Savers/IWarFoundryFileSaverTests.cs	Wed Mar 09 20:55:35 2011 +0000
@@ -120,6 +120,29 @@
 				}
 			}
 		}
+
+		[Test()]
+		public void TestSaverCreatesSingleEntryWithCorrectContentForOneArmy()
+		{
+			string tempFile = Path.GetTempFileName();
+			try
+			{
+				MockArmy obj = MockArmy.GetMockArmy();
+				GetSaver().Save(tempFile, obj);
+				ZipFile file = new ZipFile(tempFile);
+				ZipEntry zipEntry = file.GetEntry(GetEntryName(obj));
+				Stream 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()
@@ -141,6 +164,8 @@
 		
 		protected abstract string GetGameSystemExtension();
 
-		public abstract byte[] GetGameSystemContentBytes();
+		protected abstract byte[] GetGameSystemContentBytes();
+		
+		protected abstract byte[] GetArmyContentBytes();
 	}
 }