fixing and prettifying tests

snf · Mar 16, 2016 · 1bf2831 · 1bf2831
1 parent 1c1ebc5
commit 1bf2831
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -5,19 +5,24 @@ pub use assembler::{Arch, Reloc, assemble, assemble2};
 
 #[test]
 fn x86_int3() {
- //assert_eq!(assemble(Arch::X86, "int3").unwrap(), [0xcc]);
+ assert_eq!(assemble(Arch::X86, "int3").unwrap(), [0xcc]);
 }
 
 #[test]
 fn x86_jcc() {
- let bytes = assemble(Arch::X86_64, "nop; jz here; nop; nop; here:").expect("couldn't compile the code");
+ let bytes = assemble(Arch::X86_64, "nop; jz here; nop; nop; here:").unwrap();
  assert_eq!(bytes, [0x90, 0x74, 0x02, 0x90, 0x90]);
 }
 
 #[test]
 fn x86_jmp_rel() {
  let addr = 0x1000;
  let reloc = Reloc::new("here", 0x1003);
- let bytes = assemble2(Arch::X86_64, "jmp here", addr, &[reloc]).expect("couldn't compile the code");
+ let bytes = assemble2(Arch::X86_64, "jmp here", addr, &[reloc]).unwrap();
  assert_eq!(bytes, [0xeb, 0x01]);
 }
+
+#[test]
+fn missing_label() {
+ assert_eq!(assemble(Arch::X86_64, "jz here"), None);
+}