initial mapper test

Author: zohassadar

tests/src/main.rs

@@ -9,6 +9,7 @@ mod video;
 mod drought;
 mod floor;
 mod garbage;
+mod mapper;
 mod pushdown;
 mod rng;
 mod score;
@@ -47,6 +48,8 @@ fn main() {
     if options.test {
         // garbage::test_garbage4_crash();
         // println!(">> garbage4 ✅");
+        mapper::test();
+        println!(">> mappers ✅");
         floor::test();
         println!(">> floor ✅");
         tspins::test();

tests/src/mapper.rs

@@ -0,0 +1,57 @@
+use crate::util;
+use rusticnes_core::nes::NesState;
+
+pub fn get_expected_tilesets() -> (Vec<u8>, Vec<u8>) {
+
+    // tilesets as ordered in the game
+    let title_menu_tileset = include_bytes!("../../src/chr/title_menu_tileset.chr");
+    let game_tileset = include_bytes!("../../src/chr/game_tileset.chr");
+    let rocket_tileset = include_bytes!("../../src/chr/rocket_tileset.chr");
+    let empty_tileset: Vec<u8> = vec![0; 0x1000];
+    
+    // cnrom limited to reading 8k banks, so pair tilesets accordingly
+    let mut tileset1: Vec<u8> = vec![0; 0x2000];
+    tileset1[..0x1000].copy_from_slice(title_menu_tileset);
+    tileset1[0x1000..].copy_from_slice(game_tileset);
+    
+    let mut tileset2: Vec<u8> = vec![0; 0x2000];
+    tileset2[..0x1000].copy_from_slice(rocket_tileset);
+    tileset2[0x1000..].copy_from_slice(&empty_tileset);
+
+    (tileset1, tileset2)
+    }
+
+
+pub fn get_current_tilesets(emu: &mut NesState) -> Vec<u8>{
+    let mut current_tileset = vec![0; 0x2000];
+    for i in 0..0x2000 {
+        current_tileset[i] = emu.ppu.read_byte(&mut *emu.mapper, i as u16)
+        }
+        current_tileset
+    }
+
+pub fn get_tile_select(emu: &mut NesState) -> u8 {
+    let bg_select = emu.ppu.control & 0x10 >> 4;
+    let sprite_select = emu.ppu.control & 0x08 >> 3;
+
+    // validate here that background and sprite tile selects match
+    assert_eq!(bg_select, sprite_select);
+
+    bg_select
+}
+
+pub fn test() {
+    let mut emu = util::emulator(None);
+
+    let (tileset1, tileset2) = get_expected_tilesets();
+
+    for _ in 0..4 {
+        emu.run_until_vblank();
+    }
+
+    // test menu tileset is active
+    let tile_select = get_tile_select(&mut emu);
+    let current_tileset = get_current_tilesets(&mut emu);
+    assert_eq!(tile_select, 0);
+    assert_eq!(current_tileset, tileset1);
+}