cuetools.net/CUETools/CUETools.TestParity/CDRepairTest.cs

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);
		}
	}
}