public function test_modify() { $state = new SsoState(); $state->addSsoSession($session1 = new SsoSessionState()); $session1->setIdpEntityId('idp-1'); $state->addSsoSession($session2 = new SsoSessionState()); $session2->setIdpEntityId('idp-2'); $state->modify(function (SsoSessionState $session) use($session1) { return $session->getIdpEntityId() != $session1->getIdpEntityId(); }); $sessions = $state->getSsoSessions(); $this->assertCount(1, $sessions); $this->assertEquals($session2->getIdpEntityId(), $sessions[0]->getIdpEntityId()); }