Kouhei Sutou
null+****@clear*****
Mon Dec 12 11:31:45 JST 2016
Kouhei Sutou 2016-12-12 11:31:45 +0900 (Mon, 12 Dec 2016) New Revision: 0a0a311c12293de6b6051f2ed8c5e845866904d4 https://github.com/ranguba/groonga-client/commit/0a0a311c12293de6b6051f2ed8c5e845866904d4 Message: Import text helper from groonga-client-rails Added files: lib/groonga/client/spec-helper.rb lib/groonga/client/test-helper.rb lib/groonga/client/test/fixture.rb lib/groonga/client/test/groonga-server-runner.rb Added: lib/groonga/client/spec-helper.rb (+37 -0) 100644 =================================================================== --- /dev/null +++ lib/groonga/client/spec-helper.rb 2016-12-12 11:31:45 +0900 (5046fab) @@ -0,0 +1,37 @@ +# 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/test/fixture" + +module Groonga + class Client + module SpecHelper + include Test::Fixture + + class << self + def included(target) + target.before(:each) do + setup_groonga + end + + target.after(:each) do + teardown_groonga + end + end + end + end + end +end Added: lib/groonga/client/test-helper.rb (+37 -0) 100644 =================================================================== --- /dev/null +++ lib/groonga/client/test-helper.rb 2016-12-12 11:31:45 +0900 (f70cfb9) @@ -0,0 +1,37 @@ +# 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/test/fixture" + +module Groonga + class Client + module TestHelper + include Test::Fixture + + class << self + def included(target) + target.setup do + setup_groonga + end + + target.teardown do + teardown_groonga + end + end + end + end + end +end Added: lib/groonga/client/test/fixture.rb (+35 -0) 100644 =================================================================== --- /dev/null +++ lib/groonga/client/test/fixture.rb 2016-12-12 11:31:45 +0900 (8420a67) @@ -0,0 +1,35 @@ +# 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/test/groonga-server-runner" + +module Groonga + class Client + module Test + module Fixture + 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 Added: lib/groonga/client/test/groonga-server-runner.rb (+164 -0) 100644 =================================================================== --- /dev/null +++ lib/groonga/client/test/groonga-server-runner.rb 2016-12-12 11:31:45 +0900 (5aaaf64) @@ -0,0 +1,164 @@ +# 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 Test + class GroongaServerRunner + def initialize + @pid = nil + @using_running_server = false + @url = build_url + @groonga = find_groonga + @tmp_dir = nil + end + + def run + if groonga_server_running? + @using_running_server = true + else + return if****@groon*****? + @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 + sync_schema + end + + def stop + if @using_running_server + Groonga::Client.open do |client| + schema = client.schema + schema.tables.each do |name, _| + client.delete(table: name, + filter: "true") + end + end + else + if @pid + Groonga::Client.open do |client| + client.shutdown + end + wait_groonga_shutdown + end + if @tmp_dir + FileUtils.rm_rf(@tmp_dir) + end + 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 groonga_server_running? + begin + TCPSocket.open(@url.host, @url.port) do + end + rescue SystemCallError + false + else + true + end + end + + def sync_schema + ::Rails.application.eager_load! + ObjectSpace.each_object(Class) do |klass| + klass.sync_schema if klass < Searcher + end + 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.#{Process.pid}" + 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) + if groonga_server_running? + break + else + begin + pid = Process.waitpid(@pid, Process::WNOHANG) + rescue SystemCallError + @pid = nil + break + end + 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. + groonga_server_running? + + 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 -------------- next part -------------- HTML����������������������������...Download
