Coverage for yaptide/routes/main_routes.py: 100%

1from flask_restful import Api, Resource 

2 

3from yaptide.routes.auth_routes import (AuthLogIn, AuthLogOut, AuthRefresh, 

4 AuthRegister, AuthStatus) 

5from yaptide.routes.batch_routes import Clusters, JobsBatch 

6from yaptide.routes.celery_routes import ConvertResource, JobsDirect 

7from yaptide.routes.common_sim_routes import (JobsResource, 

8 InputsResource, 

9 LogfilesResource, 

10 ResultsResource) 

11from yaptide.routes.keycloak_routes import AuthKeycloak 

12from yaptide.routes.task_routes import TasksResource 

13from yaptide.routes.user_routes import UserSimulations, UserUpdate 

14from yaptide.routes.utils.response_templates import yaptide_response 

15 

16 

17class HelloWorld(Resource): 

18 """Root route""" 

19 

20 @staticmethod 

21 def get(): 

22 """Root route get method""" 

23 return yaptide_response(message="Hello World!", code=200) 

24 

25 

26def initialize_routes(api: Api): 

27 """Function initializing routes""" 

28 api.add_resource(HelloWorld, "/") 

29 

30 api.add_resource(JobsDirect, "/jobs/direct") 

31 api.add_resource(JobsBatch, "/jobs/batch") 

32 

33 api.add_resource(JobsResource, "/jobs") 

34 

35 api.add_resource(TasksResource, "/tasks") 

36 

37 api.add_resource(ResultsResource, "/results") 

38 api.add_resource(InputsResource, "/inputs") 

39 api.add_resource(LogfilesResource, "/logfiles") 

40 

41 api.add_resource(ConvertResource, "/convert") 

42 

43 api.add_resource(UserSimulations, "/user/simulations") 

44 api.add_resource(UserUpdate, "/user/update") 

45 

46 api.add_resource(AuthRegister, "/auth/register") 

47 api.add_resource(AuthLogIn, "/auth/login") 

48 api.add_resource(AuthRefresh, "/auth/refresh") 

49 api.add_resource(AuthStatus, "/auth/status") 

50 api.add_resource(AuthLogOut, "/auth/logout") 

51 

52 api.add_resource(AuthKeycloak, "/auth/keycloak") 

53 

54 api.add_resource(Clusters, "/clusters")