test/test_service.rb in pinch_hitter-0.2 vs test/test_service.rb in pinch_hitter-0.3

@@ -2,13 +2,15 @@
 
 require 'pinch_hitter/service/replay_ws'
 require 'minitest/autorun'
 require 'rack/test'
 require 'nokogiri'
+require_relative 'message_assertions'
 
 class TestService < MiniTest::Unit::TestCase
   include Rack::Test::Methods
+  include MessageAssertions
 
   def app
     PinchHitter::Service::ReplayWs
   end
 
@@ -89,25 +91,20 @@
     #Out of queue
     post '/endpoint1', ''
     assert_equal '', last_response.body
   end
 
-  def xml_message
-    message = %Q{<?xml version="1.0" encoding="UTF-8"?>
-      <env:Envelope xmlns:env="http://www.w3.org/2003/05/soap-envelope" xmlns:replay="http://www.leandog.com/replay">
-        <env:Body>
-          <replay:Response>BARK!</replay:Response>
-        </env:Body>
-    </env:Envelope>}
-  end
+  def test_module
+    post '/register_module?endpoint=stuff', Marshal.dump(TestModule)
+    post '/stuff', ''
 
-  def yml_message
-   %Q~{"menu": {
-      "id": "file",
-      "value": "File"
-    }}~
+    assert_received xml_message
   end
 
-  def assert_received(message)
-    assert_equal message.gsub(/\n\s*/, ''), last_response.body.strip
+  module TestModule
+    include MessageAssertions
+    def respond_to(message)
+      xml_message
+    end
   end
+
 end