Accept open-uri as create_rich_menu_image argument

Author: Yoshiyuki Kinjo

lib/line/bot/client.rb

@@ -507,11 +507,17 @@ def get_rich_menu_image(rich_menu_id)
       def create_rich_menu_image(rich_menu_id, file)
         channel_token_required
 
-        content_type = case file.path
-                       when /\.jpe?g\z/i then 'image/jpeg'
-                       when /\.png\z/i   then 'image/png'
+        content_type = if file.respond_to?(:content_type)
+                         content_type = file.content_type
+                         raise ArgumentError, "invalid content type: #{content_type}" unless ['image/jpeg', 'image/png'].include?(content_type)
+                         content_type
                        else
-                         raise ArgumentError, "invalid file extension: #{file.path}"
+                         case file.path
+                         when /\.jpe?g\z/i then 'image/jpeg'
+                         when /\.png\z/i   then 'image/png'
+                         else
+                           raise ArgumentError, "invalid file extension: #{file.path}"
+                         end
                        end
 
         endpoint_path = "/bot/richmenu/#{rich_menu_id}/content"

spec/line/bot/rich_menu_spec.rb

@@ -2,6 +2,7 @@
 require 'webmock/rspec'
 require 'json'
 require 'tempfile'
+require 'open-uri'
 
 RICH_MENU_CONTENT = <<"EOS"
 {
@@ -177,6 +178,24 @@
       .with(body: File.open(RICH_MENU_IMAGE_FILE_PATH).read)
   end
 
+  it 'uploads and attaches am image to a rich menu from uri' do
+    uri_template = Addressable::Template.new Line::Bot::API::DEFAULT_BLOB_ENDPOINT + '/bot/richmenu/1234567/content'
+
+    stub_request(:post, uri_template).to_return(body: '{}', status: 200).with do |request|
+      expect(request.headers["Content-Type"]).to eq('image/png')
+    end
+
+    image_url = 'https://line.example.org/rich_menu.png'
+    image_content = File.open(RICH_MENU_IMAGE_FILE_PATH).read
+    image_content.force_encoding('ASCII-8BIT')
+    stub_request(:get, image_url).to_return(body: image_content, status: 200, headers: {'Content-type': 'image/png' })
+
+    client.create_rich_menu_image('1234567', open(image_url))
+
+    expect(WebMock).to have_requested(:post, Line::Bot::API::DEFAULT_BLOB_ENDPOINT + '/bot/richmenu/1234567/content')
+      .with(body: image_content)
+  end
+
   it "uploads invalid extension's file" do
     uri_template = Addressable::Template.new Line::Bot::API::DEFAULT_ENDPOINT + '/bot/richmenu/1234567/content'
     stub_request(:post, uri_template).to_return(body: '{}', status: 200)
@@ -186,4 +205,17 @@
       end
     end.to raise_error(ArgumentError)
   end
+
+  it "uploads invalid content type uri" do
+    uri_template = Addressable::Template.new Line::Bot::API::DEFAULT_ENDPOINT + '/bot/richmenu/1234567/content'
+    stub_request(:post, uri_template).to_return(body: '{}', status: 200)
+
+    text_url = 'https://line.example.org/rich_menu.txt'
+    text_content = File.open(RICH_MENU_INVALID_FILE_EXTENSION_PATH).read
+    stub_request(:get, text_url).to_return(body: text_content, status: 200, headers: {'Content-type': 'plain/text' })
+
+    expect do
+      client.create_rich_menu_image('1234567', open(text_url))
+    end.to raise_error(ArgumentError)
+  end
 end