/** * @param $objects * @param $stage */ public function deleteFromStage($objects, $stage) { $stages = array_slice(func_get_args(), 1); foreach ($stages as $stage) { $this->batchWriter->deleteFromStage($objects, $stage); } }
/** * */ public function testWriteToStages_ManyPages_WritesObjectsToStage() { $sizes = array(10, 30, 100, 300); foreach ($sizes as $size) { $writer = new \BatchedWriter($size); $pages = array(); for ($i = 0; $i < 100; $i++) { $page = new DogPage(); $page->Title = 'Wonder Pup ' . $i; $pages[] = $page; } $writer->writeToStage($pages, 'Stage'); $writer->finish(); $currentStage = \Versioned::current_stage(); \Versioned::reading_stage('Stage'); $this->assertEquals(100, DogPage::get()->Count()); \Versioned::reading_stage('Live'); $this->assertEquals(0, DogPage::get()->Count()); $writer->writeToStage($pages, 'Live'); $writer->finish(); \Versioned::reading_stage('Live'); $this->assertEquals(100, DogPage::get()->Count()); \Versioned::reading_stage($currentStage); $writer->deleteFromStage($pages, 'Stage', 'Live'); $writer->finish(); $this->assertEquals(0, DogPage::get()->Count()); } }