updated

Author: shubhangi-google

google-cloud-storage-control/samples/acceptance/helper.rb

@@ -16,6 +16,7 @@
 require "minitest/autorun"
 require "minitest/focus"
 require "minitest/hooks/default"
+require "google/cloud/storage/control"
 
 def random_bucket_name
   t = Time.now.utc.iso8601.gsub ":", "-"
@@ -61,3 +62,33 @@ def retry_resource_exhaustion
   end
   raise Google::Cloud::ResourceExhaustedError, "Maybe take a break from creating and deleting buckets for a bit"
 end
+
+# Waits until all Anywhere Caches for a given bucket are deleted.
+#
+# This method polls the Storage Control API, listing the Anywhere Caches
+# associated with the specified bucket. If caches are found, it waits and
+# retries with an exponential backoff strategy until no caches remain.
+#
+# @param bucket_name [String] The name of the Google Cloud Storage bucket.
+# @return [Integer] The final count of Anywhere Caches, which will be 0
+# the method completes successfully after all caches are deleted.
+def count_anywhere_caches bucket_name = nil
+  storage_control_client = Google::Cloud::Storage::Control.storage_control
+
+  # Set project to "_" to signify global bucket
+  parent = "projects/_/buckets/#{bucket_name}"
+  request = Google::Cloud::Storage::Control::V2::ListAnywhereCachesRequest.new(
+    parent: parent
+  )
+  result = storage_control_client.list_anywhere_caches request
+  min_delay = 180 # 3 minutes
+  max_delay = 900 # 15 minutes
+  while result.response.anywhere_caches.count != 0
+    puts "Cache not deleted yet, Retrying in #{min_delay} seconds."
+    sleep min_delay
+    min_delay = [min_delay * 2, max_delay].min # Exponential backoff with a max delay
+    result = storage_control_client.list_anywhere_caches request
+  end
+
+  result.response.anywhere_caches.count
+end 

google-cloud-storage-control/samples/acceptance/storage_control_anywhere_cache_test.rb

@@ -13,15 +13,13 @@
 # limitations under the License.
 
 require_relative "helper"
-require "google/cloud/storage/control"
 require_relative "../storage_control_create_anywhere_cache"
 require_relative "../storage_control_list_anywhere_caches"
 require_relative "../storage_control_get_anywhere_cache"
 require_relative "../storage_control_update_anywhere_cache"
 require_relative "../storage_control_pause_anywhere_cache"
 require_relative "../storage_control_resume_anywhere_cache"
 require_relative "../storage_control_disable_anywhere_cache"
-require 'pry'
 
 describe "Storage Control Anywhere Cache" do
   let(:bucket_name) { random_bucket_name }
@@ -30,11 +28,11 @@
   let(:anywhere_cache_name) { "projects/_/buckets/#{bucket_name}/anywhereCaches/#{zone}" }
 
   before :all do
-    bucket = create_bucket_helper bucket_name
+    create_bucket_helper bucket_name
   end
 
   after :all do
-    delete_bucket_helper bucket_name #{}until count_anywhere_caches(bucket_name).zero?
+    delete_bucket_helper bucket_name until count_anywhere_caches(bucket_name).zero?
   end
 
   it "handles Anywhere cache lifecycle in sequence" do
@@ -74,15 +72,3 @@
     assert_includes out_disable, "AnywhereCache #{anywhere_cache_name} disabled"
   end
 end
-
-# def count_anywhere_caches bucket_name
-#   sleep 900
-#   storage_control_client = Google::Cloud::Storage::Control.storage_control
-#   # Set project to "_" to signify global bucket
-#   parent = "projects/_/buckets/#{bucket_name}"
-#   request = Google::Cloud::Storage::Control::V2::ListAnywhereCachesRequest.new(
-#     parent: parent
-#   )
-#   result = storage_control_client.list_anywhere_caches request
-#   result.response.anywhere_caches.count
-# end

google-cloud-storage-control/samples/storage_control_create_anywhere_cache.rb

@@ -16,7 +16,7 @@
 require "google/cloud/storage/control"
 
 def create_anywhere_cache bucket_name:, zone:
-  # The ID of your GCS bucket
+  # The Name of your GCS bucket
   # bucket_name = "your-unique-bucket-name"
 
   # Zone where you want to create cache
@@ -41,25 +41,21 @@ def create_anywhere_cache bucket_name:, zone:
   # The cache is created in the specified bucket.
   begin
     operation = storage_control_client.create_anywhere_cache request
-    if operation.instance_of? Gapic::Operation
-      puts "AnywhereCache operation created - #{operation.name}"
-      get_request = Google::Cloud::Storage::Control::V2::GetAnywhereCacheRequest.new(
-        name: name
-      )
+    puts "AnywhereCache operation created - #{operation.name}"
+    get_request = Google::Cloud::Storage::Control::V2::GetAnywhereCacheRequest.new(
+      name: name
+    )
+    result = storage_control_client.get_anywhere_cache get_request
+    min_delay = 180 # 3 minutes
+    max_delay = 900 # 15 minutes
+    while result.state != "running"
+      puts "Cache not running yet, current state is #{result.state}. Retrying in #{min_delay} seconds."
+      sleep min_delay
+      min_delay = [min_delay * 2, max_delay].min # Exponential backoff with a max delay
       result = storage_control_client.get_anywhere_cache get_request
-      min_delay = 900 # 15 minutes
-      max_delay = 1800 # 30 minutes
-      while result.state != "running"
-        puts "Cache not running yet, current state is #{result.state}. Retrying in #{min_delay} seconds."
-        sleep min_delay
-        min_delay = [min_delay * 2, max_delay].min # Exponential backoff with a max delay
-        result = storage_control_client.get_anywhere_cache get_request
-      end
-      puts "AnywhereCache created - #{result.name}"
-    else
-      puts "AnywhereCache create operation failed"
     end
-  rescue Google::Cloud::Error=> e
+    puts "AnywhereCache created - #{result.name}"
+  rescue Google::Cloud::Error => e
     puts "Error creating AnywhereCache: #{e.message}"
   end
 end

google-cloud-storage-control/samples/storage_control_disable_anywhere_cache.rb

@@ -16,7 +16,7 @@
 require "google/cloud/storage/control"
 
 def disable_anywhere_cache bucket_name:, anywhere_cache_id:
-  # The ID of your GCS bucket
+  # The Name of your GCS bucket
   # bucket_name = "your-unique-bucket-name"
 
   # A value that, along with the bucket's name, uniquely identifies the cache
@@ -37,7 +37,7 @@ def disable_anywhere_cache bucket_name:, anywhere_cache_id:
   begin
     result = storage_control_client.disable_anywhere_cache request
     puts "AnywhereCache #{result.name} #{result.state} "
-  rescue StandardError => e
+  rescue Google::Cloud::Error => e
     puts "Error disabling AnywhereCache: #{e.message}"
   end
 end

google-cloud-storage-control/samples/storage_control_get_anywhere_cache.rb

@@ -16,7 +16,7 @@
 require "google/cloud/storage/control"
 
 def get_anywhere_cache bucket_name:, anywhere_cache_id:
-  # The ID of your GCS bucket
+  # The Name of your GCS bucket
   # bucket_name = "your-unique-bucket-name"
 
   # A value that, along with the bucket's name, uniquely identifies the cache
@@ -39,7 +39,7 @@ def get_anywhere_cache bucket_name:, anywhere_cache_id:
   begin
     result = storage_control_client.get_anywhere_cache request
     puts "AnywhereCache #{result.name} fetched"
-  rescue StandardError => e
+  rescue Google::Cloud::Error => e
     puts "Error fetching AnywhereCache: #{e.message}"
   end
 end

google-cloud-storage-control/samples/storage_control_list_anywhere_caches.rb

@@ -16,7 +16,7 @@
 require "google/cloud/storage/control"
 
 def list_anywhere_caches bucket_name:
-  # The ID of your GCS bucket
+  # The Name of your GCS bucket
   # bucket_name = "your-unique-bucket-name"
 
   # Create a client object. The client can be reused for multiple calls.
@@ -35,7 +35,7 @@ def list_anywhere_caches bucket_name:
     result.response.anywhere_caches.each do |item|
       puts "AnywhereCache #{item.name} found in list"
     end
-  rescue StandardError => e
+  rescue Google::Cloud::Error => e
     puts "Error listing AnywhereCaches: #{e.message}"
   end
 end

google-cloud-storage-control/samples/storage_control_pause_anywhere_cache.rb

@@ -16,7 +16,7 @@
 require "google/cloud/storage/control"
 
 def pause_anywhere_cache bucket_name:, anywhere_cache_id:
-  # The ID of your GCS bucket
+  # The Name of your GCS bucket
   # bucket_name = "your-unique-bucket-name"
 
   # A value that, along with the bucket's name, uniquely identifies the cache
@@ -38,7 +38,7 @@ def pause_anywhere_cache bucket_name:, anywhere_cache_id:
   begin
     result = storage_control_client.pause_anywhere_cache request
     puts "AnywhereCache #{result.name} #{result.state}"
-  rescue StandardError => e
+  rescue Google::Cloud::Error => e
     puts "Error pausing AnywhereCache: #{e.message}"
   end
 end

google-cloud-storage-control/samples/storage_control_resume_anywhere_cache.rb

@@ -16,7 +16,7 @@
 require "google/cloud/storage/control"
 
 def resume_anywhere_cache bucket_name:, anywhere_cache_id:
-  # The ID of your GCS bucket
+  # The Name of your GCS bucket
   # bucket_name = "your-unique-bucket-name"
 
   # A value that, along with the bucket's name, uniquely identifies the cache
@@ -38,7 +38,7 @@ def resume_anywhere_cache bucket_name:, anywhere_cache_id:
   begin
     result = storage_control_client.resume_anywhere_cache request
     puts "AnywhereCache #{result.name} #{result.state}"
-  rescue StandardError => e
+  rescue Google::Cloud::Error => e
     puts "Error resuming AnywhereCache: #{e.message}"
   end
 end

google-cloud-storage-control/samples/storage_control_update_anywhere_cache.rb

@@ -17,7 +17,7 @@
 require "google/cloud/storage/control/v2"
 
 def update_anywhere_cache bucket_name:, anywhere_cache_id:
-  # The ID of your GCS bucket
+  # The Name of your GCS bucket
   # bucket_name = "your-unique-bucket-name"
   # A value that, along with the bucket's name, uniquely identifies the cache
   # anywhere_cache_id = "us-east1-b"
@@ -43,17 +43,22 @@ def update_anywhere_cache bucket_name:, anywhere_cache_id:
   # The cache is identified by the specified ID.
   begin
     operation = storage_control_client.update_anywhere_cache request
-    if operation.instance_of? Gapic::Operation
-      puts "AnywhereCache operation created - #{operation.name}"
-      get_request = Google::Cloud::Storage::Control::V2::GetAnywhereCacheRequest.new(
-        name: name
-      )
+    puts "AnywhereCache operation created - #{operation.name}"
+    get_request = Google::Cloud::Storage::Control::V2::GetAnywhereCacheRequest.new(
+      name: name
+    )
+    result = storage_control_client.get_anywhere_cache get_request
+    min_delay = 120 # 2 minutes
+    max_delay = 600 # 10 minutes
+    while result.state != "running"
+      puts "Cache not running yet, current state is #{result.state}. Retrying in #{min_delay} seconds."
+      sleep min_delay
+      min_delay = [min_delay * 2, max_delay].min # Exponential backoff with a max delay
       result = storage_control_client.get_anywhere_cache get_request
-      puts "AnywhereCache #{result.name} updated"
-    else
-      puts "AnywhereCache update operation failed"
     end
-  rescue StandardError => e
+    puts "AnywhereCache #{result.name} updated"
+    
+  rescue Google::Cloud::Error => e
     puts "Error updating AnywhereCache: #{e.message}"
   end
 end