Fixes to decoder, stdio, textprinter.

1 files changed, 8 insertions, 2 deletions

diff --git a/tests/test_decoder.c b/tests/test_decoder.c
index 0e6f19c..ed5a77e 100644
--- a/tests/test_decoder.c
+++ b/tests/test_decoder.c
@@ -16,13 +16,19 @@ int main() {
   upb_decoder *d = upb_decoder_new(upb_downcast_msgdef(fds));
   upb_decoder_reset(d, upb_stdio_bytesrc(in));
   upb_textprinter *p = upb_textprinter_new();
-  upb_textprinter_reset(p, upb_stdio_bytesink(out), false);
+  upb_handlers handlers;
+  upb_handlers_init(&handlers);
+  upb_textprinter_reset(p, &handlers, upb_stdio_bytesink(out), false);
+  upb_src *src = upb_decoder_src(d);
+  upb_src_sethandlers(src, &handlers);
 
   upb_status status = UPB_STATUS_INIT;
-  upb_streamdata(upb_decoder_src(d), upb_textprinter_sink(p), &status);
+  upb_src_run(src, &status);
 
+  upb_printerr(&status);
   assert(upb_ok(&status));
 
+
   upb_stdio_free(in);
   upb_stdio_free(out);
   upb_decoder_free(d);