def test_rejects_non_integer_capacity(self):
try:
capacity = "not an integer"
TailDrop(self.queue, capacity)
self.fail("AssertionError expected!")
except AssertionError as error:
self.assertEqual(error.args[0],
TailDrop.INVALID_CAPACITY.format(object=capacity))
def test_rejects_negative_capacity(self):
try:
capacity = -5
TailDrop(self.queue, capacity)
self.fail("AssertionError expected!")
except AssertionError as error:
self.assertEqual(
error.args[0],
TailDrop.NEGATIVE_CAPACITY.format(capacity=capacity))
class TailDropTests(TestCase):
def setUp(self):
self.capacity = 50
self.queue = MagicMock(TaskPool)
self.pool_size(50)
self.throttling = TailDrop(self.queue, self.capacity)
def pool_size(self, length):
type(self.queue).size = PropertyMock(return_value=length)
def test_rejects_non_integer_capacity(self):
try:
capacity = "not an integer"
TailDrop(self.queue, capacity)
self.fail("AssertionError expected!")
except AssertionError as error:
self.assertEqual(error.args[0],
TailDrop.INVALID_CAPACITY.format(object=capacity))
def test_rejects_negative_capacity(self):
try:
capacity = -5
TailDrop(self.queue, capacity)
self.fail("AssertionError expected!")
except AssertionError as error:
self.assertEqual(
error.args[0],
TailDrop.NEGATIVE_CAPACITY.format(capacity=capacity))
def test_reject_at_capacity(self):
self.pool_size(self.capacity)
self.assertFalse(self.throttling._accepts(DUMMY_TASK))
def test_reject_beyond_capacity(self):
self.pool_size(self.capacity + 1)
self.assertFalse(self.throttling._accepts(DUMMY_TASK))
def test_accept_before_capacity(self):
self.pool_size(self.capacity - 1)
self.assertTrue(self.throttling._accepts(DUMMY_TASK))
class TailDropTests(TestCase):
def setUp(self):
self.capacity = 50
self.queue = MagicMock(TaskPool)
self.pool_size(50)
self.throttling = TailDrop(self.queue, self.capacity)
def pool_size(self, length):
type(self.queue).size = PropertyMock(return_value=length)
def test_rejects_non_integer_capacity(self):
try:
capacity = "not an integer"
TailDrop(self.queue, capacity)
self.fail("AssertionError expected!")
except AssertionError as error:
self.assertEqual(error.args[0], TailDrop.INVALID_CAPACITY.format(object=capacity))
def test_rejects_negative_capacity(self):
try:
capacity = -5
TailDrop(self.queue, capacity)
self.fail("AssertionError expected!")
except AssertionError as error:
self.assertEqual(error.args[0], TailDrop.NEGATIVE_CAPACITY.format(capacity=capacity))
def test_reject_at_capacity(self):
self.pool_size(self.capacity)
self.assertFalse(self.throttling._accepts(DUMMY_TASK))
def test_reject_beyond_capacity(self):
self.pool_size(self.capacity + 1)
self.assertFalse(self.throttling._accepts(DUMMY_TASK))
def test_accept_before_capacity(self):
self.pool_size(self.capacity - 1)
self.assertTrue(self.throttling._accepts(DUMMY_TASK))
def create_tail_drop(self, environment, capacity, task_pool):
return ThrottlingWrapper(environment, TailDrop(task_pool, capacity))
def setUp(self):
self.capacity = 50
self.queue = MagicMock(TaskPool)
self.pool_size(50)
self.throttling = TailDrop(self.queue, self.capacity)
def setUp(self):
self.capacity = 50
self.queue = MagicMock(TaskPool)
self.pool_size(50)
self.throttling = TailDrop(self.queue, self.capacity)
