UrlFetchTitle: テスト追加、動作するように微調整

refs #5
cre-ne-jp · Mar 6, 2015 · 9d1f6f5 · 9d1f6f5
1 parent ca98a80
commit 9d1f6f5
Show file tree

Hide file tree

Showing 3 changed files with 58 additions and 0 deletions.
diff --git a/lib/rgrb/plugin/url_fetch_title/generator.rb b/lib/rgrb/plugin/url_fetch_title/generator.rb
@@ -2,6 +2,7 @@
 
 require 'uri'
 require 'mechanize'
+require 'rgrb/version'
 require 'rgrb/plugin/configurable_generator'
 
 module RGRB

diff --git a/spec/rgrb/plugin/url_fetch_title/data/cre_ne_jp.html b/spec/rgrb/plugin/url_fetch_title/data/cre_ne_jp.html
@@ -0,0 +1,10 @@
+<!DOCTYPE html>
+<html lang="ja">
+  <head>
+    <meta charset="UTF-8" />
+    <title>クリエイターズネットワーク</title>
+  </head>
+  <body>
+    <p><a href="http://www.cre.ne.jp/">クリエイターズネットワーク ホームページ</a></p>
+  </body>
+</html>
diff --git a/spec/rgrb/plugin/url_fetch_title/generator_spec.rb b/spec/rgrb/plugin/url_fetch_title/generator_spec.rb
@@ -0,0 +1,47 @@
+# vim: fileencoding=utf-8
+
+require_relative '../../../spec_helper'
+require 'rgrb/plugin/url_fetch_title/generator'
+
+describe RGRB::Plugin::UrlFetchTitle::Generator do
+  let(:generator) { described_class.new }
+
+  describe '#fetch_title' do
+    let(:default_prefix) { 'Fetch title: ' }
+    let(:default_suffix) { '[end]' }
+
+    before do
+      config = {
+        :@no_ssl_verify => [],
+        :@reply_prefix => default_prefix,
+        :@reply_suffix => default_suffix
+      }
+      config.each do |var, value|
+        generator.instance_variable_set(var, value)
+      end
+    end
+
+    context 'HTML ファイル' do
+      let(:url) { 'http://www.cre.ne.jp/' }
+
+      before do
+        html_path = File.expand_path(
+          'data/cre_ne_jp.html', File.dirname(__FILE__)
+        )
+        response = {
+          body: File.read(html_path),
+          status: 200,
+          headers: {
+            'Content-Type' => 'text/html; charset=UTF-8'
+          }
+        }
+        stub_request(:get, url).to_return(response)
+      end
+
+      subject { generator.fetch_title(url) }
+      it do
+        should eq("#{default_prefix}クリエイターズネットワーク#{default_suffix}")
+      end
+    end
+  end
+end