Add specs for Cloud Vision sample

Author: remi Taylor

vision-sample/spec/vision_samples_spec.rb

@@ -14,12 +14,60 @@
 
 require_relative "../vision_samples"
 require "rspec"
-require "gcloud"
+require "tempfile"
 
 describe "Vision sample" do
 
-  it "detect labels"
-  it "detect landmark"
-  it "detect faces"
+  # Returns full path to sample image included in repository for testing
+  def image_path filename
+    File.expand_path "../images/#{filename}", __dir__
+  end
 
+  # Capture and return STDOUT output by block
+  def capture &block
+    real_stdout = $stdout
+    $stdout = StringIO.new
+    block.call
+    @captured_output = $stdout.string
+  ensure
+    $stdout = real_stdout
+  end
+  attr_reader :captured_output
+
+# cat.jpg
+# eiffel_tower.jpg
+# face.png
+
+  example "detect labels" do
+    capture { detect_labels image_path("cat.jpg") }
+
+    expect(captured_output).to start_with "Image labels:"
+    expect(captured_output).to include "cat"
+    expect(captured_output).to include "mammal"
+  end
+
+  example "detect landmark" do
+    expect { detect_landmark image_path("eiffel_tower.jpg") }.to output(
+      "Found landmark: Eiffel Tower\n"
+    ).to_stdout
+  end
+
+  example "detect faces" do
+    output_image_file = Tempfile.new "cloud-vision-testing"
+    expect(File.size output_image_file.path).to eq 0
+
+    begin
+      capture { detect_faces image_path("face.png"), output_image_file.path }
+
+      expect(captured_output).to include "Face bounds:"
+      expect(captured_output).to include "(154, 33)"
+      expect(captured_output).to include "(301, 33)"
+      expect(captured_output).to include "(301, 180)"
+      expect(captured_output).to include "(154, 180)"
+      expect(File.size output_image_file.path).to be > 0
+    ensure
+      output_image_file.close
+      output_image_file.unlink
+    end
+  end
 end

vision-sample/vision_samples.rb

@@ -84,8 +84,9 @@ def detect_faces path_to_image_file, path_to_output_file
   image = Magick::Image.read(path_to_image_file)[0]
 
   faces.each do |face|
+    puts "Face bounds:"
     face.bounds.face.each do |vector|
-      puts "#{vector.x}, #{vector.y}"
+      puts "(#{vector.x}, #{vector.y})"
     end
 
     draw        = Magick::Draw.new
@@ -116,22 +117,22 @@ def detect_faces path_to_image_file, path_to_output_file
   when "labels"
     detect_labels ARGV.shift
   when "landmarks"
-    detect_landmarks ARGV.shift
+    detect_landmark ARGV.shift
   when "faces"
     detect_faces ARGV.shift, ARGV.shift
   else
     puts <<-usage
 Usage: ruby vision_samples.rb <command> [arguments]
 
 Commands:
-  labels    <image-path>
-  landmarks <image-path>
-  faces     <image-path> <output-image-path>
+  labels   <image-path>
+  landmark <image-path>
+  faces    <image-path> <output-image-path>
 
 Examples:
-  ruby vision_samples.rb labels    /path/to/cat.jpg
-  ruby vision_samples.rb landmarks /path/to/grand-canyon.jpg
-  ruby vision_samples.rb faces     /path/to/faces.jpg output-image.jpg
+  ruby vision_samples.rb labels   /path/to/cat.jpg
+  ruby vision_samples.rb landmark /path/to/grand-canyon.jpg
+  ruby vision_samples.rb faces    /path/to/faces.jpg output-image.jpg
     usage
   end
 end