Flower bucket

src/main/java/flower/store/Chamomile.java

@@ -0,0 +1,11 @@
+package flower.store;
+import lombok.Getter;
+import lombok.Setter;
+
+@Getter
+@Setter
+public class Chamomile extends Flower {
+    public Chamomile() {
+        this.setFlowerType(FlowerType.CHAMOMILE);
+    }
+}

src/main/java/flower/store/Flower.java

@@ -18,4 +18,14 @@ public class Flower {
     public String getColor() {
         return color.toString();
     }
-}
+
+    public void setPrice(double price) {
+        this.price = price;
+    }
+
+    public double getPrice() {
+        return price;
+    }
+
+
+}

src/main/java/flower/store/FlowerBucket.java

@@ -0,0 +1,24 @@
+package flower.store;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class FlowerBucket {
+    private List<FlowerPack> flowerPacks = new ArrayList<>();
+
+    public void add(FlowerPack flowerPack) {
+        flowerPacks.add(flowerPack);
+    }
+
+    public double getPrice() {
+        double totalPrice = 0.0;
+        for (FlowerPack pack : flowerPacks) {
+            totalPrice += pack.getPrice();
+        }
+        return totalPrice;
+    }
+
+    public List<FlowerPack> getFlowerPacks() {
+        return flowerPacks;
+    }
+}

src/main/java/flower/store/FlowerPack.java

@@ -1,6 +1,18 @@
 package flower.store;
-
 public class FlowerPack {
     private Flower flower;
     private int quantity;
-}
+
+    public FlowerPack(Flower flower, int quantity) {
+        this.flower = flower;
+        this.quantity = quantity;
+    }
+
+    public double getPrice() {
+        return flower.getPrice() * quantity;
+    }
+
+    public Flower getFlower() {
+        return flower;
+    }
+}

src/main/java/flower/store/Rose.java

@@ -0,0 +1,12 @@
+package flower.store;
+
+import lombok.Getter;
+import lombok.Setter;
+
+@Getter
+@Setter
+public class Rose extends Flower {
+    public Rose() {
+        this.setFlowerType(FlowerType.ROSE);
+    }
+}

src/main/java/flower/store/Store.java

@@ -0,0 +1,20 @@
+package flower.store;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.stream.Collectors;
+
+public class Store {
+    private List<FlowerBucket> buckets = new ArrayList<>();
+
+    public void addBucket(FlowerBucket bucket) {
+        buckets.add(bucket);
+    }
+
+    public List<FlowerBucket> search(FlowerType flowerType) {
+        return buckets.stream()
+                .filter(bucket -> bucket.getFlowerPacks().stream()
+                    .anyMatch(pack -> pack.getFlower().getFlowerType() == flowerType))
+                .collect(Collectors.toList());
+    }
+}

src/main/java/flower/store/Tulip.java

@@ -0,0 +1,11 @@
+package flower.store;
+import lombok.Getter;
+import lombok.Setter;
+
+@Getter
+@Setter
+public class Tulip extends Flower {
+    public Tulip() {
+        this.setFlowerType(FlowerType.TULIP);
+    }
+}

src/test/java/flower/store/FlowerBucketTest.java

@@ -0,0 +1,35 @@
+package flower.store;
+
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.Assertions;
+
+import java.util.Random;
+
+public class FlowerBucketTest {
+    private static final Random RANDOM_GENERATOR = new Random();
+    private static final int MAX_QUANTITY = 1000;
+    private static final int MAX_PRICE = 100;
+
+    private FlowerBucket flowerBucket;
+
+    @BeforeEach
+    public void init() {
+        flowerBucket = new FlowerBucket();
+    }
+
+    @Test
+    public void testPrice() {
+    int price = 10;
+    int quantity = RANDOM_GENERATOR.nextInt(MAX_QUANTITY);
+    Flower flower = new Rose();
+    flower.setPrice(price);
+    FlowerPack flowerPack = new FlowerPack(flower, quantity);
+    flowerBucket.add(flowerPack);
+    int expectedPrice = price * quantity;
+    Assertions.assertEquals(expectedPrice, flowerBucket.getPrice());
+    }
+
+
+
+}