private void RunTest(string sourceFile, string testFile)
{
using (FileUnitTester fut = new FileUnitTester(testFile))
{
this.RewriteProgramMsdos(sourceFile, Address.SegPtr(0xC00, 0));
program.Procedures.Values[0].Write(false, fut.TextWriter);
fut.TextWriter.WriteLine();
var cfgc = new ControlFlowGraphCleaner(program.Procedures.Values[0]);
cfgc.Transform();
program.Procedures.Values[0].Write(false, fut.TextWriter);
fut.AssertFilesEqual();
}
}
private void RunTest(ProcedureBuilder mock, string outFilename)
{
Procedure proc = mock.Procedure;
using (FileUnitTester fut = new FileUnitTester(outFilename))
{
ControlFlowGraphCleaner cfgc = new ControlFlowGraphCleaner(proc);
cfgc.Transform();
proc.Write(false, fut.TextWriter);
fut.TextWriter.WriteLine();
StructureAnalysis sa = new StructureAnalysis(proc);
sa.Structure();
CodeFormatter fmt = new CodeFormatter(new TextFormatter(fut.TextWriter));
fmt.Write(proc);
fut.TextWriter.WriteLine("===========================");
fut.AssertFilesEqual();
}
}
private void RunTest16(string sourceFilename, string outFilename, Address addrBase)
{
using (FileUnitTester fut = new FileUnitTester(outFilename))
{
RewriteProgramMsdos(sourceFilename, addrBase);
foreach (Procedure proc in program.Procedures.Values)
{
var cfgc = new ControlFlowGraphCleaner(program.Procedures.Values[0]);
cfgc.Transform();
proc.Write(false, fut.TextWriter);
fut.TextWriter.WriteLine();
var sa = new StructureAnalysis(proc);
sa.Structure();
var fmt = new CodeFormatter(new TextFormatter(fut.TextWriter));
fmt.Write(proc);
fut.TextWriter.WriteLine("===========================");
}
fut.AssertFilesEqual();
}
}
private void RunTest32(string expected, Program program)
{
var sw = new StringWriter();
foreach (var proc in program.Procedures.Values)
{
var cfgc = new ControlFlowGraphCleaner(proc);
cfgc.Transform();
var sa = new StructureAnalysis(proc);
sa.Structure();
var fmt = new CodeFormatter(new TextFormatter(sw));
fmt.Write(proc);
sw.WriteLine("===");
}
Console.WriteLine(sw);
Assert.AreEqual(expected, sw.ToString());
}