spec/mock_data_spec.rb in cistern-0.11.1 vs spec/mock_data_spec.rb in cistern-0.11.2.pre2

@@ -1,49 +1,43 @@
 require 'spec_helper'
 
 describe 'mock data' do
-  class Patient < Cistern::Service
-    request :diagnosis
-    request :treat
+  class Diagnosis < Sample::Request
+    def real(diagnosis)
+    end
 
-    class Real
-      def diagnosis(options={})
-      end
+    def mock(diagnosis)
+      service.data.store(:diagnosis, service.data.fetch(:diagnosis) + [diagnosis])
+    end
+  end
 
-      def treat(options={})
-      end
+  class Treat < Sample::Request
+    def real(treatment)
     end
 
-    class Mock
-      def diagnosis(diagnosis)
-        #self.data[:diagnosis] << rand(2) == 0 ? "sick" : "healthy"
-        self.data.store(:diagnosis, self.data.fetch(:diagnosis) + [diagnosis])
-      end
-
-      def treat(treatment)
-        self.data[:treatments] += [treatment]
-      end
+    def mock(treatment)
+      service.data[:treatments] += [treatment]
     end
   end
 
   shared_examples "mock_data#backend" do |backend, options|
     it "should store mock data" do
-      Patient.mock!
-      Patient::Mock.store_in(backend, options)
-      Patient.reset!
+      Sample.mock!
+      Sample::Mock.store_in(backend, options)
+      Sample.reset!
 
-      p = Patient.new
+      p = Sample.new
       p.diagnosis("sick")
       expect(p.data[:diagnosis]).to eq(["sick"])
 
       p.reset!
 
       expect(p.data[:diagnosis]).to eq([])
 
       p.treat("healthy")
       expect(p.data[:treatments]).to eq(["healthy"])
 
-      Patient.reset!
+      Sample.reset!
 
       expect(p.data[:treatments]).to eq([])
     end
 
   end