/**
* Add entity to the batch deletion
* @param EntityInterface $entity
* @throws \Aws\DynamoDb\Exception\DynamoDBException
*/
public function deleteBatch(EntityInterface $entity)
{
if (is_null($this->deleteBatch)) {
$this->deleteBatch = WriteRequestBatch::factory($this->dynamoDb);
}
$this->deleteBatch->add(new DeleteRequest($this->formatKeyCondition($entity), $this->getEntityTable($entity)));
}
public function testFlushesWhenSizeMeetsThreshold()
{
$t = $this->getMock('Guzzle\\Batch\\BatchTransferInterface', array('transfer'));
$d = $this->getMock('Guzzle\\Batch\\BatchDivisorInterface', array('createBatches'));
$batch = new Batch($t, $d);
$queue = $this->readAttribute($batch, 'queue');
$d->expects($this->exactly(2))->method('createBatches')->will($this->returnCallback(function () use($queue) {
$items = array();
foreach ($queue as $item) {
$items[] = $item;
}
return array($items);
}));
$t->expects($this->exactly(2))->method('transfer');
$flush = new FlushingBatch($batch, 3);
$this->assertEquals(3, $flush->getThreshold());
$flush->setThreshold(2);
$flush->add('foo')->add('baz')->add('bar')->add('bee')->add('boo');
$this->assertEquals(1, count($flush));
}