Add test helper

lib/groonga/client/rails/fixture.rb

@@ -0,0 +1,47 @@
+# Copyright (C) 2016  Kouhei Sutou <kou@clear-code.com>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+
+require "groonga/client/rails/groonga_server_runner"
+
+module Groonga
+  class Client
+    module Rails
+      module Fixture
+        extend ActiveSupport::Concern
+
+        included do
+          setup do
+            setup_groonga
+          end
+
+          teardown do
+            teardown_groonga
+          end
+        end
+
+        def setup_groonga
+          @groonga_server_runner = GroongaServerRunner.new
+          @groonga_server_runner.run
+        end
+
+        def teardown_groonga
+          return if @groonga_server_runner.nil?
+          @groonga_server_runner.stop
+        end
+      end
+    end
+  end
+end

lib/groonga/client/rails/groonga_server_runner.rb

@@ -0,0 +1,137 @@
+# Copyright (C) 2016  Kouhei Sutou <kou@clear-code.com>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+
+require "rbconfig"
+require "fileutils"
+
+module Groonga
+  class Client
+    module Rails
+      class GroongaServerRunner
+        def initialize
+          @pid = nil
+          @url = build_url
+          @groonga = find_groonga
+          @tmp_dir = nil
+        end
+
+        def run
+          return if @groonga.nil?
+          @tmp_dir = create_tmp_dir
+          db_path = @tmp_dir + "db"
+          @pid = spawn(@groonga,
+                       "--port", @url.port.to_s,
+                       "--log-path", (@tmp_dir + "groonga.log").to_s,
+                       "--query-log-path", (@tmp_dir + "query.log").to_s,
+                       "--protocol", "http",
+                       "-s",
+                       "-n", db_path.to_s)
+          wait_groonga_ready
+        end
+
+        def stop
+          if @pid
+            Groonga::Client.open do |client|
+              client.shutdown
+            end
+            wait_groonga_shutdown
+          end
+          if @tmp_dir
+            FileUtils.rm_rf(@tmp_dir)
+          end
+        end
+
+        private
+        def build_url
+          default_options = Groonga::Client.default_options
+          url = default_options[:url]
+          if url.nil?
+            host = default_options[:host] || default_options[:address]
+            port = default_options[:port] || 10041
+            path = default_options[:path]
+            url = URI("http://#{host}:#{port}#{path}")
+          end
+          url
+        end
+
+        def find_groonga
+          paths = ENV["PATH"].split(File::PATH_SEPARATOR)
+          exeext = RbConfig::CONFIG["EXEEXT"]
+          paths.each do |path|
+            groonga = File.join(path, "groonga#{exeext}")
+            return groonga if File.executable?(groonga)
+          end
+          nil
+        end
+
+        def create_tmp_dir
+          tmpfs_dir = "/dev/shm"
+          if File.directory?(tmpfs_dir)
+            base_tmp_dir = Pathname(tmpfs_dir)
+          else
+            base_tmp_dir = ::Rails.root + "tmp"
+          end
+          tmp_dir = base_tmp_dir + "groonga-client-rails"
+          FileUtils.rm_rf(tmp_dir)
+          FileUtils.mkdir_p(tmp_dir)
+          tmp_dir
+        end
+
+        def wait_groonga_ready
+          n_retried = 0
+          while n_retried <= 20
+            n_retried += 1
+            sleep(0.05)
+            begin
+              TCPSocket.open(@url.host, @url.port) do
+              end
+            rescue SystemCallError
+              begin
+                pid = Process.waitpid(@pid, Process::WNOHANG)
+              rescue SystemCallError
+                @pid = nil
+                break
+              end
+            else
+              break
+            end
+          end
+        end
+
+        def wait_groonga_shutdown
+          # TODO: Remove me when Groonga 6.0.1 has been released.
+          # Workaround to shutdown as soon as possible.
+          begin
+            TCPSocket.open(@url.host, @url.port) do
+            end
+          rescue SystemCallError
+          end
+
+          n_retried = 0
+          while n_retried <= 20
+            n_retried += 1
+            sleep(0.05)
+            pid = Process.waitpid(@pid, Process::WNOHANG)
+            return if pid
+          end
+
+          Process.kill(:KILL, @pid)
+          Process.waitpid(@pid)
+        end
+      end
+    end
+  end
+end

lib/groonga/client/rails/test_help.rb

@@ -0,0 +1,17 @@
+# Copyright (C) 2016  Kouhei Sutou <kou@clear-code.com>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+
+require "groonga/client/rails/fixture"