MediaCodecTest: testVendorParameters to use input surface for video encoders

Bug: 229146532
Test: atest android.media.codec.cts.MediaCodecTest#testVendorParameters
Change-Id: I1a6b9fe09485046bc628fea901586e3b6ab6766e
diff --git a/tests/tests/media/codec/src/android/media/codec/cts/MediaCodecTest.java b/tests/tests/media/codec/src/android/media/codec/cts/MediaCodecTest.java
index 6ea7ef9..0c58173 100644
--- a/tests/tests/media/codec/src/android/media/codec/cts/MediaCodecTest.java
+++ b/tests/tests/media/codec/src/android/media/codec/cts/MediaCodecTest.java
@@ -2472,7 +2472,7 @@
                         format.setInteger(MediaFormat.KEY_I_FRAME_INTERVAL, IFRAME_INTERVAL);
                         format.setInteger(
                                 MediaFormat.KEY_COLOR_FORMAT,
-                                CodecCapabilities.COLOR_FormatYUV420Flexible);
+                                CodecCapabilities.COLOR_FormatSurface);
                     }
                 } else {
                     Log.i(TAG, info.getName() + " is in neither audio nor video domain; skipped");
@@ -2482,6 +2482,10 @@
                 codec.configure(
                         format, null, null,
                         info.isEncoder() ? MediaCodec.CONFIGURE_FLAG_ENCODE : 0);
+                Surface inputSurface = null;
+                if (videoCaps != null && info.isEncoder()) {
+                    inputSurface = codec.createInputSurface();
+                }
                 codec.start();
                 codec.unsubscribeFromVendorParameters(vendorParams);
                 codec.stop();