accounts/common/tests/unit/behaviors/PrimaryKeyValueBehaviorTest.php

<?php
declare(strict_types=1);

namespace common\tests\unit\behaviors;

use common\behaviors\PrimaryKeyValueBehavior;
use common\tests\unit\TestCase;
use yii\db\ActiveRecord;

class PrimaryKeyValueBehaviorTest extends TestCase {

    public function testGenerateValueForThePrimaryKey() {
        $model = $this->createDummyModel();
        $behavior = $this->createPartialMock(PrimaryKeyValueBehavior::class, ['isValueExists']);
        $behavior->method('isValueExists')->willReturn(false);
        $behavior->value = function() {
            return 'mock';
        };

        $model->attachBehavior('primary-key-value-behavior', $behavior);
        $behavior->setPrimaryKeyValue();
        $this->assertSame('mock', $model->id);
    }

    public function testShouldRegenerateValueWhenGeneratedAlreadyExists() {
        $model = $this->createDummyModel();
        $behavior = $this->createPartialMock(PrimaryKeyValueBehavior::class, ['isValueExists', 'generateValue']);
        $behavior->expects($this->exactly(3))->method('generateValue')->willReturnOnConsecutiveCalls('1', '2', '3');
        $behavior->expects($this->exactly(3))->method('isValueExists')->willReturnOnConsecutiveCalls(true, true, false);

        $model->attachBehavior('primary-key-value-behavior', $behavior);
        $behavior->setPrimaryKeyValue();
        $this->assertSame('3', $model->id);
    }

    private function createDummyModel() {
        return new class extends ActiveRecord {
            public $id;

            public static function primaryKey() {
                return ['id'];
            }
        };
    }

}