Reland "Add DefaultUITraceListener to InputTests"

This reverts commit 30385832d2a806a74b127bf481995954e53427f5.

Only collect traces per-class to reduce test module run time.

Bug: 332714237
Bug: 210460522
Change-Id: I8a7e36b9a20c31548c6287a5b4ed901ce18950db
diff --git a/tests/Input/Android.bp b/tests/Input/Android.bp
index f367c38..06c2651 100644
--- a/tests/Input/Android.bp
+++ b/tests/Input/Android.bp
@@ -48,6 +48,7 @@
         "testables",
         "testng",
         "truth",
+        "ui-trace-collector",
     ],
     libs: [
         "android.test.mock",