Add PricePlanTest

Author: superxcgm

rest/domain/PricePlan.h

@@ -2,32 +2,52 @@
 #define DEVELOPER_JOYOFENERGY_CPP_BEAST_PRICEPLAN_H
 
 #include <string>
+#include <utility>
 #include <vector>
 
 class PricePlan {
   // todo:
  public:
+  using time_point_type = std::chrono::time_point<std::chrono::system_clock>;
   class PeakTimeMultiplier {
+   public:
     enum DayOfWeek {
+      SUNDAY,
       MONDAY,
       TUESDAY,
+      WEDNESDAY,
+      THURSDAY,
+      FRIDAY,
+      SATURDAY
     };
 
+    PeakTimeMultiplier(DayOfWeek dayOfWeek, int multiplier) : dayOfWeek(dayOfWeek), multiplier(multiplier) {}
+
     DayOfWeek dayOfWeek;
     int multiplier;
-
-    PeakTimeMultiplier(DayOfWeek dayOfWeek, int multiplier) : dayOfWeek(dayOfWeek), multiplier(multiplier) {}
   };
 
   PricePlan(std::string planName, std::string energySupplier, int unitRate, std::vector<PeakTimeMultiplier> peakTimeMultipliers)
-      : planName(planName), energySupplier(energySupplier), unitRate(unitRate), peakTimeMultipliers(peakTimeMultipliers) {}
+      : planName(std::move(planName)), energySupplier(std::move(energySupplier)), unitRate(unitRate), peakTimeMultipliers(std::move(peakTimeMultipliers)) {}
 
   std::string getEnergySupplier() const { return energySupplier; }
 
   std::string getPlanName() const { return planName; }
 
   int getUnitRate() const { return unitRate; }
 
+  int getPrice(time_point_type dateTime) const {
+    auto time_t_dateTime = std::chrono::system_clock::to_time_t(dateTime);
+    auto t = std::localtime(&time_t_dateTime);
+    auto it = std::find_if(peakTimeMultipliers.begin(), peakTimeMultipliers.end(), [=](auto &p) {
+      return p.dayOfWeek == t->tm_wday;
+    });
+    if (it == peakTimeMultipliers.end()) {
+      return unitRate;
+    }
+    return unitRate * it->multiplier;
+  }
+
  private:
   const std::string energySupplier;
   const std::string planName;

test/CMakeLists.txt

@@ -8,5 +8,10 @@ add_executable(controller_test)
 target_sources(controller_test PRIVATE controller/MeterReadingControllerTest.cpp controller/PricePlanComparatorControllerTest.cpp)
 target_link_libraries(controller_test PRIVATE GTest::gmock_main nlohmann_json::nlohmann_json rest)
 
+add_executable(domain_test)
+target_sources(domain_test PRIVATE domain/PricePlanTest.cpp)
+target_link_libraries(domain_test PRIVATE GTest::gmock_main rest)
+
 add_test(endpoint_test endpoint_test)
-add_test(controller_test controller_test)
+add_test(controller_test controller_test)
+add_test(domain_test domain_test)

test/domain/PricePlanTest.cpp

@@ -0,0 +1,44 @@
+#include <gmock/gmock.h>
+#include <rest/domain/PricePlan.h>
+#include <rest/controller/MeterReadingController.h>
+
+using ::testing::Eq;
+
+TEST(PricePlanTest, GetEnergySupplierShouldReturnTheEnergySupplierGivenInTheConstructor) {
+  const std::string energy_supplier = "Energy Supplier Name";
+
+  PricePlan p("", energy_supplier, 0, {});
+
+  EXPECT_THAT(p.getEnergySupplier(), energy_supplier);
+}
+
+TEST(PricePlanTest, GetPriceShouldReturnTheBasePriceGivenAnOrdinaryDateTime) {
+  auto time = detail::fromRfc3339("2021-09-30T06:42:15.725202Z");
+  PricePlan::PeakTimeMultiplier peak_time_multiplier(PricePlan::PeakTimeMultiplier::DayOfWeek::WEDNESDAY, 10);
+  PricePlan price_plan("", "", 1, {peak_time_multiplier});
+
+  auto price = price_plan.getPrice(time);
+
+  EXPECT_THAT(price, 1);
+}
+
+TEST(PricePlanTest, GetPriceShouldReturnAnExceptionPriceGivenExceptionalDateTime) {
+  auto time = detail::fromRfc3339("2021-09-29T06:42:15.725202Z");
+  PricePlan::PeakTimeMultiplier peak_time_multiplier(PricePlan::PeakTimeMultiplier::DayOfWeek::WEDNESDAY, 10);
+  PricePlan price_plan("", "", 1, {peak_time_multiplier});
+
+  auto price = price_plan.getPrice(time);
+
+  EXPECT_THAT(price, 10);
+}
+
+TEST(PricePlanTest, GetPriceShouldReceiveMultipleExceptionalDateTimes) {
+  auto time = detail::fromRfc3339("2021-09-29T06:42:15.725202Z");
+  PricePlan::PeakTimeMultiplier peak_time_multiplier(PricePlan::PeakTimeMultiplier::DayOfWeek::WEDNESDAY, 10);
+  PricePlan::PeakTimeMultiplier another_peak_time_multiplier(PricePlan::PeakTimeMultiplier::DayOfWeek::THURSDAY, 10);
+  PricePlan price_plan("", "", 1, {peak_time_multiplier, another_peak_time_multiplier});
+
+  auto price = price_plan.getPrice(time);
+
+  EXPECT_THAT(price, 10);
+}