Tests\Michaelc\Voting\StvElectionTest

Complexity

Total Complexity

9

Size/Duplication

Total Lines	100
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	5

Importance

Changes

0

<?php
class StvElectionTest extends \PHPUnit_Framework_TestCase
{
    public function testNewElection()
    {
        $winners = 2;
        $candidateCount = 6;

        $candidates = $ballots = [];

        for ($i = 1; $i <= $candidateCount; ++$i) {
            $candidates[$i] = new Candidate($i);
        }

        $ballots[] = new Ballot([4, 5, 6]);
        $ballots[] = new Ballot([1, 2, 3]);

        $election = new Election($winners, $candidates, $ballots);

        $this->assertEquals($candidates[3], $election->getCandidate(3));
        $this->assertEquals($candidateCount, $election->getCandidateCount());
        $this->assertEquals($candidateCount, $election->getActiveCandidateCount());
        $this->assertEquals($winners, $election->getWinnersCount());
        $this->assertEquals(2, $election->getNumBallots());

        $candidateIds = $election->getCandidateIds();
        $this->assertCount($candidateCount, $candidateIds);

        for ($i = 1; $i <= $candidateCount; ++$i) {
            $this->assertContains($i, $candidateIds);
        }
    }

    public function testElectionRunner()
    {
        $election = $this->getSampleElection();
        $logger = $this->createMock(Logger::class);

        $runner = new VoteHandler($election, $logger);
    }

    public function testCandidatesStateSetting()
    {
        $election = $this->getSampleElection();

        $election->getCandidates()[5]->setState(Candidate::ELECTED);
        $election->getCandidates()[6]->setState(Candidate::ELECTED);

        $election->getCandidates()[10]->setState(Candidate::DEFEATED);
        $election->getCandidates()[11]->setState(Candidate::DEFEATED);

        $this->assertEquals(Candidate::ELECTED, $election->getCandidates()[5]->getState());
        $this->assertEquals(Candidate::ELECTED, $election->getCandidates()[6]->getState());

        $this->assertEquals(Candidate::DEFEATED, $election->getCandidates()[10]->getState());
        $this->assertEquals(Candidate::DEFEATED, $election->getCandidates()[11]->getState());

        $this->assertEquals(Candidate::RUNNING, $election->getCandidates()[1]->getState());
        $this->assertEquals(Candidate::RUNNING, $election->getCandidates()[2]->getState());
    }

    public function testCandidatesFetchingByState()
    {
        $election = $this->getSampleElection();

        $election->getCandidates()[5]->setState(Candidate::ELECTED);
        $election->getCandidates()[6]->setState(Candidate::ELECTED);

        $election->getCandidates()[10]->setState(Candidate::DEFEATED);

        $active = $election->getActiveCandidates();
        $defeated = $election->getDefeatedCandidates();
        $elected = $election->getElectedCandidates();

        $this->assertContains($election->getCandidates()[5], $elected);
        $this->assertContains($election->getCandidates()[6], $elected);
        $this->assertContains($election->getCandidates()[10], $defeated);
        $this->assertContains($election->getCandidates()[1], $active);

        $this->assertCount(2, $elected);
        $this->assertCount(1, $defeated);
        $this->assertCount(($election->getCandidateCount() - 3), $active);
    }

    protected function getSampleElection()
    {
        $candidates = $ballots = [];

        for ($i = 1; $i <= 20; ++$i) {
            $candidates[$i] = new Candidate($i);
        }

        for ($i = 0; $i <= 40; ++$i) {
            $ballots[] = new Ballot([random_int(1, 20), random_int(1, 20), random_int(1, 20)]);
        }

        $election = new Election(12, $candidates, $ballots);

        return $election;
    }
}