TestAssemblerConditionals.java

/*
  TestAssemblerConditionals.java

  (c) 2009-2013 Edward Swartz

  All rights reserved. This program and the accompanying materials
  are made available under the terms of the Eclipse Public License v1.0
  which accompanies this distribution, and is available at
  http://www.eclipse.org/legal/epl-v10.html
 */
package v9t9.tests.asm9900;

import java.util.List;

import v9t9.common.asm.IInstruction;
import v9t9.tests.inst9900.BaseTest9900;
import v9t9.tools.asm.Assembler;
import v9t9.tools.asm.ContentEntry;
import v9t9.tools.asm.Equate;
import v9t9.tools.asm.Symbol;
import v9t9.tools.asm.inst9900.Assembler9900;

public class TestAssemblerConditionals extends BaseTest9900 {

	Assembler assembler = new Assembler9900();
	

	public void testConditionalIf() throws Exception {
		String text =
			"SP equ 10\n"+
			" aorg >100\n"+
			" #if FOO\n" +
			" ai SP, 6\n"+
			" #fi\n"+
			" rt\n"+
			"";
		
		testFileContent(
				new String[] { "FOO" },
				text,
				0x100,
				new String[] { "ai r10, 6", "b *r11" },
				new Symbol[] { new Equate(assembler.getSymbolTable(), "SP", 10) });
		
	}
	public void testConditionalIfElse() throws Exception {
		String text =
			"SP equ 10\n"+
			" aorg >100\n"+
			" #if FOO\n" +
			" ai SP, 6\n"+
			" #else\n"+
			" ai SP, 4\n"+
			" #fi\n"+
			" rt\n"+
			"";
		
		testFileContent(
				new String[] { "FOO=1" },
				text,
				0x100,
				new String[] { "ai r10, 6", "b *r11"},
				new Symbol[] { new Equate(assembler.getSymbolTable(), "SP", 10) });
		
		testFileContent(
				new String[] { "FOO=0" },
				text,
				0x100,
				new String[] { "ai r10, 4", "b *r11"},
				new Symbol[] { new Equate(assembler.getSymbolTable(), "SP", 10) });
		
		
	}
	
	public void testConditionalIfUndefined() throws Exception {
		String text =
			"SP equ 10\n"+
			" aorg >100\n"+
			" #if FOO\n" +
			" ai SP, 6\n"+
			" #else\n"+
			" ai SP, 4\n"+
			" #fi\n"+
			" rt\n"+
			"";
		
		// undefined symbols parse as 0
		testFileContent(
				new String[] { },
				text,
				0x100,
				new String[] { "ai r10, 4", "b *r11"},
				new Symbol[] { new Equate(assembler.getSymbolTable(), "SP", 10) });
	}

	public void testConditionalIfEquates() throws Exception {
		String text =
			" #if FOO\n" +
			"ADDR1 equ 1\n"+
			"ADDR2 equ 2\n"+
			" #else\n"+
			"ADDR1 equ 2\n"+
			"ADDR2 equ 1\n"+
			" #fi\n"+
			"";
		
		testFileContent(
				new String[] { "FOO=1" },
				text,
				0x100,
				new String[] { },
				new Symbol[] { new Equate(assembler.getSymbolTable(), "ADDR1", 1),
						new Equate(assembler.getSymbolTable(), "ADDR2", 2)});
	}
	private void testFileContent(
			String[] equates,
			String text, int pc, String[] stdInsts, Symbol[] symbols) throws Exception {
		String caller = new Exception().fillInStackTrace().getStackTrace()[1].getMethodName();
		assembler.getSymbolTable().clear();

		for (String equ : equates)
			assembler.defineEquate(equ);
		assembler.pushContentEntry(new ContentEntry(caller + ".asm", text));
		List<IInstruction> asminsts = assembler.parse();
		List<IInstruction> realinsts = assembler.resolve(asminsts);
		realinsts = assembler.optimize(realinsts);

		testGeneratedContent(assembler, pc, stdInsts, symbols, realinsts);
	}
}