From a9da6e4fbbfff3da75fa039661583de5af3e5178 Mon Sep 17 00:00:00 2001
From: Jacek <jacek@suchenia.pl>
Date: Sun, 17 Nov 2024 12:09:50 +0100
Subject: [PATCH] Implement pagination for list_jobs() and list_job_templates()

---
 moto/iot/models.py    | 40 ++++++----------------------------------
 moto/iot/responses.py | 10 +++++-----
 moto/iot/utils.py     | 14 +++++++++++++-
 3 files changed, 24 insertions(+), 40 deletions(-)

diff --git a/moto/iot/models.py b/moto/iot/models.py
index bfce837cfdce..d5d0f77c0e5f 100644
--- a/moto/iot/models.py
+++ b/moto/iot/models.py
@@ -2303,28 +2303,12 @@ def cancel_job(
     def get_job_document(self, job_id: str) -> FakeJob:
         return self.jobs[job_id]
 
-    def list_jobs(
-        self, max_results: int, token: Optional[str]
-    ) -> Tuple[List[Dict[str, Any]], Optional[str]]:
+    @paginate(PAGINATION_MODEL)  # type: ignore[misc]
+    def list_jobs(self) -> List[Dict[str, Any]]:
         """
         The following parameter are not yet implemented: Status, TargetSelection, ThingGroupName, ThingGroupId
         """
-        all_jobs = [_.to_dict() for _ in self.jobs.values()]
-        filtered_jobs = all_jobs
-
-        if token is None:
-            jobs = filtered_jobs[0:max_results]
-            next_token = str(max_results) if len(filtered_jobs) > max_results else None
-        else:
-            int_token = int(token)
-            jobs = filtered_jobs[int_token : int_token + max_results]
-            next_token = (
-                str(int_token + max_results)
-                if len(filtered_jobs) > int_token + max_results
-                else None
-            )
-
-        return jobs, next_token
+        return [_.to_dict() for _ in self.jobs.values()]
 
     def describe_job_execution(
         self, job_id: str, thing_name: str, execution_number: int
@@ -2658,21 +2642,9 @@ def create_job_template(
         self.jobs_templates[job_template_id] = job_template
         return job_template
 
-    def list_job_templates(
-        self, max_results: int, current_token: Optional[str]
-    ) -> Tuple[List[Dict[str, Any]], Optional[str]]:
-        all_job_templates = [_.to_dict() for _ in self.jobs_templates.values()]
-
-        if current_token is None:
-            start_index = 0
-        else:
-            start_index = int(current_token)
-
-        end_index = start_index + max_results
-        job_templates = all_job_templates[start_index:end_index]
-        next_token = str(end_index) if len(all_job_templates) > end_index else None
-
-        return job_templates, next_token
+    @paginate(PAGINATION_MODEL)  # type: ignore[misc]
+    def list_job_templates(self) -> List[Dict[str, Any]]:
+        return [_.to_dict() for _ in self.jobs_templates.values()]
 
     def delete_job_template(self, job_template_id: str) -> None:
         if job_template_id not in self.jobs_templates:
diff --git a/moto/iot/responses.py b/moto/iot/responses.py
index bd550f572a84..69885e2ca8e5 100644
--- a/moto/iot/responses.py
+++ b/moto/iot/responses.py
@@ -228,7 +228,7 @@ def list_jobs(self) -> str:
         max_results = self._get_int_param("maxResults", 50)
         previous_next_token = self._get_param("nextToken")
         jobs, next_token = self.iot_backend.list_jobs(
-            max_results=max_results, token=previous_next_token
+            max_results=max_results, next_token=previous_next_token
         )
 
         return json.dumps(dict(jobs=jobs, nextToken=next_token))
@@ -867,12 +867,12 @@ def create_job_template(self) -> str:
 
     def list_job_templates(self) -> str:
         max_results = self._get_int_param("maxResults", 50)
-        previous_next_token = self._get_param("nextToken")
-        job_templates, next_token = self.iot_backend.list_job_templates(
-            max_results=max_results, current_token=previous_next_token
+        current_next_token = self._get_param("nextToken")
+        job_templates, future_next_token = self.iot_backend.list_job_templates(
+            max_results=max_results, next_token=current_next_token
         )
 
-        return json.dumps(dict(jobTemplates=job_templates, nextToken=next_token))
+        return json.dumps(dict(jobTemplates=job_templates, nextToken=future_next_token))
 
     def delete_job_template(self) -> str:
         job_template_id = self._get_param("jobTemplateId")
diff --git a/moto/iot/utils.py b/moto/iot/utils.py
index be24e21ea924..fb71f639a484 100644
--- a/moto/iot/utils.py
+++ b/moto/iot/utils.py
@@ -6,7 +6,19 @@
         "limit_key": "max_results",
         "limit_default": 100,
         "unique_attribute": "jobId",
-    }
+    },
+    "list_job_templates": {
+        "input_token": "next_token",
+        "limit_key": "max_results",
+        "limit_default": 100,
+        "unique_attribute": "jobTemplateId",
+    },
+    "list_jobs": {
+        "input_token": "next_token",
+        "limit_key": "max_results",
+        "limit_default": 100,
+        "unique_attribute": "jobId",
+    },
 }