Allow overriding annotation service

ankane · Jun 13, 2022 · dd40653 · dd40653
1 parent 5b4126a
commit dd40653
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/app/controllers/blazer/queries_controller.rb b/app/controllers/blazer/queries_controller.rb
@@ -232,7 +232,7 @@ def continue_run
       def render_run
         @checks = @query ? @query.checks.order(:id) : []
 
-        @annotations = Blazer::Annotations.new(@data_source.annotations).call(@result)
+        @annotations = Blazer.annotations.new(@data_source.annotations).call(@result)
 
         @first_row = @rows.first || []
         @column_types = []

diff --git a/lib/blazer.rb b/lib/blazer.rb
@@ -83,6 +83,7 @@ class << self
   self.async = false
   self.images = false
   self.override_csp = false
+  self.annotations = Blazer::Annotations
 
   VARIABLE_MESSAGE = "Variable cannot be used in this position"
   TIMEOUT_MESSAGE = "Query timed out :("