HP-2496/Added methods findBehaviorByClass, findPricesByTypeName and u… (#103)

* HP-2496/Added methods findBehaviorByClass, findPricesByTypeName and unit tests for them

* HP-2496/Removed not used namespaces

* HP-2496/Fixed after review

---------

Co-authored-by: Drahma <[email protected]>

Author: ValeriyShnurovoy

src/product/TariffTypeDefinition.php

@@ -4,6 +4,7 @@
 
 use Google\Service\TPU;
 use hiqdev\php\billing\product\behavior\BehaviorCollectionInterface;
+use hiqdev\php\billing\product\behavior\BehaviorInterface;
 use hiqdev\php\billing\product\behavior\BehaviorTariffTypeCollection;
 use hiqdev\php\billing\product\behavior\TariffTypeBehaviorRegistry;
 use hiqdev\php\billing\product\Domain\Model\TariffTypeInterface;
@@ -12,6 +13,7 @@
 use hiqdev\php\billing\product\price\PriceTypeDefinitionCollection;
 use hiqdev\php\billing\product\price\PriceTypeDefinitionCollectionInterface;
 use hiqdev\php\billing\product\price\PriceTypeDefinitionFactory;
+use hiqdev\php\billing\product\price\PriceTypeDefinitionInterface;
 use hiqdev\php\billing\product\trait\HasLock;
 
 /**
@@ -87,6 +89,24 @@ public function withPrices()
         return $this->prices;
     }
 
+    public function findPricesByTypeName(string $typeName): ?array
+    {
+        $prices = null;
+        $this->ensureNotLocked();
+
+        foreach ($this->prices as $price) {
+            if ($this->matchesPriceType($price, $typeName)) {
+                $prices[] = $price;
+            }
+        }
+        return $prices;
+    }
+
+    private function matchesPriceType(PriceTypeDefinitionInterface $price, string $typeName): bool
+    {
+        return str_ends_with($price->type()->getName(), ",$typeName");
+    }
+
     /**
      * @return BehaviorCollectionInterface<TariffTypeDefinition>
      * @psalm-suppress ImplementedReturnTypeMismatch
@@ -105,6 +125,11 @@ public function hasBehavior(string $behaviorClassName): bool
         return $this->tariffTypeBehaviorRegistry->hasBehavior($behaviorClassName);
     }
 
+    public function findBehaviorByClass(string $class): ?BehaviorInterface
+    {
+        return $this->tariffTypeBehaviorRegistry->findBehaviorByClass($class);
+    }
+
     public function end(): TariffTypeDefinitionInterface
     {
         $this->ensureProductExists();

src/product/TariffTypeDefinitionInterface.php

@@ -37,4 +37,6 @@ public function setPricesSuggester(string $suggesterClass): static;
     public function withPrices();
 
     public function end();
+
+    public function findPricesByTypeName(string $typeName): ?array;
 }

src/product/behavior/HasBehaviorsInterface.php

@@ -18,4 +18,6 @@ interface HasBehaviorsInterface
     public function withBehaviors();
 
     public function hasBehavior(string $behaviorClassName): bool;
+
+    public function findBehaviorByClass(string $class): ?BehaviorInterface;
 }

src/product/behavior/TariffTypeBehaviorRegistry.php

@@ -60,6 +60,17 @@ public function hasBehavior(string $behaviorClassName): bool
         return false;
     }
 
+    public function findBehaviorByClass(string $class): ?BehaviorInterface
+    {
+        foreach ($this->getBehaviors() as $behavior) {
+            if ($behavior instanceof $class) {
+                return $behavior;
+            }
+        }
+
+        return null;
+    }
+
     public function lock(): void
     {
         $this->behaviorCollection->lock();

src/product/price/PriceTypeDefinition.php

@@ -5,6 +5,7 @@
 
 use hiqdev\php\billing\product\AggregateInterface;
 use hiqdev\php\billing\product\behavior\BehaviorCollectionInterface;
+use hiqdev\php\billing\product\behavior\BehaviorInterface;
 use hiqdev\php\billing\product\behavior\HasBehaviorsInterface;
 use hiqdev\php\billing\product\Exception\AggregateNotDefinedException;
 use hiqdev\php\billing\product\behavior\BehaviorPriceTypeDefinitionCollection;
@@ -194,6 +195,17 @@ public function hasBehavior(string $behaviorClassName): bool
         return false;
     }
 
+    public function findBehaviorByClass(string $class): ?BehaviorInterface
+    {
+        foreach ($this->behaviorCollection as $behavior) {
+            if ($behavior instanceof $class) {
+                return $behavior;
+            }
+        }
+
+        return null;
+    }
+
     /**
      * @inerhitDoc
      */