diff --git a/test/searchjoy_test.rb b/test/searchjoy_test.rb
index <HASH>..<HASH> 100644
--- a/test/searchjoy_test.rb
+++ b/test/searchjoy_test.rb
@@ -39,6 +39,18 @@ class SearchjoyTest < Minitest::Test
     assert_equal "All Indices", search.search_type
   end
 
+  def test_additional_attributes
+    Product.search("APPLE", track: {source: "web"})
+    search = Searchjoy::Search.last
+    assert_equal "web", search.source
+  end
+
+  def test_override_attributes
+    Product.search("APPLE", track: {search_type: "Item"})
+    search = Searchjoy::Search.last
+    assert_equal "Item", search.search_type
+  end
+
   def test_no_track
     Product.search("apple")
     assert_equal 0, Searchjoy::Search.count
diff --git a/test/test_helper.rb b/test/test_helper.rb
index <HASH>..<HASH> 100644
--- a/test/test_helper.rb
+++ b/test/test_helper.rb
@@ -28,6 +28,7 @@ ActiveRecord::Migration.create_table :searchjoy_searches do |t|
   t.timestamp :created_at
   t.references :convertable, polymorphic: true, index: {name: "index_searchjoy_searches_on_convertable"}
   t.timestamp :converted_at
+  t.string :source
 end
 
 class Product < ActiveRecord::Base
Added tests for additional attributes and overriding attributes