server-tools/fetchmail_attach_from_folder/tests/test_match_algorithms.py

# -*- coding: utf-8 -*-
# Copyright - 2015-2018 Therp BV <https://acme.com>.
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
from odoo import models
from odoo.tests.common import TransactionCase

from ..match_algorithm import email_exact, email_domain, odoo_standard

MSG_BODY = [
    ('1 (RFC822 {1149}',
     'Return-Path: <ronald@acme.com>\r\n'
     'Delivered-To: demo@yourcompany.example.com\r\n'
     'Received: from localhost (localhost [127.0.0.1])\r\n'
     '\tby vanaheim.acme.com (Postfix) with ESMTP id 14A3183163\r\n'
     '\tfor <demo@yourcompany.example.com>;'
     ' Mon, 26 Mar 2018 16:03:52 +0200 (CEST)\r\n'
     'To: Test User <demo@yourcompany.example.com>\r\n'
     'From: Ronald Portier <ronald@acme.com>\r\n'
     'Subject: test\r\n'
     'Message-ID: <485a8041-d560-a981-5afc-d31c1f136748@acme.com>\r\n'
     'Date: Mon, 26 Mar 2018 16:03:51 +0200\r\n'
     'User-Agent: Mock Test\r\n'
     'MIME-Version: 1.0\r\n'
     'Content-Type: text/plain; charset=utf-8\r\n'
     'Content-Language: en-US\r\n'
     'Content-Transfer-Encoding: 7bit\r\n\r\n'
     'Hallo Wereld!\r\n'),
    ')']


class MockConnection():

    def select(self, path):
        """Mock selecting a folder."""
        return ('OK', )

    def store(self, msgid, msg_item, value):
        """Mock store command."""
        return 'OK'

    def fetch(self, msgid, parts):
        """Return RFC822 formatted message."""
        return ('OK', MSG_BODY)

    def search(self, charset, criteria):
        """Return some msgid's."""
        return ('OK', ['123 456'])


class TestMatchAlgorithms(TransactionCase):

    def _get_base_folder(self):
        server_model = self.env['fetchmail.server']
        folder_model = self.env['fetchmail.server.folder']
        folder = folder_model.browse([models.NewId()])
        folder.model_id = self.env.ref('base.model_res_partner').id
        folder.model_field = 'email'
        folder.match_algorithm = 'EmailExact'
        folder.mail_field = 'to,from'
        folder.server_id = server_model.browse([models.NewId()])
        return folder

    def do_matching(
            self, match_algorithm, expected_xmlid, folder, mail_message,
            mail_message_org=None):
        matcher = match_algorithm()
        matches = matcher.search_matches(folder, mail_message)
        self.assertEqual(len(matches), 1)
        self.assertEqual(
            matches[0], self.env.ref(expected_xmlid))
        connection = MockConnection()
        matcher.handle_match(
            connection, matches[0], folder, mail_message, mail_message_org,
            None)

    def test_email_exact(self):
        mail_message = {
            'subject': 'Testsubject',
            'to': 'demo@yourcompany.example.com',
            'from': 'someone@else.com',
        }
        folder = self._get_base_folder()
        folder.match_algorithm = 'EmailExact'
        self.do_matching(
            email_exact.EmailExact, 'base.user_demo_res_partner',
            folder, mail_message)
        self.assertEqual(
            self.env.ref('base.user_demo_res_partner').message_ids.subject,
            mail_message['subject'])

    def test_email_domain(self):
        mail_message = {
            'subject': 'Testsubject',
            'to': 'test@seagate.com',
            'from': 'someone@else.com',
            'attachments': [('hello.txt', 'Hello World!')]}
        folder = self._get_base_folder()
        folder.match_algorithm = 'EmailDomain'
        folder.use_first_match = True
        self.do_matching(
            email_domain.EmailDomain, 'base.res_partner_address_31',
            folder, mail_message)
        self.assertEqual(
            self.env.ref('base.res_partner_address_31').message_ids.subject,
            mail_message['subject'])

    def test_odoo_standard(self):
        mail_message_org = (
            "To: demo@yourcompany.example.com\n"
            "From: someone@else.com\n"
            "Subject: testsubject\n"
            "Message-Id: 42\n"
            "Hello world"
        )
        folder = self._get_base_folder()
        folder.match_algorithm = 'OdooStandard'
        matcher = odoo_standard.OdooStandard()
        matches = matcher.search_matches(folder, None)
        self.assertEqual(len(matches), 1)
        matcher.handle_match(
            None, matches[0], folder, None, mail_message_org, None)
        self.assertIn(
            'Hello world',
            self.env['mail.message']
                .search([('subject', '=', 'testsubject')]).body)

    def test_apply_matching_exact(self):
        folder = self._get_base_folder()
        folder.match_algorithm = 'EmailExact'
        connection = MockConnection()
        msgid = "<485a8041-d560-a981-5afc-d31c1f136748@acme.com>"
        matcher = email_exact.EmailExact()
        folder.apply_matching(connection, msgid, matcher)

    def test_retrieve_imap_folder_domain(self):
        folder = self._get_base_folder()
        folder.match_algorithm = 'EmailDomain'
        connection = MockConnection()
        folder.retrieve_imap_folder(connection)

    def test_field_view_get(self):
        """For the moment just check execution withouth errors."""
        server_model = self.env['fetchmail.server']
        view = server_model.fields_view_get()
        self.assertTrue(view)
        self.assertIn(
            'match_algorithm',
            view['fields']['folder_ids']['views']['form']['arch'])