import unittest.mock as mock
from .. import mgr
-from ..controllers import Controller, RESTController
-from . import ControllerTestCase, KVStoreMockMixin # pylint: disable=no-name-in-module
+from ..controllers import RESTController, Router
+from ..tests import ControllerTestCase, KVStoreMockMixin
# pylint: disable=W0613
-@Controller('/foo', secure=False)
+@Router('/foo', secure=False)
class FooResource(RESTController):
def create(self, password):
pass