spec/integration/repositories/extending_relations_spec.rb in rom-0.6.2 vs spec/integration/repositories/extending_relations_spec.rb in rom-0.7.0

@@ -20,9 +20,18 @@
         end
       end
     end
   end
 
+  after do
+    ROM::Memory::Relation.class_eval do
+      undef_method :freaking_cool?
+      class << self
+        undef_method :freaking_awesome?
+      end
+    end
+  end
+
   shared_examples_for 'extended relation' do
     it 'can extend relation class' do
       expect(rom.relations.users.class).to be_freaking_awesome
     end