test_liborgdate_utf8.py

# -*- coding: utf-8 -*-

import sys
import unittest
import locale
import threading

from datetime import date
from contextlib import contextmanager

from orgmode.py3compat.unicode_compatibility import *

sys.path.append(u'../ftplugin')
from orgmode.liborgmode.orgdate import OrgDate

class OrgDateUtf8TestCase(unittest.TestCase):
    u"""
    Tests OrgDate with utf-8 enabled locales
    """
    LOCALE_LOCK = threading.Lock()
    UTF8_LOCALE = "fr_FR.utf-8"

    @contextmanager
    def setlocale(self, name):
        with self.LOCALE_LOCK:
            saved = locale.setlocale(locale.LC_ALL)
            try:
                yield locale.setlocale(locale.LC_ALL, name)
            finally:
                locale.setlocale(locale.LC_ALL, saved)

    def setUp(self):
        self.year = 2016
        self.month = 5
        self.day = 7
        self.text = u'<2016-05-07 sam.>'
        self.textinactive = u'[2016-05-07 sam.]'

    def test_OrdDate_str_unicode_active(self):
        with self.setlocale(self.UTF8_LOCALE):
            od = OrgDate(True, self.year, self.month, self.day)
            self.assertEqual(self.text, unicode(od))

    def test_OrdDate_str_unicode_inactive(self):
        with self.setlocale(self.UTF8_LOCALE):
            od = OrgDate(False, self.year, self.month, self.day)
            self.assertEqual(self.textinactive, unicode(od))

def suite():
    return unittest.TestLoader().loadTestsFromTestCase(OrgDateUtf8TestCase)