Build API: Make autotest more generic.

BUG=chromium:933984
TEST=run_tests

Change-Id: I2764716ff89b8668db7621b3e9e6cfe808c8018a
Reviewed-on: https://chromium-review.googlesource.com/1480630
Commit-Ready: Alex Klein <saklein@chromium.org>
Tested-by: Alex Klein <saklein@chromium.org>
Reviewed-by: Alex Klein <saklein@chromium.org>
Reviewed-by: C Shapiro <shapiroc@chromium.org>
diff --git a/api/build_api.py b/api/build_api.py
index 00959eb..3df89e0 100644
--- a/api/build_api.py
+++ b/api/build_api.py
@@ -13,10 +13,10 @@
 from google.protobuf import symbol_database
 
 from chromite.api import service
-from chromite.api.gen import autotest_pb2
 from chromite.api.gen import build_api_pb2
 from chromite.api.gen import depgraph_pb2
 from chromite.api.gen import image_pb2
+from chromite.api.gen import test_archive_pb2
 from chromite.lib import commandline
 from chromite.lib import cros_build_lib
 from chromite.lib import osutils
@@ -232,9 +232,9 @@
   Args:
     router (Router): The router.
   """
-  router.Register(autotest_pb2)
   router.Register(depgraph_pb2)
   router.Register(image_pb2)
+  router.Register(test_archive_pb2)
 
 
 def main(argv):