From c4a60df691a0822258489105b71018ff6b3d73c6 Mon Sep 17 00:00:00 2001
From: Xavier Morel <xmo@openerp.com>
Date: Tue, 31 Jul 2012 18:36:31 +0200
Subject: [PATCH] [TEST] o2m db-id export and import

bzr revid: xmo@openerp.com-20120731163631-n97nxph9gm2f6kdp
---
 openerp/tests/test_export.py |  17 +++++-
 openerp/tests/test_import.py | 113 ++++++++++++++++++++++++++++++++++-
 2 files changed, 127 insertions(+), 3 deletions(-)

diff --git a/openerp/tests/test_export.py b/openerp/tests/test_export.py
index d60e5a09dcb..ed47b580365 100644
--- a/openerp/tests/test_export.py
+++ b/openerp/tests/test_export.py
@@ -376,6 +376,21 @@ class test_o2m(CreatorCase):
                 u'4', u','.join(self.names)
             ]])
 
+    def test_multiple_records_id(self):
+        export = self.export(self.commands, fields=['const', 'value/.id'])
+        O2M_c = self.registry('export.one2many.child')
+        ids = O2M_c.browse(self.cr, openerp.SUPERUSER_ID,
+                           O2M_c.search(self.cr, openerp.SUPERUSER_ID, []))
+        self.assertEqual(
+            export,
+            [
+                ['4', str(ids[0].id)],
+                ['', str(ids[1].id)],
+                ['', str(ids[2].id)],
+                ['', str(ids[3].id)],
+                ['', str(ids[4].id)],
+            ])
+
     def test_multiple_records_with_name_before(self):
         self.assertEqual(
             self.export(self.commands, fields=['const', 'value', 'value/value']),
@@ -383,7 +398,7 @@ class test_o2m(CreatorCase):
                 u'4', u','.join(self.names), u'4'
             ]])
 
-    def test_multiple_records_with_name_before(self):
+    def test_multiple_records_with_name_after(self):
         self.assertEqual(
             self.export(self.commands, fields=['const', 'value/value', 'value']),
             [ # completely ignores name_get request
diff --git a/openerp/tests/test_import.py b/openerp/tests/test_import.py
index c2534caa929..c23d74d080a 100644
--- a/openerp/tests/test_import.py
+++ b/openerp/tests/test_import.py
@@ -23,8 +23,8 @@ def error(row, message, record=None, **kwargs):
         "Line %d : %s" % (row, message),
         '')
 
-def values(seq):
-    return [item['value'] for item in seq]
+def values(seq, field='value'):
+    return [item[field] for item in seq]
 
 def setupModule():
     openerp.tools.config['update'] = {'base': 1}
@@ -718,5 +718,114 @@ class test_m2m(ImporterCase):
 class test_o2m(ImporterCase):
     model_name = 'export.one2many'
 
+    def test_single(self):
+        self.assertEqual(
+            self.import_(['const', 'value/value'], [
+                ['5', '63']
+            ]),
+            ok(1))
+
+        (b,) = self.browse()
+        self.assertEqual(b.const, 5)
+        self.assertEqual(values(b.value), [63])
+
+    def test_multicore(self):
+        self.assertEqual(
+            self.import_(['const', 'value/value'], [
+                ['5', '63'],
+                ['6', '64'],
+            ]),
+            ok(2))
+
+        b1, b2 = self.browse()
+        self.assertEqual(b1.const, 5)
+        self.assertEqual(values(b1.value), [63])
+        self.assertEqual(b2.const, 6)
+        self.assertEqual(values(b2.value), [64])
+
+    def test_multisub(self):
+        self.assertEqual(
+            self.import_(['const', 'value/value'], [
+                ['5', '63'],
+                ['', '64'],
+                ['', '65'],
+                ['', '66'],
+            ]),
+            ok(4))
+
+        (b,) = self.browse()
+        self.assertEqual(values(b.value), [63, 64, 65, 66])
+
+    def test_multi_subfields(self):
+        self.assertEqual(
+            self.import_(['value/str', 'const', 'value/value'], [
+                ['this', '5', '63'],
+                ['is', '', '64'],
+                ['the', '', '65'],
+                ['rhythm', '', '66'],
+            ]),
+            ok(4))
+
+        (b,) = self.browse()
+        self.assertEqual(values(b.value), [63, 64, 65, 66])
+        self.assertEqual(
+            values(b.value, 'str'),
+            'this is the rhythm'.split())
+
+    def test_link(self):
+        id1 = self.registry('export.one2many.child').create(self.cr, openerp.SUPERUSER_ID, {
+            'str': 'Bf', 'value': 109
+        })
+        id2 = self.registry('export.one2many.child').create(self.cr, openerp.SUPERUSER_ID, {
+            'str': 'Me', 'value': 262
+        })
+
+        self.assertEqual(
+            self.import_(['const', 'value/.id'], [
+                ['42', str(id1)],
+                ['', str(id2)],
+            ]),
+            ok(2))
+
+        # No record values alongside id => o2m resolution skipped altogether,
+        # creates 2 records
+        b, b1 = self.browse()
+        self.assertEqual(b.const, 42)
+        self.assertEqual(values(b.value), [])
+        self.assertEqual(b1.const, 4)
+        self.assertEqual(values(b1.value), [])
+
+    def test_link_2(self):
+        O2M_c = self.registry('export.one2many.child')
+        id1 = O2M_c.create(self.cr, openerp.SUPERUSER_ID, {
+            'str': 'Bf', 'value': 109
+        })
+        id2 = O2M_c.create(self.cr, openerp.SUPERUSER_ID, {
+            'str': 'Me', 'value': 262
+        })
+
+        self.assertEqual(
+            self.import_(['const', 'value/.id', 'value/value'], [
+                ['42', str(id1), '1'],
+                ['', str(id2), '2'],
+            ]),
+            ok(2))
+
+        (b,) = self.browse()
+        # if an id (db or xid) is provided, expectations that objects are
+        # *already* linked and emits UPDATE (1, id, {}).
+        # Noid => CREATE (0, ?, {})
+        # TODO: xid ignored aside from getting corresponding db id?
+        self.assertEqual(b.const, 42)
+        self.assertEqual(values(b.value), [])
+
+        # FIXME: updates somebody else's records?
+        self.assertEqual(
+            O2M_c.read(self.cr, openerp.SUPERUSER_ID, id1),
+            {'id': id1, 'str': 'Bf', 'value': 1, 'parent_id': False})
+        self.assertEqual(
+            O2M_c.read(self.cr, openerp.SUPERUSER_ID, id2),
+            {'id': id2, 'str': 'Me', 'value': 2, 'parent_id': False})
+
 # function, related, reference: written to db as-is...
 # => function uses @type for value coercion/conversion