YUKI Hiroshi
null+****@clear*****
Wed Nov 27 22:00:54 JST 2013
YUKI Hiroshi 2013-11-27 22:00:54 +0900 (Wed, 27 Nov 2013) New Revision: 078de9feaf6c0e13c49807a99234e865f2013949 https://github.com/droonga/fluent-plugin-droonga/commit/078de9feaf6c0e13c49807a99234e865f2013949 Message: Add test for search distributor Added files: test/unit/plugin/distributor/test_search.rb Added: test/unit/plugin/distributor/test_search.rb (+205 -0) 100644 =================================================================== --- /dev/null +++ test/unit/plugin/distributor/test_search.rb 2013-11-27 22:00:54 +0900 (3201485) @@ -0,0 +1,205 @@ +# 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/distributor/search" + +class SearchDistributorTest < Test::Unit::TestCase + include PluginHelper + + def setup + setup_database + setup_plugin(Droonga::SearchDistributor) + end + + def teardown + teardown_plugin + teardown_database + end + + def test_distribute + envelope = { + "type" => "search", + "dataset" => "Droonga", + "body" => { + "queries" => { + "no_output" => { + "source" => "User", + }, + "no_records" => { + "source" => "User", + "output" => { + "elements" => ["count"], + }, + }, + "no_limit" => { + "source" => "User", + "output" => { + "format" => "complex", + "elements" => ["count", "records"], + }, + }, + "have_records" => { + "source" => "User", + "output" => { + "format" => "complex", + "elements" => ["count", "records"], + "attributes" => ["_key", "name", "age"], + "offset" => 1, + "limit" => 2, + }, + }, + # XXX we should write cases for... + # - sortBy(simple) + # - sortBy(rich) + # - sortBy(rich) with offset + # - sortBy(rich) with limit + # - sortBy(rich) with offset and limit + # - sortBy(simple) + output(limit, offset) + # - sortBy(rich) + # + output(limit, offset) + # - sortBy(rich) with offset + # + output(limit, offset) + # - sortBy(rich) with limit + # + output(limit, offset) + # - sortBy(rich) with offset and limit + # + output(limit, offset) + # and, we have to write cases for both unlimited and limited cases... + }, + }, + } + + @plugin.process("search", envelope) + + message = [] + no_records_reducer = { + "type" => "reduce", + "body" => { + "no_records" => { + "no_records_reduced" => { + "count" => { + "type" => "sum", + }, + }, + }, + }, + "inputs" => ["no_records"], + "outputs" => ["no_records_reduced"], + } + message << no_records_reducer + no_limit_reducer = { + "type" => "reduce", + "body" => { + "no_limit" => { + "no_limit_reduced" => { + "count" => { + "type" => "sum", + }, + "records" => { + "type" => "sort", + "order" => ["<"], + "offset" => 0, + "limit" => 0, + }, + }, + }, + }, + "inputs" => ["no_limit"], + "outputs" => ["no_limit_reduced"], + } + message << no_limit_reducer + have_records_reducer = { + "type" => "reduce", + "body" => { + "have_records" => { + "have_records_reduced" => { + "count" => { + "type" => "sum", + }, + "records" => { + "type" => "sort", + "order" => ["<"], + "offset" => 1, + "limit" => 2, + }, + }, + }, + }, + "inputs" => ["have_records"], + "outputs" => ["have_records_reduced"], + } + message << have_records_reducer + + gatherer = { + "type" => "gather", + "body" => { + "no_records_reduced" => "no_records", + "no_limit_reduced" => "no_limit", + "have_records_reduced" => "have_records", + }, + "inputs" => [ + "no_records_reduced", + "no_limit_reduced", + "have_records_reduced", + ], + "post" => true, + } + message << gatherer + searcher = { + "type" => "broadcast", + "dataset" => "Droonga", + "body" => { + "queries" => { + "no_output" => { + "source" => "User", + }, + "no_records" => { + "source" => "User", + "output" => { + "elements" => ["count"], + }, + }, + "no_limit" => { + "source" => "User", + "output" => { + "format" => "complex", + "elements" => ["count", "records"], + "offset" => 0, + "limit" => -1, + }, + }, + "have_records" => { + "source" => "User", + "output" => { + "format" => "complex", + "elements" => ["count", "records"], + "attributes" => ["_key", "name", "age"], + "offset" => 0, + "limit" => -1, + }, + }, + }, + }, + "command" => "search", + "outputs" => [ + "no_records", + "no_limit", + "have_records", + ], + "replica" => "random", + } + message << searcher + assert_equal(message, @posted.last.last) + end +end -------------- next part -------------- HTML����������������������������...Download
