def multiple_resources(self):
wp = AutoThreadedWorkProvider()
node = MockWorkNode()
wp._work_nodes = [node]
node.resources = [1, 2, 3, 4]
wp._cur_resources = [2, 3]
self.assertIsNone(wp.get_next_node())
def test_single_resource(self):
wp = AutoThreadedWorkProvider()
node = MockWorkNode()
wp._work_nodes = [node]
node.resources = [1]
self.assertEqual(node, wp.get_next_node())
wp._cur_resources = [1]
self.assertIsNone(wp.get_next_node())
def test_empty_work_pool(self):
wp = AutoThreadedWorkProvider()
self.assertIsNone(wp.get_next_node())
def test_no_resources(self):
wp = AutoThreadedWorkProvider()
node = MockWorkNode()
wp._work_nodes = [node]
self.assertEqual(node, wp.get_next_node())
