def setUp(self):
self.grouper = CustomGroupingHelper()
def setUp(self): self.grouper = CustomGroupingHelper()
class CustomGroupingTest(unittest.TestCase):
STREAM_ID = "default"
TASK_IDS = [2, 3, 5, 7, 11, 13, 17]
SRC_COMP_NAME = "component1"
VALUES = ['test', 'values']
def setUp(self):
self.grouper = CustomGroupingHelper()
def tearDown(self):
self.grouper = None
def test_all_target(self):
# custom grouping returns all targets
all_grouping = MockCustomGrouping(MockCustomGrouping.ALL_TARGET_MODE)
self.grouper.add(self.STREAM_ID, self.TASK_IDS, all_grouping,
self.SRC_COMP_NAME)
self.assertIn(self.STREAM_ID, self.grouper.targets)
self.grouper.prepare(None)
ret = self.grouper.choose_tasks(self.STREAM_ID, self.VALUES)
self.assertEqual(ret, self.TASK_IDS)
ret2 = self.grouper.choose_tasks("not_registered_stream", self.VALUES)
self.assertIsNotNone(ret2)
self.assertIsInstance(ret2, list)
self.assertEqual(len(ret2), 0)
def test_random_target(self):
# custom grouping returns a random subset of given targets
random_grouping = MockCustomGrouping(
MockCustomGrouping.RANDOM_TARGET_MODE)
self.grouper.add(self.STREAM_ID, self.TASK_IDS, random_grouping,
self.SRC_COMP_NAME)
self.grouper.prepare(None)
ret = self.grouper.choose_tasks(self.STREAM_ID, self.VALUES)
self.assertEqual(len(set(ret)), len(ret))
self.assertTrue(set(ret) <= set(self.TASK_IDS))
def test_wrong_return_type(self):
wrong_rettype = MockCustomGrouping(
MockCustomGrouping.WRONG_RETURN_TYPE_MODE)
self.grouper.add(self.STREAM_ID, self.TASK_IDS, wrong_rettype,
self.SRC_COMP_NAME)
self.grouper.prepare(None)
with self.assertRaises(TypeError):
self.grouper.choose_tasks(self.STREAM_ID, self.VALUES)
def test_wrong_return_value(self):
wrong_retvalue = MockCustomGrouping(
MockCustomGrouping.WRONG_CHOOSE_TASK_MODE)
self.grouper.add(self.STREAM_ID, self.TASK_IDS, wrong_retvalue,
self.SRC_COMP_NAME)
self.grouper.prepare(None)
with self.assertRaises(ValueError):
self.grouper.choose_tasks(self.STREAM_ID, self.VALUES)
class CustomGroupingTest(unittest.TestCase):
STREAM_ID = "default"
TASK_IDS = [2, 3, 5, 7, 11, 13, 17]
SRC_COMP_NAME = "component1"
VALUES = ['test', 'values']
def setUp(self):
self.grouper = CustomGroupingHelper()
def tearDown(self):
self.grouper = None
def test_all_target(self):
# custom grouping returns all targets
all_grouping = MockCustomGrouping(MockCustomGrouping.ALL_TARGET_MODE)
self.grouper.add(self.STREAM_ID, self.TASK_IDS, all_grouping, self.SRC_COMP_NAME)
self.assertIn(self.STREAM_ID, self.grouper.targets)
self.grouper.prepare(None)
ret = self.grouper.choose_tasks(self.STREAM_ID, self.VALUES)
self.assertEqual(ret, self.TASK_IDS)
ret2 = self.grouper.choose_tasks("not_registered_stream", self.VALUES)
self.assertIsNotNone(ret2)
self.assertIsInstance(ret2, list)
self.assertEqual(len(ret2), 0)
def test_random_target(self):
# custom grouping returns a random subset of given targets
random_grouping = MockCustomGrouping(MockCustomGrouping.RANDOM_TARGET_MODE)
self.grouper.add(self.STREAM_ID, self.TASK_IDS, random_grouping, self.SRC_COMP_NAME)
self.grouper.prepare(None)
ret = self.grouper.choose_tasks(self.STREAM_ID, self.VALUES)
self.assertEqual(len(set(ret)), len(ret))
self.assertTrue(set(ret) <= set(self.TASK_IDS))
def test_wrong_return_type(self):
wrong_rettype = MockCustomGrouping(MockCustomGrouping.WRONG_RETURN_TYPE_MODE)
self.grouper.add(self.STREAM_ID, self.TASK_IDS, wrong_rettype, self.SRC_COMP_NAME)
self.grouper.prepare(None)
with self.assertRaises(TypeError):
self.grouper.choose_tasks(self.STREAM_ID, self.VALUES)
def test_wrong_return_value(self):
wrong_retvalue = MockCustomGrouping(MockCustomGrouping.WRONG_CHOOSE_TASK_MODE)
self.grouper.add(self.STREAM_ID, self.TASK_IDS, wrong_retvalue, self.SRC_COMP_NAME)
self.grouper.prepare(None)
with self.assertRaises(ValueError):
self.grouper.choose_tasks(self.STREAM_ID, self.VALUES)
