Merge 5339980 into 5c8a7db

bugsnag · Jan 17, 2019 · eb49c30 · eb49c30
2 parents 5c8a7db + 5339980
commit eb49c30
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/spec/session_tracker_spec.rb b/spec/session_tracker_spec.rb
@@ -2,14 +2,6 @@
 require 'webrick'
 require 'spec_helper'
 require 'json'
-require 'concurrent'
-
-# Enable reset of session_counts between each example
-module Bugsnag
-  class SessionTracker
-    attr_accessor :session_counts
-  end
-end
 
 describe Bugsnag::SessionTracker do
   server = nil
@@ -27,7 +19,7 @@ class SessionTracker
     Thread.new{ server.start }
   end
 
-  after(:each) do
+  before(:each) do
     Bugsnag.instance_variable_set(:@session_tracker, Bugsnag::SessionTracker.new)
   end
 
@@ -40,6 +32,10 @@ class SessionTracker
     queue.clear
   end
 
+  after(:all) do
+    Bugsnag.instance_variable_set(:@session_tracker, Bugsnag::SessionTracker.new)
+  end
+
   it 'adds session object to queue' do
     tracker = Bugsnag::SessionTracker.new
     tracker.start_session