spec/mock_data_spec.rb in cistern-2.4.1 vs spec/mock_data_spec.rb in cistern-2.5.0

@@ -1,24 +1,26 @@
 require 'spec_helper'
 
 describe 'mock data' do
-  class Diagnosis < Sample::Request
-    def real(diagnosis)
-    end
+  before {
+    class Sample::Diagnosis < Sample::Request
+      def real(diagnosis)
+      end
 
-    def mock(diagnosis)
-      cistern.data.store(:diagnosis, cistern.data.fetch(:diagnosis) + [diagnosis])
+      def mock(diagnosis)
+        cistern.data.store(:diagnosis, cistern.data.fetch(:diagnosis) + [diagnosis])
+      end
     end
-  end
 
-  class Treat < Sample::Request
-    def real(treatment)
-    end
+    class Sample::Treat < Sample::Request
+      def real(treatment)
+      end
 
-    def mock(treatment)
-      cistern.data[:treatments] += [treatment]
+      def mock(treatment)
+        cistern.data[:treatments] += [treatment]
+      end
     end
-  end
+  }
 
   shared_examples 'mock_data#backend' do |backend, options|
     it 'should store mock data' do
       Sample.mock!
       Sample::Mock.store_in(backend, options)