salt/tests/unit/utils/test_jid.py

# -*- coding: utf-8 -*-
"""
Tests for salt.utils.jid
"""

# Import Python libs
from __future__ import absolute_import, unicode_literals

import datetime
import os

# Import Salt libs
import salt.utils.jid
from tests.support.mock import patch
from tests.support.unit import TestCase


class JidTestCase(TestCase):
    def test_jid_to_time(self):
        test_jid = 20131219110700123489
        expected_jid = "2013, Dec 19 11:07:00.123489"
        self.assertEqual(salt.utils.jid.jid_to_time(test_jid), expected_jid)

        # Test incorrect lengths
        incorrect_jid_length = 2012
        self.assertEqual(salt.utils.jid.jid_to_time(incorrect_jid_length), "")

    def test_is_jid(self):
        self.assertTrue(salt.utils.jid.is_jid("20131219110700123489"))  # Valid JID
        self.assertFalse(salt.utils.jid.is_jid(20131219110700123489))  # int
        self.assertFalse(
            salt.utils.jid.is_jid("2013121911070012348911111")
        )  # Wrong length

    def test_gen_jid(self):
        now = datetime.datetime(2002, 12, 25, 12, 0, 0, 0)
        with patch("salt.utils.jid._utc_now", return_value=now):
            ret = salt.utils.jid.gen_jid({})
            self.assertEqual(ret, "20021225120000000000")
            salt.utils.jid.LAST_JID_DATETIME = None
            ret = salt.utils.jid.gen_jid({"unique_jid": True})
            self.assertEqual(ret, "20021225120000000000_{0}".format(os.getpid()))
            ret = salt.utils.jid.gen_jid({"unique_jid": True})
            self.assertEqual(ret, "20021225120000000001_{0}".format(os.getpid()))

    def test_gen_jid_utc(self):
        utcnow = datetime.datetime(2002, 12, 25, 12, 7, 0, 0)
        with patch("salt.utils.jid._utc_now", return_value=utcnow):
            ret = salt.utils.jid.gen_jid({"utc_jid": True})
            self.assertEqual(ret, "20021225120700000000")

    def test_gen_jid_utc_unique(self):
        utcnow = datetime.datetime(2002, 12, 25, 12, 7, 0, 0)
        with patch("salt.utils.jid._utc_now", return_value=utcnow):
            salt.utils.jid.LAST_JID_DATETIME = None
            ret = salt.utils.jid.gen_jid({"utc_jid": True, "unique_jid": True})
            self.assertEqual(ret, "20021225120700000000_{0}".format(os.getpid()))
            ret = salt.utils.jid.gen_jid({"utc_jid": True, "unique_jid": True})
            self.assertEqual(ret, "20021225120700000001_{0}".format(os.getpid()))