updating test cases with delay

Author: shubhangi-google

google-cloud-storage-control/README.md

@@ -141,7 +141,3 @@ might need to use one if there is no modern client available for the service.
 However, if a modern client is available, we generally recommend it over the
 older `google-apis-` clients.
 
-## Note
-
-Tests for Anywhere cache Sample are not included due to the long operation times typical for cache operations.
-

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

@@ -13,6 +13,7 @@
 # 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"
@@ -21,69 +22,66 @@
 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 }
-  let(:storage_client) { Google::Cloud::Storage.new }
-  let(:zone) {'us-east1-b'}
-#   let(:zone) {@bucket.location}
+  let(:zone) { "us-east1-b" }
+  # Set project to "_" to signify global bucket
+  let(:anywhere_cache_name) { "projects/_/buckets/#{bucket_name}/anywhereCaches/#{zone}" }
 
   before :all do
-    @bucket= storage_client.bucket bucket_name
-     @bucket = create_bucket_helper bucket_name
+    @bucket = create_bucket_helper bucket_name
   end
 
-  # after do
-  #   delete_bucket_helper bucket_name
-  # end
-
-  it "create Anywhere cache" do
-    create_anywhere_cache bucket_name: bucket_name, zone: zone
+  after do
+    delete_bucket_helper bucket_name until count_anywhere_caches bucket_name == 0
   end
 
-  # it "list Anywhere cache" do
-  #   # create_anywhere_cache bucket_name: bucket_name, zone: zone
+  it "handles Anywhere cache lifecycle in sequence" do
+    out_create, _err = capture_io do
+      create_anywhere_cache bucket_name: bucket_name, zone: zone
+    end
+    assert_includes out_create, "AnywhereCache created - #{anywhere_cache_name}"
 
-  #   out, _err = capture_io do
-  #       list_anywhere_caches bucket_name: bucket_name
-  #   end
+    out_list, _err = capture_io do
+      list_anywhere_caches bucket_name: bucket_name
+    end
+    assert_includes out_list, "AnywhereCache #{anywhere_cache_name} found in list"
 
-  #   assert_includes out, "#{bucket_name}/anywhereCaches/#{zone}"
-  # end
+    out_get, _err = capture_io do
+      get_anywhere_cache bucket_name: bucket_name, anywhere_cache_id: zone
+    end
+    assert_includes out_get, "AnywhereCache #{anywhere_cache_name} fetched"
 
-  # it "Get Anywhere cache" do
-  #   out, _err = capture_io do
-  #       get_anywhere_cache bucket_name: bucket_name, anywhere_cache_id: zone
-  #   end
-  #   assert_includes out, "#{bucket_name}/anywhereCaches/#{zone}"
-  # end
+    out_update, _err = capture_io do
+      update_anywhere_cache bucket_name: bucket_name, anywhere_cache_id: zone
+    end
+    assert_includes out_update, "AnywhereCache #{anywhere_cache_name} updated"
 
-  # it "Pause Anywhere cache" do
-  #   out, _err = capture_io do
-  #       pause_anywhere_cache bucket_name: bucket_name, anywhere_cache_id: zone
-  #   end
-  #   assert_includes out, "#{bucket_name}/anywhereCaches/#{zone} paused"
-  # end
+    out_pause, _err = capture_io do
+      pause_anywhere_cache bucket_name: bucket_name, anywhere_cache_id: zone
+    end
+    assert_includes out_pause, "AnywhereCache #{anywhere_cache_name} paused"
 
-  # it "Resume Anywhere cache" do
-  #   out, _err = capture_io do
-  #       resume_anywhere_cache bucket_name: bucket_name, anywhere_cache_id: zone
-  #   end
-  #   assert_includes out, "#{bucket_name}/anywhereCaches/#{zone} running"
-  # end
+    out_resume, _err = capture_io do
+      resume_anywhere_cache bucket_name: bucket_name, anywhere_cache_id: zone
+    end
+    assert_includes out_resume, "AnywhereCache #{anywhere_cache_name} running"
 
-  # it "Disable Anywhere cache" do
-  #   out, _err = capture_io do
-  #      disable_anywhere_cache bucket_name: bucket_name, anywhere_cache_id: zone
-  #   end
-  #   assert_includes out, "#{bucket_name}/anywhereCaches/#{zone} disabled"
-  # end
+    out_disable, _err = capture_io do
+      disable_anywhere_cache bucket_name: bucket_name, anywhere_cache_id: zone
+    end
+    assert_includes out_disable, "AnywhereCache #{anywhere_cache_name} disabled"
+  end
+end
 
-  # it "Update Anywhere cache" do
-  #   out, _err = capture_io do
-  #       update_anywhere_cache bucket_name: bucket_name, anywhere_cache_id: zone
-  #   end
-  #   assert_includes out, "#{bucket_name}/anywhereCaches/#{zone}"
-  # 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

@@ -24,6 +24,7 @@ def create_anywhere_cache bucket_name:, zone:
 
   # Create a client object. The client can be reused for multiple calls.
   storage_control_client = Google::Cloud::Storage::Control.storage_control
+  # Set project to "_" to signify global bucket
   parent = "projects/_/buckets/#{bucket_name}"
   name = "#{parent}/anywhereCaches/#{zone}"
 
@@ -40,24 +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.response? == false
-
-      puts "********************AnywhereCache operation created - #{operation.name}"
-      if !operation.done?
-        get_request = Google::Cloud::Storage::Control::V2::GetAnywhereCacheRequest.new(
-            name: name
-          )
+    if operation.class == Gapic::Operation
+      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
+      until result.state == "running"
+        sleep 1800 # Wait for 1/2 hour before checking again
         result = storage_control_client.get_anywhere_cache get_request
-        while result.state == "creating"
-          sleep 1800 # Wait for 1/2 hour before checking again
-          result = storage_control_client.get_anywhere_cache get_request
-          puts "********************AnywhereCache operation refreshed"
-          puts "********************AnywhereCache operation status check retried"
-        end
+        puts "AnywhereCache status check retried"
       end
+      puts "AnywhereCache created - #{result.name}"
+    else
+      puts "AnywhereCache create operation failed"
     end
-    puts "********************AnywhereCache create operation completed - #{operation.name}"
-
   rescue StandardError => e
     puts "Error creating AnywhereCache: #{e.message}"
   end

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

@@ -24,8 +24,9 @@ def disable_anywhere_cache bucket_name:, anywhere_cache_id:
 
   # Create a client object. The client can be reused for multiple calls.
   storage_control_client = Google::Cloud::Storage::Control.storage_control
+  # Set project to "_" to signify global bucket
   parent = "projects/_/buckets/#{bucket_name}"
-  name =  "#{parent}/anywhereCaches/#{anywhere_cache_id}"
+  name = "#{parent}/anywhereCaches/#{anywhere_cache_id}"
 
   # Create a request.
   request = Google::Cloud::Storage::Control::V2::DisableAnywhereCacheRequest.new(
@@ -35,7 +36,7 @@ def disable_anywhere_cache bucket_name:, anywhere_cache_id:
   # The cache can be re-enabled later.
   begin
     result = storage_control_client.disable_anywhere_cache request
-    puts "AnywhereCache #{result.name} #{result.state}"
+    puts "AnywhereCache #{result.name} #{result.state} "
   rescue StandardError => e
     puts "Error disabling AnywhereCache: #{e.message}"
   end

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

@@ -24,10 +24,11 @@ def get_anywhere_cache bucket_name:, anywhere_cache_id:
 
   # Create a client object. The client can be reused for multiple calls.
   storage_control_client = Google::Cloud::Storage::Control.storage_control
+  # Set project to "_" to signify global bucket
   parent = "projects/_/buckets/#{bucket_name}"
-  name =  "#{parent}/anywhereCaches/#{anywhere_cache_id}"
+  name = "#{parent}/anywhereCaches/#{anywhere_cache_id}"
 
-  # Create a request. Replace the placeholder values with actual data.
+  # Create a request.
   request = Google::Cloud::Storage::Control::V2::GetAnywhereCacheRequest.new(
     name: name
   )
@@ -37,7 +38,7 @@ def get_anywhere_cache bucket_name:, anywhere_cache_id:
 
   begin
     result = storage_control_client.get_anywhere_cache request
-    puts "AnywhereCache fetched - #{result.name}"
+    puts "AnywhereCache #{result.name} fetched"
   rescue StandardError => e
     puts "Error fetching AnywhereCache: #{e.message}"
   end

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

@@ -21,8 +21,10 @@ def list_anywhere_caches bucket_name:
 
   # Create a client object. The client can be reused for multiple calls.
   storage_control_client = Google::Cloud::Storage::Control.storage_control
+  # Set project to "_" to signify global bucket
   parent = "projects/_/buckets/#{bucket_name}"
 
+  # Create a request.
   request = Google::Cloud::Storage::Control::V2::ListAnywhereCachesRequest.new(
     parent: parent
   )
@@ -31,7 +33,7 @@ def list_anywhere_caches bucket_name:
   begin
     result = storage_control_client.list_anywhere_caches request
     result.response.anywhere_caches.each do |item|
-      puts item.name
+      puts "AnywhereCache #{item.name} found in list"
     end
   rescue StandardError => e
     puts "Error listing AnywhereCaches: #{e.message}"

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

@@ -24,10 +24,11 @@ def pause_anywhere_cache bucket_name:, anywhere_cache_id:
 
   # Create a client object. The client can be reused for multiple calls.
   storage_control_client = Google::Cloud::Storage::Control.storage_control
+  # Set project to "_" to signify global bucket
   parent = "projects/_/buckets/#{bucket_name}"
-  name =  "#{parent}/anywhereCaches/#{anywhere_cache_id}"
+  name = "#{parent}/anywhereCaches/#{anywhere_cache_id}"
 
-  # Create a request. Replace the placeholder values with actual data.
+  # Create a request.
   request = Google::Cloud::Storage::Control::V2::PauseAnywhereCacheRequest.new(
     name: name
   )

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

@@ -25,6 +25,7 @@ def resume_anywhere_cache bucket_name:, anywhere_cache_id:
   # Create a client object. The client can be reused for multiple calls.
   storage_control_client = Google::Cloud::Storage::Control.storage_control
   parent = "projects/_/buckets/#{bucket_name}"
+  # Set project to "_" to signify global bucket
   name = "#{parent}/anywhereCaches/#{anywhere_cache_id}"
 
   # Create a request.

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

@@ -14,22 +14,24 @@
 
 # [START storage_control_update_anywhere_cache]
 require "google/cloud/storage/control"
+require "google/cloud/storage/control/v2"
 
 def update_anywhere_cache bucket_name:, anywhere_cache_id:
   # The ID 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"
-  require "google/cloud/storage/control/v2"
 
   # Create a client object. The client can be reused for multiple calls.
   storage_control_client = Google::Cloud::Storage::Control.storage_control
+  # Set project to "_" to signify global bucket
   parent = "projects/_/buckets/#{bucket_name}"
   name = "#{parent}/anywhereCaches/#{anywhere_cache_id}"
+  ttl_in_seconds= 7200
 
   anywhere_cache = Google::Cloud::Storage::Control::V2::AnywhereCache.new(
     name: name,
-    ttl: 7200
+    ttl: ttl_in_seconds
   )
   mask = Google::Protobuf::FieldMask.new paths: ["ttl"]
   # Create a request.
@@ -40,8 +42,17 @@ def update_anywhere_cache bucket_name:, anywhere_cache_id:
   # The request updates the cache in the specified bucket.
   # The cache is identified by the specified ID.
   begin
-    result = storage_control_client.update_anywhere_cache request
-    puts "AnywhereCache updated - #{result.name}"
+    operation = storage_control_client.update_anywhere_cache request
+    if operation.class == Gapic::Operation
+      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
+      puts "AnywhereCache #{result.name} updated" if operation.class == Gapic::Operation
+    else
+      puts "AnywhereCache update operation failed"
+    end
   rescue StandardError => e
     puts "Error updating AnywhereCache: #{e.message}"
   end