Skip to content

Commit c57973f

Browse files
author
sonhd91
committed
[15.0] connector-interfaces: fix test cases mock import
1 parent e452bd1 commit c57973f

File tree

10 files changed

+27
-20
lines changed

10 files changed

+27
-20
lines changed

connector_importer/models/import_type.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,10 @@ def _check_options(self):
107107
)
108108

109109
def _load_options(self):
110-
return yaml.safe_load(self.options or "") or []
110+
options = []
111+
if self.options:
112+
options = yaml.safe_load(self.options)
113+
return options if isinstance(options, list) else [options]
111114

112115
def available_importers(self):
113116
self.ensure_one()

connector_importer/tests/common.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,11 @@ def _setup_records(cls):
8080
"key": "fake",
8181
"options": """
8282
- model: res.partner
83-
importer: fake.partner.importer
83+
importer:
84+
name: fake.partner.importer
85+
options:
86+
record_handler:
87+
match_domain: "[('name', '=', values['name'])]"
8488
""",
8589
}
8690
)

connector_importer/tests/test_event_listeners.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,8 @@
22
# Copyright 2023 Camptocamp SA
33
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
44

5-
import mock
5+
from unittest import mock
6+
67
from odoo_test_helper import FakeModelLoader
78

89
from odoo.tools import mute_logger
@@ -56,7 +57,11 @@ def setUpClass(cls):
5657
{
5758
"options": f"""
5859
- model: res.partner
59-
importer: fake.partner.importer
60+
importer:
61+
name: fake.partner.importer
62+
options:
63+
record_handler:
64+
match_domain: "[('name', '=', values['name'])]"
6065
- model: {FakeImportedModel._name}
6166
options:
6267
record_handler:

connector_importer/tests/test_import_type.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,8 @@ def test_available_importers_legacy(self):
4242
importers,
4343
(
4444
{
45-
"importer": "partner.importer",
4645
"model": "res.partner",
46+
"importer": "partner.importer",
4747
"is_last_importer": False,
4848
"context": {},
4949
"options": {
@@ -54,8 +54,8 @@ def test_available_importers_legacy(self):
5454
},
5555
},
5656
{
57-
"importer": "user.importer",
5857
"model": "res.users",
58+
"importer": "user.importer",
5959
"is_last_importer": False,
6060
"context": {},
6161
"options": {
@@ -66,8 +66,8 @@ def test_available_importers_legacy(self):
6666
},
6767
},
6868
{
69-
"importer": "import.withspaces",
7069
"model": "another.one",
70+
"importer": "import.withspaces",
7171
"is_last_importer": True,
7272
"context": {},
7373
"options": {

connector_importer/tests/test_recordset_importer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
# Copyright 2018 Camptocamp SA
33
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
44

5-
import mock
5+
from unittest import mock
66

77
from odoo.tools import mute_logger
88

connector_importer/tests/test_source.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@
33
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
44

55

6-
import mock
6+
from unittest import mock
7+
78
from odoo_test_helper import FakeModelLoader
89

910
from .common import BaseTestCase

connector_importer/utils/misc.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ def to_b64(file_content):
6666
"""Safe convertion to b64"""
6767
try:
6868
# py > 3.9
69-
return base64.encodestring(file_content)
69+
return base64.encodebytes(file_content)
7070
except AttributeError:
7171
# py <= 3.9
7272
return base64.b64encode(file_content)

connector_importer_source_sftp/tests/common.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,7 @@
22
# @author: Simone Orsi <[email protected]>
33
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
44

5-
from odoo.addons.component.tests.common import (
6-
TransactionComponentCase,
7-
TransactionComponentRegistryCase,
8-
)
5+
from odoo.addons.component.tests.common import TransactionComponentCase
96
from odoo.addons.connector_importer.tests.common import TestImporterMixin
107

118

@@ -28,7 +25,7 @@ def setUpClass(cls):
2825

2926

3027
class SFTPSourceTransactionComponentRegistryCase(
31-
TransactionComponentRegistryCase, TestImporterMixin, TestSourceCSVSFTPMixin
28+
TestImporterMixin, TestSourceCSVSFTPMixin
3229
):
3330
@classmethod
3431
def setUpClass(cls):

connector_importer_source_sftp/tests/test_csv_sftp_source.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,17 +5,15 @@
55
import base64
66
import io
77

8-
import mock
9-
108
from odoo.tools import mute_logger
9+
from unittest import mock
1110

1211
from odoo.addons.storage_backend_sftp.tests.test_sftp import PARAMIKO_PATH
1312

1413
from .common import SFTPSourceTransactionComponentCase
1514

1615

1716
class TestSourceCSV(SFTPSourceTransactionComponentCase):
18-
1917
extra_fields = [
2018
"chunk_size",
2119
"csv_filesize",

connector_importer_source_sftp/tests/test_event_listener.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,7 @@
44

55
import base64
66
import io
7-
8-
import mock
7+
from unittest import mock
98

109
from odoo.tools import mute_logger
1110

0 commit comments

Comments
 (0)