server-tools/datetime_formatter/tests/test_formatter.py

# -*- coding: utf-8 -*-
# © 2015 Grupo ESOC Ingeniería de Servicios, S.L.U. - Jairo Llopis
# © 2016 Tecnativa, S.L. - Vicent Cubells
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
import datetime
from random import random
from openerp.tests.common import TransactionCase
from openerp.tools import (DEFAULT_SERVER_DATE_FORMAT,
                           DEFAULT_SERVER_TIME_FORMAT,
                           DEFAULT_SERVER_DATETIME_FORMAT)
from ..models import MODE_DATE, MODE_TIME, MODE_DATETIME


class FormatterCase(TransactionCase):
    def setUp(self):
        super(FormatterCase, self).setUp()
        self.rl = self.env["res.lang"]
        self.bm = self.rl.best_match()
        self.dt = datetime.datetime.now()
        self.d_fmt = self.bm.date_format or DEFAULT_SERVER_DATE_FORMAT
        self.t_fmt = self.bm.time_format or DEFAULT_SERVER_TIME_FORMAT
        self.kwargs = dict()

    def tearDown(self):
        # This should be returned
        self.expected = self.dt.strftime(self.format)

        # Pass a datetime object
        self.assertEqual(
            self.expected,
            self.rl.datetime_formatter(
                self.dt,
                **self.kwargs))

        # When the date comes as a string
        if isinstance(self.dt, datetime.datetime):
            self.dt_str = self.dt.strftime(DEFAULT_SERVER_DATETIME_FORMAT)
        elif isinstance(self.dt, datetime.date):
            self.dt_str = self.dt.strftime(DEFAULT_SERVER_DATE_FORMAT)
        elif isinstance(self.dt, datetime.time):
            self.dt_str = self.dt.strftime(DEFAULT_SERVER_TIME_FORMAT)

        # Pass a string
        self.assertEqual(
            self.expected,
            self.rl.datetime_formatter(
                self.dt_str,
                **self.kwargs))

        # Pass a unicode
        self.assertEqual(
            self.expected,
            self.rl.datetime_formatter(
                unicode(self.dt_str),
                **self.kwargs))

        super(FormatterCase, self).tearDown()

    def test_datetime(self):
        """Format a datetime."""
        self.format = "%s %s" % (self.d_fmt, self.t_fmt)
        self.kwargs = {"template": MODE_DATETIME}

    def test_date(self):
        """Format a date."""
        self.format = self.d_fmt
        self.kwargs = {"template": MODE_DATE}
        self.dt = self.dt.date()

    def test_time(self):
        """Format times, including float ones."""
        self.format = self.t_fmt
        self.kwargs = {"template": MODE_TIME}
        self.dt = self.dt.time()

        # Test float times
        for n in range(50):
            n = n + random()

            # Patch values with >= 24 hours
            fmt = self.format.replace("%H", "%02d" % n)

            time = (datetime.datetime.min +
                    datetime.timedelta(hours=n)).time()
            self.assertEqual(
                time.strftime(fmt),
                self.rl.datetime_formatter(n, **self.kwargs))

    def test_custom_separator(self):
        """Format a datetime with a custom separator."""
        sep = "T"
        self.format = "%s%s%s" % (self.d_fmt, sep, self.t_fmt)
        self.kwargs = {"template": MODE_DATETIME, "separator": sep}