Add logic to support pagination in resource listing

Author: Gildas Le Nadan

lib/awspec/helper/finder/backup.rb

@@ -4,30 +4,62 @@ module Awspec::Helper
   module Finder
     module Backup
       def find_backup_vault(id)
-        res = backup_client.list_backup_vaults
-        res.backup_vault_list.select do |v|
-          v.backup_vault_name == id || v.backup_vault_arn == id
-        end.single_resource(id)
+        selected = []
+        req = {}
+        loop do
+          res = backup_client.list_backup_vaults(req)
+          selected += res.backup_vault_list.select do |v|
+            v.backup_vault_name == id || v.backup_vault_arn == id
+          end
+          break if res.next_token.nil?
+
+          req[:next_token] = res.next_token
+        end
+        selected.single_resource(id)
       rescue Aws::Backup::Errors::ResourceNotFoundException
         nil
       end
 
       def find_backup_plan(id)
-        res = backup_client.list_backup_plans
-        res.backup_plans_list.select do |p|
-          p.backup_plan_name == id || p.backup_plan_arn == id || p.backup_plan_id == id
-        end.single_resource(id)
+        selected = []
+        req = {}
+        loop do
+          res = backup_client.list_backup_plans(req)
+          selected += res.backup_plans_list.select do |p|
+            p.backup_plan_name == id || p.backup_plan_arn == id || p.backup_plan_id == id
+          end
+          break if res.next_token.nil?
+
+          req[:next_token] = res.next_token
+        end
+        selected.single_resource(id)
       rescue Aws::Backup::Errors::ResourceNotFoundException
         nil
       end
 
       def find_backup_selection(id)
+        backup_plans = []
+        req = {}
+        loop do
+          res = backup_client.list_backup_plans(req)
+          backup_plans += res.backup_plans_list.map { |p| p.backup_plan_id }
+          break if res.next_token.nil?
+
+          req[:next_token] = res.next_token
+        end
+
         selected = []
-        plans = backup_client.list_backup_plans
-        plans.backup_plans_list.each do |p|
-          res = backup_client.list_backup_selections({ backup_plan_id: p.backup_plan_id })
-          selected += res.backup_selections_list.select do |s|
-            s.selection_id == id || s.selection_name == id
+        next_token = nil
+
+        backup_plans.each do |plan_id|
+          loop do
+            res = backup_client.list_backup_selections({ backup_plan_id: plan_id, next_token: next_token })
+            selected += res.backup_selections_list.select do |s|
+              s.selection_id == id || s.selection_name == id
+            end
+            break if res.next_token.nil?
+
+            next_token = res.next_token
           end
         end
         selected.single_resource(id)

lib/awspec/stub/backup_selection.rb

@@ -26,7 +26,8 @@
         creation_date: Time.new(2016, 10, 4, 9, 00, 00, '+00:00'),
         creator_request_id: nil,
         iam_role_arn: 'arn:aws:iam::111122223333:role/service-role/my-backup-service-role'
-      ]
+      ],
+      next_token: nil
     }
   }
 }