IsTraitTest.php

<?php

declare(strict_types=1);

namespace Arkitect\Tests\Unit\Expressions\ForClasses;

use Arkitect\Analyzer\ClassDescriptionBuilder;
use Arkitect\Expression\ForClasses\IsTrait;
use Arkitect\Rules\Violations;
use PHPUnit\Framework\TestCase;

class IsTraitTest extends TestCase
{
    public function test_it_should_return_violation_error(): void
    {
        $isTrait = new IsTrait();

        $classDescription = (new ClassDescriptionBuilder())
            ->setFilePath('src/Foo.php')
            ->setClassName('HappyIsland')
            ->build();

        $because = 'we want to add this rule for our software';
        $violationError = $isTrait->describe($classDescription, $because)->toString();

        $violations = new Violations();
        $isTrait->evaluate($classDescription, $violations, $because);

        self::assertNotEquals(0, $violations->count());
        self::assertEquals('HappyIsland should be trait because we want to add this rule for our software', $violationError);
    }

    public function test_it_should_return_true_if_is_trait(): void
    {
        $isTrait = new IsTrait();

        $classDescription = (new ClassDescriptionBuilder())
            ->setFilePath('src/Foo.php')
            ->setClassName('HappyIsland')
            ->setTrait(true)
            ->build();

        $because = 'we want to add this rule for our software';
        $violations = new Violations();
        $isTrait->evaluate($classDescription, $violations, $because);

        self::assertEquals(0, $violations->count());
    }
}