106 lines
2.4 KiB
Ruby
106 lines
2.4 KiB
Ruby
require 'test_helper'
|
|
|
|
class Elasticsearch::Model::MultipleTest < Test::Unit::TestCase
|
|
|
|
context "Adapter for multiple models" do
|
|
|
|
class ::DummyOne
|
|
include Elasticsearch::Model
|
|
|
|
index_name 'dummy'
|
|
document_type 'dummy_one'
|
|
|
|
def self.find(ids)
|
|
ids.map { |id| new(id) }
|
|
end
|
|
|
|
attr_reader :id
|
|
|
|
def initialize(id)
|
|
@id = id.to_i
|
|
end
|
|
end
|
|
|
|
module ::Namespace
|
|
class DummyTwo
|
|
include Elasticsearch::Model
|
|
|
|
index_name 'dummy'
|
|
document_type 'dummy_two'
|
|
|
|
def self.find(ids)
|
|
ids.map { |id| new(id) }
|
|
end
|
|
|
|
attr_reader :id
|
|
|
|
def initialize(id)
|
|
@id = id.to_i
|
|
end
|
|
end
|
|
end
|
|
|
|
class ::DummyTwo
|
|
include Elasticsearch::Model
|
|
|
|
index_name 'other_index'
|
|
document_type 'dummy_two'
|
|
|
|
def self.find(ids)
|
|
ids.map { |id| new(id) }
|
|
end
|
|
|
|
attr_reader :id
|
|
|
|
def initialize(id)
|
|
@id = id.to_i
|
|
end
|
|
end
|
|
|
|
HITS = [{_index: 'dummy',
|
|
_type: 'dummy_two',
|
|
_id: '2',
|
|
}, {
|
|
_index: 'dummy',
|
|
_type: 'dummy_one',
|
|
_id: '2',
|
|
}, {
|
|
_index: 'other_index',
|
|
_type: 'dummy_two',
|
|
_id: '1',
|
|
}, {
|
|
_index: 'dummy',
|
|
_type: 'dummy_two',
|
|
_id: '1',
|
|
}, {
|
|
_index: 'dummy',
|
|
_type: 'dummy_one',
|
|
_id: '3'}]
|
|
|
|
setup do
|
|
@multimodel = Elasticsearch::Model::Multimodel.new(DummyOne, DummyTwo, Namespace::DummyTwo)
|
|
end
|
|
|
|
context "when returning records" do
|
|
setup do
|
|
@multimodel.class.send :include, Elasticsearch::Model::Adapter::Multiple::Records
|
|
@multimodel.expects(:response).at_least_once.returns(stub(response: { 'hits' => { 'hits' => HITS } }))
|
|
end
|
|
|
|
should "keep the order from response" do
|
|
assert_instance_of Module, Elasticsearch::Model::Adapter::Multiple::Records
|
|
records = @multimodel.records
|
|
|
|
assert_equal 5, records.count
|
|
|
|
assert_kind_of ::Namespace::DummyTwo, records[0]
|
|
assert_kind_of ::DummyOne, records[1]
|
|
assert_kind_of ::DummyTwo, records[2]
|
|
assert_kind_of ::Namespace::DummyTwo, records[3]
|
|
assert_kind_of ::DummyOne, records[4]
|
|
|
|
assert_equal [2, 2, 1, 1, 3], records.map(&:id)
|
|
end
|
|
end
|
|
end
|
|
end
|