1 files changed, 49 insertions, 23 deletions
diff --git a/tests/display.rs b/tests/display.rs
index 887db53..a6d6eb1 100644
--- a/tests/display.rs
+++ b/tests/display.rs
@@ -1,27 +1,5 @@
-/*
-#[test]
-fn sinks_are_equivalent() {
-    use yaxpeax_arch::display::NoColorsSink;
-    use yaxpeax_arch::testkit::DisplaySinkWriteComparator;
-
-    let mut bare = String::new();
-    let mut through_sink = String::new();
-    for i in 0..u64::MAX {
-        bare.clear();
-        through_sink.clear();
-        let mut out = NoColorsSink {
-            out: &mut through_sink
-        };
-        let mut comparator = DisplaySinkWriteComparator {
-            sink1: &mut out,
-            sink1_check: |sink| { sink.out.as_str() },
-            sink2: &mut bare,
-            sink2_check: |sink| { sink.as_str() },
-        };
-    }
-}
-*/
 
+// this was something of a misfeature for these formatters..
 #[test]
 #[allow(deprecated)]
 fn formatters_are_not_feature_gated() {
@@ -86,3 +64,51 @@ fn display_sink_write_hex_helpers() {
         assert_eq!(buf, expected);
     }
 }
+
+#[cfg(feature="alloc")]
+#[test]
+fn sinks_are_equivalent() {
+    use yaxpeax_arch::display::{DisplaySink, FmtSink};
+    use yaxpeax_arch::testkit::DisplaySinkWriteComparator;
+
+    let mut bare = String::new();
+    let mut through_sink = String::new();
+    for i in 0..u16::MAX {
+        bare.clear();
+        through_sink.clear();
+        let mut out = FmtSink::new(&mut through_sink);
+        let mut comparator = DisplaySinkWriteComparator::new(
+            &mut out,
+            |sink| { sink.inner_ref().as_str() },
+            &mut bare,
+            |sink| { sink.as_str() },
+        );
+        comparator.write_u16(i).expect("write succeeds");
+        comparator.write_prefixed_u16(i).expect("write succeeds");
+        comparator.write_prefixed_i16(i as i16).expect("write succeeds");
+    }
+}
+
+#[cfg(all(feature="alloc", feature="color-new"))]
+#[test]
+fn ansi_sink_works() {
+    use yaxpeax_arch::color_new::ansi::AnsiDisplaySink;
+    use yaxpeax_arch::display::DisplaySink;
+
+    let mut buf = String::new();
+
+    let mut ansi_sink = AnsiDisplaySink::new(&mut buf, yaxpeax_arch::color_new::DefaultColors);
+
+    ansi_sink.span_start_immediate();
+    ansi_sink.write_prefixed_u8(0x80).expect("write succeeds");
+    ansi_sink.span_end_immediate();
+    ansi_sink.write_fixed_size("(").expect("write succeeds");
+    ansi_sink.span_start_register();
+    ansi_sink.write_fixed_size("rbp").expect("write succeeds");
+    ansi_sink.span_end_register();
+    ansi_sink.write_fixed_size(")").expect("write succeeds");
+
+    drop(ansi_sink);
+
+    assert_eq!(buf, "\x1b[37m0x80\x1b[39m(\x1b[38;5;6mrbp\x1b[39m)");
+}