test/helper.rb in toadhopper-2.0 vs test/helper.rb in toadhopper-2.1

@@ -1,14 +1,51 @@
 require 'bundler/setup'
 require 'test/unit'
 require 'fakeweb'
 require 'toadhopper'
+require 'nokogiri'
 
-FakeWeb.allow_net_connect = true
+def reset_test_env
+  FakeWeb.clean_registry
+  FakeWeb.allow_net_connect = false
+end
 
 def toadhopper
-  @toadhopper ||= Toadhopper.new(ENV['AIRBRAKE_API_KEY'] || ENV['HOPTOAD_API_KEY'] || "test api key")
+  api_key = toadhopper_api_key || 'test api key'
+  Toadhopper.new api_key, toadhopper_args
 end
 
+def toadhopper_api_key
+  ENV['AIRBRAKE_API_KEY'] || ENV['HOPTOAD_API_KEY']
+end
+
+def toadhopper_args
+  ENV['AIRBRAKE_FULL_TEST'] ? {:notify_host => "https://#{Toadhopper::DEFAULT_DOMAIN}"} : {}
+end
+
+def assert_valid_airbrake_xml(body)
+  # prepare schema for validation
+  xsd_path = File.expand_path File.join('resources', 'airbrake_2_2.xsd'),
+    File.dirname(__FILE__)
+  xsd = Nokogiri::XML::Schema File.read xsd_path
+  # validate xml document
+  #doc = Nokogiri::XML body
+  doc = Nokogiri::XML body
+  assert xsd.valid?(doc), "The document is not valid:\n#{body}"
+end
+
+def posting_response_good
+"<notice>
+<id>d87i3bc8-1854-f937-184b-e2d93711cad3</id>
+<url>#{Toadhopper::DEFAULT_NOTIFY_HOST}/locate/d87i3bc8-1854-f937-184b-e2d93711cad3</url>
+</notice>"
+end
+
+def posting_response_bad_apikey
+  'We could not find a project with API key of foo.  Please double check your airbrake configuration.'
+end
+
 def error
   begin; raise "Kaboom!"; rescue => e; e end
 end
+
+reset_test_env
\ No newline at end of file