allows('isManaged')->andReturn(true); if ($divi !== null) { $schema->allows('diviEnabled')->andReturn($divi); } return $schema; } }