Skip to content

Commit 3061596

Browse files
bors[bot]jitingcn
andauthored
Merge #63
63: Add support for class descendants r=curquiza a=jitingcn Co-authored-by: 寂听 <[email protected]> Co-authored-by: Jiting <[email protected]>
2 parents 75b4498 + 5b7b8c2 commit 3061596

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

lib/meilisearch/utilities.rb

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,8 @@ def get_model_classes
77
elsif Rails.application
88
Rails.application.eager_load!
99
end
10-
MeiliSearch.instance_variable_get :@included_in
10+
klasses = MeiliSearch.instance_variable_get(:@included_in)
11+
(klasses + klasses.map{ |klass| klass.descendants }.flatten).uniq
1112
end
1213

1314
def clear_all_indexes

spec/utilities_spec.rb

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,14 +17,20 @@ def self.model_name
1717

1818
meilisearch
1919
end
20+
21+
class DummyChild < Dummy
22+
end
23+
24+
class DummyGrandChild < DummyChild
25+
end
2026
end
2127

2228
after(:each) do
2329
MeiliSearch.instance_variable_set :@included_in, @included_in
2430
end
2531

2632
it "should get the models where MeiliSearch module was included" do
27-
(MeiliSearch::Utilities.get_model_classes - [Dummy]).should == []
33+
(MeiliSearch::Utilities.get_model_classes - [Dummy, DummyChild, DummyGrandChild]).should == []
2834
end
2935

3036
end

0 commit comments

Comments
 (0)