Move exception handler configuration

2 files changed, 8 insertions, 5 deletions

diff --git a/lib/super_good/solidus_taxjar.rb b/lib/super_good/solidus_taxjar.rb
index be56453..663b3d8 100644
--- a/lib/super_good/solidus_taxjar.rb
+++ b/lib/super_good/solidus_taxjar.rb
@@ -13,9 +13,13 @@ module SuperGood
     class << self
       attr_accessor :discount_calculator
       attr_accessor :test_mode
+      attr_accessor :exception_handler
     end
 
     self.discount_calculator = ::SuperGood::SolidusTaxJar::DiscountCalculator
     self.test_mode = false
+    self.exception_handler = ->(e) {
+      Rails.logger.error "An error occurred while fetching TaxJar tax rates - #{e}: #{e.message}"
+    }
   end
 end
diff --git a/lib/super_good/solidus_taxjar/tax_calculator.rb b/lib/super_good/solidus_taxjar/tax_calculator.rb
index b9e6444..a3c0690 100644
--- a/lib/super_good/solidus_taxjar/tax_calculator.rb
+++ b/lib/super_good/solidus_taxjar/tax_calculator.rb
@@ -1,11 +1,6 @@
 module SuperGood
   module SolidusTaxJar
     class TaxCalculator
-      class_attribute :exception_handler
-      self.exception_handler = ->(e) {
-        Rails.logger.error "An error occurred while fetching TaxJar tax rates - #{e}: #{e.message}"
-      }
-
       def self.default_api
         ::SuperGood::SolidusTaxJar::API.new
       end
@@ -129,6 +124,10 @@ module SuperGood
           end
         end
       end
+
+      def exception_handler
+        SuperGood::SolidusTaxJar.exception_handler
+      end
     end
   end
 end