Add test for context proc arity validation

hashicorp · May 17, 2019 · 2a0b57a · 2a0b57a
1 parent dae0e6d
commit 2a0b57a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/spec/unit/encrypted_model_spec.rb b/spec/unit/encrypted_model_spec.rb
@@ -20,6 +20,12 @@
       }.to raise_error(Vault::Rails::ValidationFailedError)
     end
 
+    it "raises an exception if a proc is passed to :context without an arity of 1" do
+      expect {
+        klass.vault_attribute(:foo, context: ->() { })
+      }.to raise_error(Vault::Rails::ValidationFailedError, /1 argument/i)
+    end
+
     it "defines a getter" do
       klass.vault_attribute(:foo)
       expect(klass.instance_methods).to include(:foo)