public function testCriterionIsStoredForDiscarding() { $Criterion = $this->getMock(self::CoreObjectNamespace . 'ICriterion'); $Criterion->expects($this->any())->method('GetEntityType')->will($this->returnValue(self::EntityType)); $this->UnitOfWork->DiscardWhere($Criterion); $this->assertContains($Criterion, $this->UnitOfWork->GetDiscardedCriteria()); }