1 files changed, 6 insertions, 2 deletions
diff --git a/test/decode_test_driver.h b/test/decode_test_driver.h
index 76cd12e02..72a279f00 100644
--- a/test/decode_test_driver.h
+++ b/test/decode_test_driver.h
@@ -66,15 +66,19 @@ class Decoder {
   }
 
   void Control(int ctrl_id, int arg) {
+    Control(ctrl_id, arg, VPX_CODEC_OK);
+  }
+
+  void Control(int ctrl_id, const void *arg) {
     InitOnce();
     const vpx_codec_err_t res = vpx_codec_control_(&decoder_, ctrl_id, arg);
     ASSERT_EQ(VPX_CODEC_OK, res) << DecodeError();
   }
 
-  void Control(int ctrl_id, const void *arg) {
+  void Control(int ctrl_id, int arg, vpx_codec_err_t expected_value) {
     InitOnce();
     const vpx_codec_err_t res = vpx_codec_control_(&decoder_, ctrl_id, arg);
-    ASSERT_EQ(VPX_CODEC_OK, res) << DecodeError();
+    ASSERT_EQ(expected_value, res) << DecodeError();
   }
 
   const char* DecodeError() {