OperatorMatcher considers ancestor chain

lib/rspec/matchers/operator_matcher.rb

@@ -11,8 +11,17 @@ def register(klass, operator, matcher)
           registry[klass][operator] = matcher
         end
 
+        def unregister(klass, operator)
+          registry[klass] && registry[klass].delete(operator)
+        end
+
         def get(klass, operator)
-          registry[klass] && registry[klass][operator]
+          klass.ancestors.each { |ancestor|
+            matcher = registry[ancestor] && registry[ancestor][operator]
+            return matcher if matcher
+          }
+
+          nil
         end
       end
 

spec/rspec/matchers/operator_matcher_spec.rb

@@ -196,6 +196,29 @@
 
 end
 
+describe "OperatorMatcher registry" do
+  let(:custom_klass) { Class.new }
+  let(:custom_subklass) { Class.new(custom_klass) }
+
+  after {
+    RSpec::Matchers::OperatorMatcher.unregister(custom_klass, "=~")
+  }
+
+  it "allows operator matchers to be registered for types" do
+    RSpec::Matchers::OperatorMatcher.register(custom_klass, "=~", RSpec::Matchers::BuiltIn::Match)
+    expect(RSpec::Matchers::OperatorMatcher.get(custom_klass, "=~")).to eq(RSpec::Matchers::BuiltIn::Match)
+  end
+
+  it "considers ancestors when finding an operator matcher" do
+    RSpec::Matchers::OperatorMatcher.register(custom_klass, "=~", RSpec::Matchers::BuiltIn::Match)
+    expect(RSpec::Matchers::OperatorMatcher.get(custom_subklass, "=~")).to eq(RSpec::Matchers::BuiltIn::Match)
+  end
+
+  it "returns nil if there is no matcher registered for a type" do
+    expect(RSpec::Matchers::OperatorMatcher.get(custom_klass, "=~")).to be_nil
+  end
+end
+
 describe RSpec::Matchers::BuiltIn::PositiveOperatorMatcher do
 
   it "works when the target has implemented #send" do