add more unit tests

Change-Id: Id3c831fdfca03b5b70786d993e9c6bc78b3bf0fd
This commit is contained in:
Boris Pavlovic 2014-08-07 03:58:09 +04:00
parent 737f351c38
commit b3e432b9d6

View File

@ -17,12 +17,22 @@ import base64
import hashlib import hashlib
import hmac import hmac
import mock
from osprofiler import _utils as utils from osprofiler import _utils as utils
from tests import test from tests import test
class UtilsTestCase(test.TestCase): class UtilsTestCase(test.TestCase):
def test_split(self):
self.assertEqual([1, 2], utils.split([1, 2]))
self.assertEqual(["A", "B"], utils.split("A, B"))
self.assertEqual(["A", " B"], utils.split("A, B", strip=False))
def test_split_wrong_type(self):
self.assertRaises(TypeError, utils.split, 1)
def test_binary_encode_and_decode(self): def test_binary_encode_and_decode(self):
self.assertEqual("text", self.assertEqual("text",
utils.binary_decode(utils.binary_encode("text"))) utils.binary_decode(utils.binary_encode("text")))
@ -87,6 +97,12 @@ class UtilsTestCase(test.TestCase):
packed_data, hmac_data = utils.signed_pack(data, "secret") packed_data, hmac_data = utils.signed_pack(data, "secret")
self.assertIsNone(utils.signed_unpack(packed_data, hmac_data, None)) self.assertIsNone(utils.signed_unpack(packed_data, hmac_data, None))
self.assertIsNone(utils.signed_unpack(packed_data, None, "secret")) self.assertIsNone(utils.signed_unpack(packed_data, None, "secret"))
self.assertIsNone(utils.signed_unpack(packed_data, " ", "secret"))
@mock.patch("osprofiler._utils.generate_hmac")
def test_singed_unpack_generate_hmac_failed(self, mock_generate_hmac):
mock_generate_hmac.side_effect = Exception
self.assertIsNone(utils.signed_unpack("data", "hmac_data", "hmac_key"))
def test_signed_unpack_invalid_json(self): def test_signed_unpack_invalid_json(self):
hmac = "secret" hmac = "secret"