Drop temporary tracking table when finished

1 files changed, 14 insertions, 0 deletions

diff --git a/spec/support/track_untracked_uploads_helpers.rb b/spec/support/track_untracked_uploads_helpers.rb
index 5b832929602..bb700bc53f1 100644
--- a/spec/support/track_untracked_uploads_helpers.rb
+++ b/spec/support/track_untracked_uploads_helpers.rb
@@ -3,4 +3,18 @@ module TrackUntrackedUploadsHelpers
     fixture_path = Rails.root.join('spec', 'fixtures', 'rails_sample.jpg')
     fixture_file_upload(fixture_path)
   end
+
+  def recreate_temp_table_if_dropped
+    TrackUntrackedUploads.new.ensure_temporary_tracking_table_exists
+  end
+
+  RSpec.configure do |config|
+    config.after(:each, :temp_table_may_drop) do
+      recreate_temp_table_if_dropped
+    end
+
+    config.after(:context, :temp_table_may_drop) do
+      recreate_temp_table_if_dropped
+    end
+  end
 end