X-Git-Url: https://git.openstreetmap.org./rails.git/blobdiff_plain/3b3f328ae4427ba0d1e3700ceabf7d0870f78dbd..00bd64c7fb86d598eb9406e39f99fc711f58bf9b:/test/models/diary_entry_test.rb?ds=inline

diff --git a/test/models/diary_entry_test.rb b/test/models/diary_entry_test.rb
index ba70c5228..76e8bbbe0 100644
--- a/test/models/diary_entry_test.rb
+++ b/test/models/diary_entry_test.rb
@@ -1,6 +1,11 @@
 require "test_helper"
 
 class DiaryEntryTest < ActiveSupport::TestCase
+  def setup
+    # Create the default language for diary entries
+    create(:language, :code => "en")
+  end
+
   def test_diary_entry_validations
     diary_entry_valid({})
     diary_entry_valid({ :title => "" }, false)
@@ -28,6 +33,7 @@ class DiaryEntryTest < ActiveSupport::TestCase
 
   def test_diary_entry_comments
     diary = create(:diary_entry)
+    assert_equal(0, diary.comments.count)
     create(:diary_comment, :diary_entry => diary)
     assert_equal(1, diary.comments.count)
   end
@@ -43,8 +49,7 @@ class DiaryEntryTest < ActiveSupport::TestCase
   private
 
   def diary_entry_valid(attrs, result = true)
-    entry = DiaryEntry.new(attributes_for(:diary_entry))
-    entry.assign_attributes(attrs)
+    entry = build(:diary_entry, attrs)
     assert_equal result, entry.valid?, "Expected #{attrs.inspect} to be #{result}"
   end
 end