Differential D352 Diff 1180 swh/lister/tests/test_utils.py

Changeset View

Standalone View

				# Copyright (C) 2018 the Software Heritage developers
				# License: GNU General Public License version 3, or any later version
				# See top-level LICENSE file for more information

				import unittest

				from nose.tools import istest

				from swh.lister import utils


				class UtilsTest(unittest.TestCase):

				@istest
				def get(self):
				data = {
				'X-Next-Page': None,
				'x-next-page': 1,
				}
				actual_value = utils.get(data, ['X-Next-Page', 'x-next-page'])

				self.assertEqual(actual_value, 1)

				data = {
				'X-Next-Page': 10,
				'x-next-page': 1,
				}
				actual_value = utils.get(data, ['X-Next-Page', 'x-next-page'])

				self.assertEqual(actual_value, 10)

				data = {
				'x-next-page': 100,
				}
				actual_value = utils.get(data, ['X-Next-Page', 'x-next-page'])

				self.assertEqual(actual_value, 100)

				@istest
				def get_empty(self):
				self.assertIsNone(utils.get({}, []))
				self.assertIsNone(utils.get({'a': 1}, ['b']))
				self.assertIsNone(utils.get({'b': 2}, []))
				self.assertIsNone(utils.get({'b': 2}, []))

				@istest
				def get_errors(self):
				with self.assertRaises(TypeError):
				self.assertIsNone(utils.get({}, None))
				with self.assertRaises(AttributeError):
				self.assertIsNone(utils.get(None, ['a']))

				@istest
				def split_range(self):
				actual_ranges = list(utils.split_range(14, 5))
				self.assertEqual(actual_ranges, [(0, 5), (5, 10), (10, 14)])

				actual_ranges = list(utils.split_range(19, 10))
				self.assertEqual(actual_ranges, [(0, 10), (10, 19)])

				@istest
				def split_range_errors(self):
				with self.assertRaises(TypeError):
				list(utils.split_range(None, 1))

				with self.assertRaises(TypeError):
				list(utils.split_range(100, None))