test/test_document.rb in couchpillow-0.3.7 vs test/test_document.rb in couchpillow-0.3.8

@@ -213,9 +213,22 @@
     d.xyz = 123
     d.save!
   end
 
 
+  def test_whitelist
+    d = Class.new(Document) do
+      whitelist :a, :b, :c
+    end.new
+    d.update({a: 1, b: 2, c: 3, d: 4})
+    assert d.save!
+    assert_equal 1, d.a
+    assert_equal 2, d.b
+    assert_equal 3, d.c
+    assert_equal false, d.has?(:d)
+  end
+
+
   def test_to_json
     mock_time
     d = Document.new({}, "1")
     assert_equal "{\"_id\":\"1\",\"created_at\":\"#{mock_time.to_s}\",\"_type\":\"default\"}", d.to_json
   end