fluent-plugin-droonga 0.0.2

data/test/plugin/handler/test_handler_add.rb

@@ -0,0 +1,135 @@
+# Copyright (C) 2013 droonga project
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License version 2.1 as published by the Free Software Foundation.
+#
+# 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+
+require "droonga/plugin/handler_add"
+
+class AddHandlerTest < Test::Unit::TestCase
+  def setup
+    setup_database
+    setup_schema
+    setup_handler
+  end
+
+  def teardown
+    teardown_handler
+    teardown_database
+  end
+
+  private
+  def setup_database
+    FileUtils.rm_rf(@database_path.dirname.to_s)
+    FileUtils.mkdir_p(@database_path.dirname.to_s)
+    @database = Groonga::Database.create(:path => @database_path.to_s)
+  end
+
+  def setup_schema
+  end
+
+  def teardown_database
+    @database.close
+    @database = nil
+    FileUtils.rm_rf(@database_path.dirname.to_s)
+  end
+
+  def setup_handler
+    @worker = StubWorker.new
+    @handler = Droonga::AddHandler.new(@worker)
+  end
+
+  def teardown_handler
+    @handler = nil
+  end
+
+  public
+  class HasKeyTest < self
+    def setup_schema
+      Groonga::Schema.define do |schema|
+        schema.create_table("Users",
+                            :type => :hash,
+                            :key_type => :short_text) do |table|
+          table.short_text("country")
+        end
+      end
+    end
+
+    def test_empty_values
+      request = {
+        "table"  => "Users",
+        "key"    => "mori",
+        "values" => {},
+      }
+      mock(@handler).emit([true])
+      @handler.add(request)
+      table = @worker.context["Users"]
+      assert_equal(["mori"], table.collect(&:key))
+    end
+
+    def test_values
+      request = {
+        "table"  => "Users",
+        "key"    => "asami",
+        "values" => {"country" => "japan"},
+      }
+      mock(@handler).emit([true])
+      @handler.add(request)
+      table = @worker.context["Users"]
+      assert_equal(["japan"], table.collect(&:country))
+    end
+  end
+
+  class NoKeyTest < self
+    def setup_schema
+      Groonga::Schema.define do |schema|
+        schema.create_table("Books",
+                            :type => :array) do |table|
+          table.short_text("title")
+        end
+      end
+    end
+
+    def test_empty_values
+      request = {
+        "table"  => "Books",
+        "values" => {},
+      }
+      mock(@handler).emit([true])
+      @handler.add(request)
+      table = @worker.context["Books"]
+      assert_equal([nil], table.collect(&:title))
+    end
+
+    def test_with_values
+      request = {
+        "table"  => "Books",
+        "values" => {"title" => "CSS"},
+      }
+      mock(@handler).emit([true])
+      @handler.add(request)
+      table = @worker.context["Books"]
+      assert_equal(["CSS"], table.collect(&:title))
+    end
+  end
+
+  class FailureTest < self
+    def test_nonexistent_table
+      request = {
+        "table"  => "Nonexistent",
+        "values" => {},
+      }
+      mock(@handler).emit([false])
+      @handler.add(request)
+    end
+  end
+end

data/test/plugin/handler/test_handler_groonga.rb

@@ -0,0 +1,64 @@
+# Copyright (C) 2013 droonga project
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License version 2.1 as published by the Free Software Foundation.
+#
+# 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+
+require "droonga/plugin/handler_groonga"
+
+class GroongaHandlerTest < Test::Unit::TestCase
+  def setup
+    setup_database
+    setup_handler
+  end
+
+  def teardown
+    teardown_handler
+    teardown_database
+  end
+
+  private
+  def setup_database
+    FileUtils.rm_rf(@database_path.dirname.to_s)
+    FileUtils.mkdir_p(@database_path.dirname.to_s)
+    @database = Groonga::Database.create(:path => @database_path.to_s)
+  end
+
+  def teardown_database
+    @database.close
+    @database = nil
+    FileUtils.rm_rf(@database_path.dirname.to_s)
+  end
+
+  def setup_handler
+    @worker = StubWorker.new
+    @handler = Droonga::GroongaHandler.new(@worker)
+  end
+
+  def teardown_handler
+    @handler = nil
+  end
+
+  private
+  def dump
+    database_dumper = Groonga::DatabaseDumper.new(:database => @database)
+    database_dumper.dump
+  end
+
+  NORMALIZED_START_TIME = Time.parse("2013-07-11T16:04:28+0900").to_i
+  NORMALIZED_ELAPSED_TIME = 1
+  def normalize_header(header)
+    start_time = NORMALIZED_START_TIME
+    elapsed_time = NORMALIZED_ELAPSED_TIME
+    [header[0], start_time, elapsed_time]
+  end
+end

data/test/plugin/handler/test_handler_search.rb

@@ -0,0 +1,512 @@
+# Copyright (C) 2013 droonga project
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License version 2.1 as published by the Free Software Foundation.
+#
+# 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+
+require "droonga/plugin/handler_search"
+
+class SearchHandlerTest < Test::Unit::TestCase
+  def setup
+    setup_database
+    setup_handler
+  end
+
+  def teardown
+    teardown_handler
+    teardown_database
+  end
+
+  private
+  def setup_database
+    restore(fixture_data("document.grn"))
+    @database = Groonga::Database.open(@database_path.to_s)
+  end
+
+  def teardown_database
+    @database.close
+    @database = nil
+  end
+
+  def setup_handler
+    @worker = StubWorker.new
+    @handler = Droonga::SearchHandler.new(@worker)
+  end
+
+  def teardown_handler
+    @handler = nil
+  end
+
+  def search(request)
+    @handler.search(request)
+    normalize_result_set(@worker.body)
+  end
+
+  def normalize_result_set(result_set)
+    result_set.each do |name, result|
+      result["startTime"] = start_time if result["startTime"]
+      result["elapsedTime"] = elapsed_time if result["elapsedTime"]
+    end
+    result_set
+  end
+
+  def start_time
+    "2013-01-31T14:34:47+09:00"
+  end
+
+  def elapsed_time
+    0.01
+  end
+
+  def assert_search(expected, request)
+    assert_equal(expected, search(request))
+  end
+
+  class NoParameterTest < self
+    def test_empty
+      assert_search({}, {})
+    end
+  end
+
+  class QueriesTest < self
+    def test_empty
+      assert_search({}, {"queries" => {}})
+    end
+  end
+
+  class HashQueryTest < self
+    def test_string_matchTo
+      request = base_request
+      request["queries"]["sections-result"]["condition"] = {
+        "query" => "Groonga",
+        "matchTo" => "title"
+      }
+      assert_search({
+                      "sections-result" => {
+                        "records" => [
+                          { "title" => "Groonga overview" },
+                        ],
+                      },
+                    },
+                    request)
+    end
+
+    def test_array_matchTo
+      request = base_request
+      request["queries"]["sections-result"]["condition"] = {
+        "query" => "Groonga",
+        "matchTo" => ["title"]
+      }
+      assert_search({
+                      "sections-result" => {
+                        "records" => [
+                          { "title" => "Groonga overview" },
+                        ],
+                      },
+                    },
+                    request)
+    end
+
+    def base_request
+      {
+        "queries" => {
+          "sections-result" => {
+            "source" => "Sections",
+            "output" => {
+              "elements" => [
+                "records",
+              ],
+              "format" => "complex",
+              "limit" => 1,
+              "attributes" => ["title"],
+            },
+          },
+        },
+      }
+    end
+  end
+
+  class SourceTest < self
+    def test_non_existent
+      assert_raise(Droonga::Searcher::UndefinedSourceError) do
+        search({
+                 "queries" => {
+                   "non-existent-result" => {
+                     "source" => "non-existent",
+                   },
+                 },
+               })
+      end
+    end
+
+    def test_existent
+      assert_search({
+                      "sections-result" => {},
+                    },
+                    {
+                      "queries" => {
+                        "sections-result" => {
+                          "source" => "Sections",
+                          "output" => {},
+                        },
+                      },
+                    })
+    end
+  end
+
+  class OutputTest < self
+    def test_count
+      assert_search({
+                      "sections-result" => {
+                        "count" => 9,
+                      },
+                    },
+                    {
+                      "queries" => {
+                        "sections-result" => {
+                          "source" => "Sections",
+                          "output" => {
+                            "elements" => [
+                              "count",
+                            ],
+                          },
+                        },
+                      },
+                    })
+    end
+
+    def test_elapsed_time
+      assert_search({
+                      "sections-result" => {
+                        "startTime" => start_time,
+                        "elapsedTime" => elapsed_time,
+                      },
+                    },
+                    {
+                      "queries" => {
+                        "sections-result" => {
+                          "source" => "Sections",
+                          "output" => {
+                            "elements" => [
+                              "startTime",
+                              "elapsedTime",
+                            ],
+                          },
+                        },
+                      },
+                    })
+    end
+
+    class AttributesTest < self
+      def test_source_only
+        expected = {
+          "sections-result" => {
+            "records" => [
+              {
+                "_key" => "1.1",
+                "title" => "Groonga overview",
+              },
+              {
+                "_key" => "1.2",
+                "title" => "Full text search and Instant update",
+              },
+              {
+                "_key" => "1.3",
+                "title" => "Column store and aggregate query",
+              },
+            ],
+          },
+        }
+        request = {
+          "queries" => {
+            "sections-result" => {
+              "source" => "Sections",
+              "output" => {
+                "elements" => [
+                  "records",
+                ],
+                "format" => "complex",
+                "limit" => 3,
+                "attributes" => ["_key", "title"],
+              },
+            },
+          },
+        }
+        assert_search(expected, request)
+      end
+
+      def test_label
+        expected = {
+          "sections-result" => {
+            "records" => [
+              {
+                "key" => "1.1",
+                "title" => "Groonga overview",
+              },
+              {
+                "key" => "1.2",
+                "title" => "Full text search and Instant update",
+              },
+              {
+                "key" => "1.3",
+                "title" => "Column store and aggregate query",
+              },
+            ],
+          },
+        }
+        request = {
+          "queries" => {
+            "sections-result" => {
+              "source" => "Sections",
+              "output" => {
+                "elements" => [
+                  "records",
+                ],
+                "format" => "complex",
+                "limit" => 3,
+                "attributes" => [
+                  {
+                    "label" => "key",
+                    "source" => "_key",
+                  },
+                  "title",
+                ],
+              },
+            },
+          },
+        }
+        assert_search(expected, request)
+      end
+
+      def test_static_value
+        expected = {
+          "sections-result" => {
+            "records" => [
+              {
+                "single_quote_string" => "string value",
+                "double_quote_string" => "string value",
+                "integer" => 29,
+                "complex_negative_number" => -29.29,
+              },
+              {
+                "single_quote_string" => "string value",
+                "double_quote_string" => "string value",
+                "integer" => 29,
+                "complex_negative_number" => -29.29,
+              },
+            ],
+          },
+        }
+        request = {
+          "queries" => {
+            "sections-result" => {
+              "source" => "Sections",
+              "output" => {
+                "elements" => [
+                  "records",
+                ],
+                "format" => "complex",
+                "limit" => 2,
+                "attributes" => [
+                  {
+                    "label" => "single_quote_string",
+                    "source" => "'string value'",
+                  },
+                  {
+                    "label" => "double_quote_string",
+                    "source" => '"string value"',
+                  },
+                  {
+                    "label" => "integer",
+                    "source" => "29",
+                  },
+                  {
+                    "label" => "complex_negative_number",
+                    "source" => "-29.29",
+                  },
+                ],
+              },
+            },
+          },
+        }
+        assert_search(expected, request)
+      end
+
+      def test_expression
+        expected = {
+          "sections-result" => {
+            "records" => [
+              {
+                "formatted title" => "<Groonga overview>",
+                "title" => "Groonga overview",
+              },
+            ],
+          },
+        }
+        request = {
+          "queries" => {
+            "sections-result" => {
+              "source" => "Sections",
+              "output" => {
+                "elements" => [
+                  "records",
+                ],
+                "format" => "complex",
+                "limit" => 1,
+                "attributes" => [
+                  "title",
+                  {
+                    "label" => "formatted title",
+                    "source" => "'<' + title + '>'",
+                  },
+                ],
+              },
+            },
+          },
+        }
+        assert_search(expected, request)
+      end
+
+      def test_snippet_html
+        expected = {
+          "sections-result" => {
+            "records" => [
+              {
+                "title" => "Groonga overview",
+                "snippet" => [
+                  "<span class=\"keyword\">Groonga</span> overview",
+                ],
+              },
+            ],
+          },
+        }
+        request = {
+          "queries" => {
+            "sections-result" => {
+              "source" => "Sections",
+              "condition" => {
+                "query" => "Groonga",
+                "matchTo" => ["title"],
+              },
+              "output" => {
+                "elements" => [
+                  "records",
+                ],
+                "format" => "complex",
+                "limit" => 1,
+                "attributes" => [
+                  "title",
+                  {
+                    "label" => "snippet",
+                    "source" => "snippet_html(title)",
+                  },
+                ],
+              },
+            },
+          },
+        }
+        assert_search(expected, request)
+      end
+    end
+
+    class FormatTest < self
+      def test_complex
+        request = {
+          "queries" => {
+            "sections-result" => {
+              "source" => "Sections",
+              "output" => {
+                "elements" => [
+                  "records",
+                ],
+                "format" => "complex",
+                "limit" => 3,
+                "attributes" => ["_key", "title"],
+              },
+            },
+          },
+        }
+        assert_search(complex_result, request)
+      end
+
+      def test_simple
+        request = {
+          "queries" => {
+            "sections-result" => {
+              "source" => "Sections",
+              "output" => {
+                "elements" => [
+                  "records",
+                ],
+                "format" => "simple",
+                "limit" => 3,
+                "attributes" => ["_key", "title"],
+              },
+            },
+          },
+        }
+        assert_search(simple_result, request)
+      end
+
+      def test_default
+        request = {
+          "queries" => {
+            "sections-result" => {
+              "source" => "Sections",
+              "output" => {
+                "elements" => [
+                  "records",
+                ],
+                "limit" => 3,
+                "attributes" => ["_key", "title"],
+              },
+            },
+          },
+        }
+        assert_search(simple_result, request)
+      end
+
+      def complex_result
+        {
+          "sections-result" => {
+            "records" => [
+              {
+                "_key" => "1.1",
+                "title" => "Groonga overview",
+              },
+              {
+                "_key" => "1.2",
+                "title" => "Full text search and Instant update",
+              },
+              {
+                "_key" => "1.3",
+                "title" => "Column store and aggregate query",
+              },
+            ],
+          },
+        }
+      end
+
+      def simple_result
+        {
+          "sections-result" => {
+            "records" => [
+              ["1.1", "Groonga overview"],
+              ["1.2", "Full text search and Instant update"],
+              ["1.3", "Column store and aggregate query"],
+            ],
+          },
+        }
+      end
+    end
+  end
+end