1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
use std::path::PathBuf;
use vnf::{Machine, MemoryPolicy};
use xixanta::opcodes::InstructionIdentifier;
#[derive(Default)]
struct Args {
file: String,
}
fn print_help() {
println!("End-to-end tests for the vnf library\n");
println!("usage: vnf-tests [OPTIONS] <tests/ directory>\n");
println!("Options:");
println!(" -h, --help\t\t\tPrint this message and quit.");
std::process::exit(0);
}
// Print the given `message` and exit(1).
fn die(message: String) -> ! {
eprintln!("error: {message}");
std::process::exit(1);
}
fn parse_arguments() -> Args {
let mut args = std::env::args();
let mut res = Args::default();
// Skip command name.
args.next();
for arg in args {
match arg.as_str() {
"-h" | "--help" => print_help(),
_ => {
if arg.starts_with('-') {
die(format!("don't know how to handle the '{arg}' flag"));
}
if !res.file.is_empty() {
die("cannot have multiple paths".to_string());
}
res.file = arg;
}
}
}
if res.file.is_empty() {
die("you have to provide a path to the tests/ directory".to_string());
}
res
}
fn run_break_mark_test(path: &String) -> Result<(), String> {
let rom = PathBuf::from(path).join("out/stack.nes");
let roms: String = rom.to_string_lossy().into_owned().to_string();
let mut machine = Machine::from(&roms, 0x8000, MemoryPolicy::default())?;
// Get out of <start>
let _ = machine.next_iteration();
// Assert that we are at the expected location.
assert!(matches!(
machine.current_instruction.identifier,
InstructionIdentifier::Brk
));
// Before vs after running 'brk'.
assert_eq!(machine.status_register.break_mark, 0);
assert!(!machine.status_register.brk);
let _ = machine.next_iteration();
assert_eq!(machine.status_register.break_mark, 0x42);
assert!(machine.status_register.brk);
// Skip instructions we don't care about here.
for _ in 0..11 {
let _ = machine.next_iteration();
}
// Before running 'plp', everything is still as expected.
assert!(matches!(
machine.current_instruction.identifier,
InstructionIdentifier::Plp
));
assert_eq!(machine.status_register.break_mark, 0x42);
assert!(machine.status_register.brk);
// After running 'plp'.
let _ = machine.next_iteration();
assert_eq!(machine.status_register.break_mark, 0x00);
assert!(!machine.status_register.brk);
Ok(())
}
fn main() {
let args = parse_arguments();
let file = PathBuf::from(args.file.clone());
if !file.is_dir() {
die("you have to provide a path to the tests/ directory".to_string());
}
if let Err(e) = run_break_mark_test(&args.file) {
die(e)
}
}
|
