implementing feature steps (step1)

Author: Félix Carpena

src/test/java/group/rohlik/acceptance/steps/CartSteps.java

@@ -1,47 +1,121 @@
 package group.rohlik.acceptance.steps;
 
+import group.rohlik.entity.Cart;
+import group.rohlik.entity.CartLine;
+import group.rohlik.entity.CartRepository;
+import group.rohlik.entity.Discount;
+import io.cucumber.gherkin.internal.com.eclipsesource.json.JsonObject;
+import io.cucumber.java.Before;
 import io.cucumber.java.en.Given;
 import io.cucumber.java.en.Then;
 import io.cucumber.java.en.When;
+import org.junit.jupiter.api.Assertions;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.web.client.TestRestTemplate;
+import org.springframework.http.HttpEntity;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpMethod;
+import org.springframework.http.MediaType;
 
 public class CartSteps {
+
+    private final CartRepository cartRepository;
+    private final TestRestTemplate template;
+    private Long currentCartId;
+
+    @Autowired
+    public CartSteps(CartRepository cartRepository, TestRestTemplate template) {
+        this.cartRepository = cartRepository;
+        this.template = template;
+    }
+
+    @Before
+    public void setUp() {
+        currentCartId = null;
+    }
+
     @Given("I have a cart")
     public void haveCart() {
-        SetupSteps.notImplemented();
+        Cart cart = new Cart();
+        cartRepository.save(cart);
+        currentCartId = cart.getId();
     }
 
     @When("I add {int} unit(s) of product {string} to my cart")
     public void addProductUnitsToMyCart(int quantity, String sku) {
-        SetupSteps.notImplemented();
+        JsonObject body = new JsonObject();
+        body.add("sku", sku);
+        body.add("quantity", quantity);
+        HttpHeaders headers = new HttpHeaders();
+        headers.setContentType(MediaType.APPLICATION_JSON);
+
+        template.exchange(
+                String.format("/carts/%d/lines", currentCartId),
+                HttpMethod.POST,
+                new HttpEntity<>(body.toString(), headers),
+                String.class
+        );
     }
 
     @When("I remove product {string} of my cart")
     public void removeProductOfMyCart(String sku) {
-        SetupSteps.notImplemented();
+        addProductUnitsToMyCart(0, sku);
     }
 
     @Then("the cart's total cost should be {float} euro(s)")
     public void cartTotalCost(float amount) {
-        SetupSteps.notImplemented();
+        Cart cart = currentCart();
+        double totalProducts = cart
+                .getLines()
+                .stream()
+                .mapToDouble(currentCartLine -> currentCartLine.getQuantity() * currentCartLine.getProduct().getPrice())
+                .sum();
+        double totalDiscounts = cart.getDiscounts().stream().mapToDouble(Discount::getAmount).sum();
+        double totalPrice = totalProducts - totalDiscounts;
+
+        Assertions.assertEquals(Math.round(totalPrice * 100f) / 100f, amount);
     }
 
     @Then("there should be {int} unit(s) of product {string} in my cart")
     public void thereShouldBeProductUnitsInMyCart(int quantity, String sku) {
-        SetupSteps.notImplemented();
+        Cart cart = currentCart();
+        CartLine cartLine = cart.getLines()
+                .stream()
+                .filter(currentCartLine -> currentCartLine.getProduct().getSku().equals(sku))
+                .findFirst()
+                .orElseThrow();
+        Assertions.assertEquals(quantity, cartLine.getQuantity());
     }
 
     @Then("there shouldn't be product {string} in my cart")
     public void thereShouldNotBeProductInCart(String sku) {
-        SetupSteps.notImplemented();
+        Cart cart = currentCart();
+        CartLine cartLine = cart.getLines()
+                .stream()
+                .filter(currentCartLine -> currentCartLine.getProduct().getSku().equals(sku))
+                .findFirst()
+                .orElse(null);
+        Assertions.assertNull(cartLine, String.format("Product %s should not be present", sku));
     }
 
     @Then("there should be discount {long} in my cart")
     public void thereShouldBeDiscountInCart(long discountId) {
-        SetupSteps.notImplemented();
+        Cart cart = currentCart();
+        Discount discount = cart.getDiscounts()
+                .stream()
+                .filter(currentDiscount -> currentDiscount.getId() == discountId)
+                .findFirst()
+                .orElse(null);
+        Assertions.assertNotNull(discount, String.format("Discount %s should be present", discountId));
     }
 
     @Then("there shouldn't be discounts in my cart")
     public void thereShouldNotBeDiscountsInCart() {
-        SetupSteps.notImplemented();
+        Cart cart = currentCart();
+        Assertions.assertEquals(0, cart.getDiscounts().size());
+    }
+
+    private Cart currentCart() {
+        return cartRepository.findById(currentCartId).orElseThrow();
     }
 }

src/test/java/group/rohlik/acceptance/steps/DiscountSteps.java

@@ -1,12 +1,23 @@
 package group.rohlik.acceptance.steps;
 
+import group.rohlik.entity.Discount;
+import group.rohlik.entity.DiscountRepository;
+import io.cucumber.java.en.And;
 import io.cucumber.java.en.Given;
 import lombok.AllArgsConstructor;
+import org.springframework.transaction.annotation.Transactional;
 
 @AllArgsConstructor
 public class DiscountSteps {
+    private final DiscountRepository discountRepository;
+
     @Given("there is a cart discount {string} for {float} euros with a minimum total price of {float} euros")
+    @Transactional
     public void thereIsACartDiscountWithAMinimumTotalPrice(String name, float amount, float minPrice) {
-        SetupSteps.notImplemented();
+        Discount discount = new Discount();
+        discount.setName(name);
+        discount.setAmount(amount);
+        discount.setMinPrice(minPrice);
+        discountRepository.save(discount);
     }
 }

src/test/java/group/rohlik/acceptance/steps/ProductSteps.java

@@ -1,11 +1,29 @@
 package group.rohlik.acceptance.steps;
 
+import group.rohlik.entity.Product;
+import group.rohlik.entity.ProductRepository;
 import io.cucumber.datatable.DataTable;
 import io.cucumber.java.en.Given;
+import lombok.AllArgsConstructor;
+import org.springframework.transaction.annotation.Transactional;
 
+import java.util.Map;
+
+@AllArgsConstructor
 public class ProductSteps {
+    private final ProductRepository productRepository;
+
     @Given("the following products exist:")
+    @Transactional
     public void theFollowingProductsExists(DataTable table) {
-        SetupSteps.notImplemented();
+        table
+                .asMaps(String.class, String.class)
+                .forEach((Map<String, String> row) -> {
+                    Product product = new Product();
+                    product.setSku(row.get("sku"));
+                    product.setName(row.get("name"));
+                    product.setPrice(Float.parseFloat(row.get("price")));
+                    productRepository.save(product);
+                });
     }
 }

src/test/java/group/rohlik/acceptance/steps/SetupSteps.java

@@ -10,6 +10,7 @@
 
 @AllArgsConstructor
 public class SetupSteps {
+
     @PersistenceContext
     private EntityManager em;
 
@@ -28,8 +29,4 @@ public void truncateDB() {
     private void executeUpdate(String sql) {
         em.createNativeQuery(sql).executeUpdate();
     }
-
-    public static void notImplemented() {
-        throw new RuntimeException("Step method not implemented");
-    }
 }