CD parity information database test

CUETools/CUETools.TestParity/CDRepairTest.cs

@@ -0,0 +1,145 @@
+using System;
+using CUETools.Codecs;
+using CUETools.CDRepair;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+namespace CUETools.TestParity
+{
+    
+    
+    /// <summary>
+    ///This is a test class for CDRepairTest and is intended
+    ///to contain all CDRepairTest Unit Tests
+    ///</summary>
+	[TestClass()]
+	public class CDRepairTest
+	{
+
+
+		private TestContext testContextInstance;
+
+		const int finalSampleCount = 44100 * 60 * 10; // 20 minutes long
+		//const int stride = finalSampleCount * 2 / 32768;
+		const int stride = 10 * 588 * 2;
+		const int npar = 8;
+		static byte[] wav = new byte[finalSampleCount * 4];
+		static byte[] wav2 = new byte[finalSampleCount * 4];
+		static byte[] parity;
+		static uint crc;
+		static CDRepairEncode decode;
+		static CDRepairEncode decode2;
+		const int offset = 48;
+
+		/// <summary>
+		///Gets or sets the test context which provides
+		///information about and functionality for the current test run.
+		///</summary>
+		public TestContext TestContext
+		{
+			get
+			{
+				return testContextInstance;
+			}
+			set
+			{
+				testContextInstance = value;
+			}
+		}
+
+		#region Additional test attributes
+		// 
+		//You can use the following additional attributes as you write your tests:
+		//
+		//Use ClassInitialize to run code before running the first test in the class
+		[ClassInitialize()]
+		public static void MyClassInitialize(TestContext testContext)
+		{
+			new Random(2423).NextBytes(wav);
+			new Random(2423).NextBytes(wav2);
+			Random rnd = new Random(987);
+			for (int i = 0; i < stride / 4; i++ )
+				wav2[(int)(rnd.NextDouble() * (wav2.Length - 1))] = (byte)(rnd.NextDouble() * 255);
+
+			AudioBuffer buff = new AudioBuffer(AudioPCMConfig.RedBook, 0);
+			CDRepairEncode encode = new CDRepairEncode(finalSampleCount, stride, npar, false);
+			buff.Prepare(wav, finalSampleCount);
+			encode.Write(buff);
+			encode.Close();
+			parity = encode.Parity;
+			crc = encode.CRC;
+
+			decode = new CDRepairEncode(finalSampleCount, stride, npar, true);
+			buff.Prepare(wav2, finalSampleCount);
+			decode.Write(buff);
+			decode.Close();
+			decode.VerifyParity(parity);
+
+			decode2 = new CDRepairEncode(finalSampleCount, stride, npar, true);
+			buff.Prepare(new byte[offset * 4], offset);
+			decode2.Write(buff);
+			buff.Prepare(wav2, finalSampleCount - offset);
+			decode2.Write(buff);
+			decode2.Close();
+			decode2.VerifyParity(parity);
+		}
+
+		//Use ClassCleanup to run code after all tests in a class have run
+		//[ClassCleanup()]
+		//public static void MyClassCleanup()
+		//{
+		//}
+		//
+		//Use TestInitialize to run code before running each test
+		//[TestInitialize()]
+		//public void MyTestInitialize()
+		//{
+		//}		
+		//Use TestCleanup to run code after each test has run
+		//[TestCleanup()]
+		//public void MyTestCleanup()
+		//{
+		//}
+		//
+		#endregion
+
+
+		/// <summary>
+		///A test for CDRepair Constructor
+		///</summary>
+		[TestMethod()]
+		public void CDRepairFixTest()
+		{
+			Assert.IsTrue(decode.HasErrors);
+			Assert.IsTrue(decode.CanRecover);
+			Assert.AreEqual(0, decode.ActualOffset, "wrong offset");
+
+			AudioBuffer buff = new AudioBuffer(AudioPCMConfig.RedBook, 0);
+			CDRepairFix fix = new CDRepairFix(decode);
+			buff.Prepare(wav2, finalSampleCount);
+			fix.Write(buff);
+			fix.Close();
+
+			Assert.AreEqual<uint>(crc, fix.CRC);
+		}
+
+		/// <summary>
+		///Repair with offset
+		///</summary>
+		[TestMethod()]
+		public void CDRepairFixWithOffsetTest()
+		{
+			Assert.IsTrue(decode2.HasErrors);
+			Assert.IsTrue(decode2.CanRecover);
+			Assert.AreEqual(-offset, decode2.ActualOffset, "wrong offset");
+
+			AudioBuffer buff = new AudioBuffer(AudioPCMConfig.RedBook, 0);
+			CDRepairFix fix = new CDRepairFix(decode2);
+			buff.Prepare(new byte[offset * 4], offset);
+			fix.Write(buff);
+			buff.Prepare(wav2, finalSampleCount - offset);
+			fix.Write(buff);
+			fix.Close();
+
+			Assert.AreEqual<uint>(crc, fix.CRC);
+		}
+	}
+}