# HG changeset patch
# User palgarvio
# Date 1181593369 0
# Node ID 5a456b7ebc0ac2820423f8b0054b2e73c5884bbd
# Parent  e1231e97ce4d52b39aafbfde5e6523c8cd67310f
Unittest for the problem reported by pjenvey fixed on [88].

diff --git a/babel/messages/tests/catalog.py b/babel/messages/tests/catalog.py
--- a/babel/messages/tests/catalog.py
+++ b/babel/messages/tests/catalog.py
@@ -42,6 +42,20 @@
         cat.add('foo')
         cat.add(('foo', 'foos'))
         self.assertEqual(1, len(cat))
+        
+    def test_update_message_updates_comments(self):
+        cat = catalog.Catalog()
+        cat[u'foo'] = catalog.Message('foo', locations=[('main.py', 5)])
+        self.assertEqual(cat[u'foo'].comments, [])
+        # Update cat[u'foo'] with a new location and a comment
+        cat[u'foo'] = catalog.Message('foo', locations=[('main.py', 7)],
+                                      comments=['Foo Bar comment 1'])
+        self.assertEqual(cat[u'foo'].comments, ['Foo Bar comment 1'])
+        # now add yet another location with another comment
+        cat[u'foo'] = catalog.Message('foo', locations=[('main.py', 9)],
+                                      comments=['Foo Bar comment 2'])        
+        self.assertEqual(cat[u'foo'].comments,
+                         ['Foo Bar comment 1', 'Foo Bar comment 2'])
 
 
 def suite():