trivial: fix up imports and move a test

BUG=b:185593007
TEST=./run_tests

Change-Id: Ib49b327dfd3e41b720c9cce6573f4f3f91bc724a
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/2972280
Tested-by: George Engelbrecht <engeg@google.com>
Auto-Submit: George Engelbrecht <engeg@google.com>
Reviewed-by: Alex Klein <saklein@chromium.org>
Commit-Queue: George Engelbrecht <engeg@google.com>
diff --git a/api/controller/controller_util.py b/api/controller/controller_util.py
index c86dd85..341f5d2 100644
--- a/api/controller/controller_util.py
+++ b/api/controller/controller_util.py
@@ -10,8 +10,8 @@
 from chromite.lib import build_target_lib
 from chromite.lib import constants
 from chromite.lib.parser import package_info
-from chromite.lib.chroot_lib import Chroot
-from chromite.lib.sysroot_lib import Sysroot
+from chromite.lib import chroot_lib
+from chromite.lib import sysroot_lib
 
 class Error(Exception):
   """Base error class for the module."""
@@ -53,7 +53,7 @@
   if features:
     env['FEATURES'] = ' '.join(features)
 
-  chroot = Chroot(
+  chroot = chroot_lib.Chroot(
       path=path, cache_dir=cache_dir, chrome_root=chrome_root, env=env)
 
   return chroot
@@ -73,7 +73,7 @@
   """
   assert isinstance(sysroot_message, sysroot_pb2.Sysroot)
 
-  return Sysroot(sysroot_message.path)
+  return sysroot_lib.Sysroot(sysroot_message.path)
 
 
 def ParseGomaConfig(goma_message, chroot_path):