Improved test

ankane · Oct 24, 2024 · ad4d445 · ad4d445
1 parent 7540cc9
commit ad4d445
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/test/model_test.rb b/test/model_test.rb
@@ -220,13 +220,17 @@ def test_attributes_not_loaded
     skip if mongoid?
 
     User.create!(email: "test@example.org")
-    user = User.select("id").last
+    user = User.select("id", "phone_ciphertext").last
     assert_nil user.attributes["email"]
     assert !user.has_attribute?("name")
     assert !user.has_attribute?(:name)
 
-    assert_equal ["id"], user.attributes.keys
-    assert_equal ["id"], user.attribute_names
+    assert_equal ["id", "phone_ciphertext", "phone"], user.attributes.keys
+    assert_equal ["id", "phone_ciphertext", "phone"], user.attribute_names
+    assert user.has_attribute?("phone_ciphertext")
+    assert user.has_attribute?(:phone_ciphertext)
+    assert user.has_attribute?("phone")
+    assert user.has_attribute?(:phone)
     assert !user.has_attribute?("email")
     assert !user.has_attribute?(:email)