chore: setup for db migration

usnistgov · Nov 9, 2023 · 71dedfc · 71dedfc
1 parent f3cf528
commit 71dedfc
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/src/dioptra/restapi/group/service.py b/src/dioptra/restapi/group/service.py
@@ -57,7 +57,6 @@ def get_all(**kwargs) -> List[Group]:
         return Group.query.all()  # type: ignore
 
     @staticmethod
-    def get_by_id(group_id: int, **kwargs) -> Group:
     def get_by_id(group_id: int, **kwargs) -> Group:
         log: BoundLogger = kwargs.get("log", LOGGER.new())  # noqa: F841
 

diff --git a/src/dioptra/restapi/routes.py b/src/dioptra/restapi/routes.py
@@ -38,10 +38,14 @@ def register_routes(api: Api, app: Flask) -> None:
     from .queue import register_routes as attach_job_queue
     from .task_plugin import register_routes as attach_task_plugin
     from .user import register_routes as attach_user
+    from .group import register_routes as attach_group
+    from .group_membership import register_routes as attach_group_membership
 
     attach_auth(api, app)
     attach_experiment(api, app)
     attach_job(api, app)
     attach_job_queue(api, app)
     attach_task_plugin(api, app)
     attach_user(api, app)
+    attach_group(api, app)
+    attach_group_membership(api, app)
diff --git a/tests/unit/restapi/group/test_group.py b/tests/unit/restapi/group/test_group.py
@@ -70,7 +70,6 @@ def create_user(db: SQLAlchemy) -> User:
         created_on=timestamp,
         last_modified_on=timestamp,
         last_login_on=timestamp,
-        user_expire_on=user_expire_on,
         password_expire_on=password_expire_on,
     )
     db.session.add(new_user)