test: Add column_names spec

active-hash · Jul 8, 2024 · 034e9d2 · 034e9d2
1 parent 0306858
commit 034e9d2
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/spec/active_hash/base_spec.rb b/spec/active_hash/base_spec.rb
@@ -116,6 +116,16 @@ class Country < ActiveHash::Base
     end
   end
 
+  describe ".column_names" do
+    before do
+      Country.fields :name, :iso_name, "size"
+    end
+
+    it "returns an array of column names" do
+      expect(Country.column_names).to eq(["name", "iso_name", "size"])
+    end
+  end
+
   describe ".data=" do
     before do
       class Region < ActiveHash::Base