Add specs for Google Cloud Storage bucket samples

Author: remi Taylor

storage/Gemfile

@@ -14,8 +14,7 @@
 
 source "https://rubygems.org"
 
-# Google Cloud Client Library for Ruby
-gem "gcloud"
+gem "google-cloud-storage"
 
 group :test do
   gem "rake"

storage/Gemfile.lock

@@ -7,14 +7,7 @@ GEM
     digest-crc (0.4.1)
     faraday (0.9.2)
       multipart-post (>= 1.2, < 3)
-    gcloud (0.12.2)
-      digest-crc (~> 0.4)
-      google-api-client (~> 0.9.11)
-      google-protobuf (= 3.0.0.alpha.5.0.5.1)
-      grpc (= 1.0.0.pre1)
-      mime-types (>= 2.4, < 4.0)
-      zonefile (~> 1.04)
-    google-api-client (0.9.11)
+    google-api-client (0.9.13)
       addressable (~> 2.3)
       googleauth (~> 0.5)
       httpclient (~> 2.7)
@@ -23,8 +16,11 @@ GEM
       mime-types (>= 1.6)
       representable (~> 2.3.0)
       retriable (~> 2.0)
-      thor (~> 0.19)
-    google-protobuf (3.0.0.alpha.5.0.5.1)
+    google-cloud-core (0.20.1)
+    google-cloud-storage (0.20.0)
+      digest-crc (~> 0.4)
+      google-api-client (~> 0.9.11)
+      google-cloud-core (~> 0.20.0)
     googleauth (0.5.1)
       faraday (~> 0.9)
       jwt (~> 1.4)
@@ -33,17 +29,14 @@ GEM
       multi_json (~> 1.11)
       os (~> 0.9)
       signet (~> 0.7)
-    grpc (1.0.0.pre1)
-      google-protobuf (~> 3.0.0.alpha.5.0.3)
-      googleauth (~> 0.5.1)
-    httpclient (2.8.2.2)
+    httpclient (2.8.2.3)
     hurley (0.2)
     jwt (1.5.4)
     little-plugger (1.1.4)
     logging (2.1.0)
       little-plugger (~> 1.1)
       multi_json (~> 1.10)
-    memoist (0.14.0)
+    memoist (0.15.0)
     mime-types (3.1)
       mime-types-data (~> 3.2015)
     mime-types-data (3.2016.0521)
@@ -83,16 +76,14 @@ GEM
       faraday (~> 0.9)
       jwt (~> 1.5)
       multi_json (~> 1.10)
-    thor (0.19.1)
     uber (0.0.15)
     unicode-display_width (1.1.0)
-    zonefile (1.04)
 
 PLATFORMS
   ruby
 
 DEPENDENCIES
-  gcloud
+  google-cloud-storage
   rake
   rspec
   rubocop

storage/buckets.rb

@@ -16,9 +16,9 @@ def list_buckets project_id:
   # [START list_buckets]
   # project_id = "Your Google Cloud project ID"
 
-  require "gcloud"
+  require "google/cloud"
 
-  gcloud  = Gcloud.new project_id
+  gcloud  = Google::Cloud.new project_id
   storage = gcloud.storage
 
   storage.buckets.each do |bucket|
@@ -32,9 +32,9 @@ def create_bucket project_id:, bucket_name:
   # project_id  = "Your Google Cloud project ID"
   # bucket_name = "Name of Google Cloud Storage bucket to create"
 
-  require "gcloud"
+  require "google/cloud"
 
-  gcloud  = Gcloud.new project_id
+  gcloud  = Google::Cloud.new project_id
   storage = gcloud.storage
   bucket  = storage.create_bucket bucket_name
 
@@ -47,9 +47,9 @@ def delete_bucket project_id:, bucket_name:
   # project_id  = "Your Google Cloud project ID"
   # bucket_name = "Name of your Google Cloud Storage bucket to delete"
 
-  require "gcloud"
+  require "google/cloud"
 
-  gcloud  = Gcloud.new project_id
+  gcloud  = Google::Cloud.new project_id
   storage = gcloud.storage
   bucket  = storage.bucket bucket_name
 

storage/spec/buckets_spec.rb

@@ -0,0 +1,68 @@
+require_relative "../buckets"
+require "rspec"
+require "google/cloud"
+
+RSpec.describe "Google Cloud Storage bucket management" do
+
+  before :all do
+    @project_id  = ENV["GOOGLE_PROJECT_ID"]
+    @bucket_name = ENV["STORAGE_BUCKET"]
+    @gcloud      = Google::Cloud.new @project_id
+    @storage     = @gcloud.storage
+  end
+
+  before do
+    @storage.create_bucket @bucket_name unless @storage.bucket @bucket_name
+  end
+
+  after :all do
+    # Other tests assume that this bucket exists,
+    # so create it before exiting this spec suite
+    @storage.create_bucket @bucket_name unless @storage.bucket(@bucket_name)
+  end
+
+  def delete_bucket!
+    bucket = @storage.bucket @bucket_name
+
+    if bucket
+      bucket.files.each &:delete until bucket.files.empty?
+      bucket.delete
+    end
+  end
+
+  example "list buckets" do
+    expect {
+      list_buckets project_id: @project_id
+    }.to output(
+      /#{@bucket_name}/
+    ).to_stdout
+  end
+
+  example "create bucket" do
+    delete_bucket!
+
+    expect(@storage.bucket @bucket_name).to be nil
+
+    expect {
+      create_bucket project_id:  @project_id,
+                    bucket_name: @bucket_name
+    }.to output(
+      "Created bucket: #{@bucket_name}\n"
+    ).to_stdout
+
+    expect(@storage.bucket @bucket_name).not_to be nil
+  end
+
+  example "delete bucket" do
+    expect(@storage.bucket @bucket_name).not_to be nil
+
+    expect {
+      delete_bucket project_id: @project_id, bucket_name: @bucket_name
+    }.to output(
+      "Deleted bucket: #{@bucket_name}\n"
+    ).to_stdout
+
+    expect(@storage.bucket @bucket_name).to be nil
+  end
+
+end