HP-1751 created TariffTypeBehaviorRegistry

Author: VadymHrechukha

src/product/TariffTypeDefinition.php

@@ -3,6 +3,7 @@
 namespace hiqdev\php\billing\product;
 
 use hiqdev\php\billing\product\behavior\BehaviorTariffTypeCollection;
+use hiqdev\php\billing\product\behavior\TariffTypeBehaviorRegistry;
 use hiqdev\php\billing\product\Domain\Model\TariffTypeInterface;
 use hiqdev\php\billing\product\Exception\ProductNotDefinedException;
 use hiqdev\php\billing\product\Exception\TariffTypeLockedException;
@@ -17,12 +18,12 @@ class TariffTypeDefinition implements TariffTypeDefinitionInterface
 
     private PriceTypeDefinitionCollection $prices;
 
-    private BehaviorTariffTypeCollection $behaviorCollection;
+    private TariffTypeBehaviorRegistry $tariffTypeBehaviorRegistry;
 
     public function __construct(private readonly TariffTypeInterface $tariffType)
     {
         $this->prices = new PriceTypeDefinitionCollection($this, new PriceTypeDefinitionFactory());
-        $this->behaviorCollection = new BehaviorTariffTypeCollection($this, $tariffType);
+        $this->tariffTypeBehaviorRegistry = new TariffTypeBehaviorRegistry($this, $tariffType);
     }
 
     public function tariffType(): TariffTypeInterface
@@ -78,18 +79,12 @@ public function withBehaviors(): BehaviorTariffTypeCollection
     {
         $this->ensureNotLocked();
 
-        return $this->behaviorCollection;
+        return $this->tariffTypeBehaviorRegistry->withBehaviors();
     }
 
     public function hasBehavior(string $behaviorClassName): bool
     {
-        foreach ($this->behaviorCollection as $behavior) {
-            if ($behavior instanceof $behaviorClassName) {
-                return true;
-            }
-        }
-
-        return false;
+        return $this->tariffTypeBehaviorRegistry->hasBehavior($behaviorClassName);
     }
 
     public function end(): TariffTypeDefinitionInterface

src/product/behavior/TariffTypeBehaviorRegistry.php

@@ -0,0 +1,32 @@
+<?php declare(strict_types=1);
+
+namespace hiqdev\php\billing\product\behavior;
+
+use hiqdev\php\billing\product\Domain\Model\TariffTypeInterface;
+use hiqdev\php\billing\product\TariffTypeDefinitionInterface;
+
+final class TariffTypeBehaviorRegistry
+{
+    private BehaviorTariffTypeCollection $behaviorCollection;
+
+    public function __construct(TariffTypeDefinitionInterface $tariffTypeDefinition, TariffTypeInterface $tariffType)
+    {
+        $this->behaviorCollection = new BehaviorTariffTypeCollection($tariffTypeDefinition, $tariffType);
+    }
+
+    public function withBehaviors(): BehaviorTariffTypeCollection
+    {
+        return $this->behaviorCollection;
+    }
+
+    public function hasBehavior(string $behaviorClassName): bool
+    {
+        foreach ($this->behaviorCollection as $behavior) {
+            if ($behavior instanceof $behaviorClassName) {
+                return true;
+            }
+        }
+
+        return false;
+    }
+}

tests/unit/product/behavior/TariffTypeBehaviorRegistryTest.php

@@ -0,0 +1,40 @@
+<?php declare(strict_types=1);
+
+namespace hiqdev\php\billing\tests\unit\product\behavior;
+
+use hiqdev\php\billing\product\behavior\TariffTypeBehaviorRegistry;
+use hiqdev\php\billing\product\TariffTypeDefinitionInterface;
+use hiqdev\php\billing\product\Domain\Model\TariffTypeInterface;
+use hiqdev\php\billing\product\behavior\BehaviorTariffTypeCollection;
+use PHPUnit\Framework\TestCase;
+
+class TariffTypeBehaviorRegistryTest extends TestCase
+{
+    private TariffTypeBehaviorRegistry $manager;
+
+    protected function setUp(): void
+    {
+        $tariffTypeDefinition = $this->createMock(TariffTypeDefinitionInterface::class);
+        $tariffType = $this->createMock(TariffTypeInterface::class);
+        $this->manager = new TariffTypeBehaviorRegistry($tariffTypeDefinition, $tariffType);
+    }
+
+    public function testWithBehaviorsReturnsBehaviorCollection(): void
+    {
+        $this->assertInstanceOf(BehaviorTariffTypeCollection::class, $this->manager->withBehaviors());
+    }
+
+    public function testHasBehaviorReturnsFalseWhenBehaviorNotPresent(): void
+    {
+        $this->assertFalse($this->manager->hasBehavior(TestBehavior::class));
+    }
+
+    public function testHasBehaviorReturnsTrueWhenBehaviorPresent(): void
+    {
+        $behavior = $this->createMock(TestBehavior::class);
+        $behaviorCollection = $this->manager->withBehaviors();
+        $behaviorCollection->attach($behavior);
+
+        $this->assertTrue($this->manager->hasBehavior(TestBehavior::class));
+    }
+}