more tests

Author: Sajjon

crates/core/src/logic/calendar_logic.rs

@@ -932,4 +932,77 @@ mod tests {
         );
         assert!(result.is_ok());
     }
+
+    #[test]
+    fn try_from_period_anno_for_year_and_month() {
+        let sut = YearAndMonth::sample();
+        let period_anno = PeriodAnno::YearAndMonth(sut);
+        let outcome = YearAndMonth::try_from_period_anno(period_anno).unwrap();
+        assert_eq!(outcome, sut);
+    }
+
+    #[test]
+    fn try_from_period_anno_for_year_month_and_fortnight() {
+        let sut = YearMonthAndFortnight::sample();
+        let period_anno = PeriodAnno::YearMonthAndFortnight(sut);
+        let outcome = YearMonthAndFortnight::try_from_period_anno(period_anno).unwrap();
+        assert_eq!(outcome, sut);
+    }
+
+    #[test]
+    fn test_quantity_in_period_throws_when_cadence_is_biweekly_and_granularity_is_month() {
+        let target_period = YearAndMonth::january(2025);
+        let record_of_periods_off = RecordOfPeriodsOff::new([]);
+        let result = quantity_in_period(
+            &target_period,
+            Granularity::Month,
+            Cadence::BiWeekly,
+            &record_of_periods_off,
+        );
+        assert!(result.is_err());
+        if let Err(Error::CannotInvoiceForMonthWhenCadenceIsBiWeekly) = result {
+            // Expected error
+        } else {
+            panic!("Expected CannotInvoiceForMonthWhenCadenceIsBiWeekly error");
+        }
+    }
+
+    #[test]
+    fn test_quantity_in_period_granularity_fortnight_cadence_monthly_is_two() {
+        let target_period = YearAndMonth::january(2025);
+        let record_of_periods_off = RecordOfPeriodsOff::new([]);
+        let result = quantity_in_period(
+            &target_period,
+            Granularity::Fortnight,
+            Cadence::Monthly,
+            &record_of_periods_off,
+        );
+        assert!(result.is_ok());
+        assert_eq!(result.unwrap(), Quantity::TWO);
+    }
+
+    #[test]
+    fn test_quantity_in_period_granularity_fortnight_cadence_biweekly_is_one() {
+        let target_period = YearAndMonth::january(2025);
+        let record_of_periods_off = RecordOfPeriodsOff::new([]);
+        let result = quantity_in_period(
+            &target_period,
+            Granularity::Fortnight,
+            Cadence::BiWeekly,
+            &record_of_periods_off,
+        );
+        assert!(result.is_ok());
+        assert_eq!(result.unwrap(), Quantity::ONE);
+    }
+
+    #[test]
+    fn test_one_half_earlier_second() {
+        let ym = YearAndMonth::january(2025);
+        let month = YearMonthAndFortnight::year_and_month_with_half(ym, MonthHalf::Second);
+        let one_half_earlier = month.one_half_earlier();
+        assert_eq!(
+            one_half_earlier,
+            YearMonthAndFortnight::year_and_month_with_half(ym, MonthHalf::First)
+        );
+    }
 }

crates/core/src/logic/command/command.rs

@@ -626,4 +626,56 @@ mod tests {
             email_settings.smtp_app_password().expose_secret()
         );
     }
+
+    #[test]
+    fn test_record_expenses_with_base_path_throws_when_cadence_is_monthly_and_period_is_fortnight()
+    {
+        let tempdir = tempfile::tempdir().expect("Failed to create temp dir");
+        let period = YearMonthAndFortnight::builder()
+            .year(Year::from(2025))
+            .month(Month::January)
+            .half(MonthHalf::First)
+            .build();
+        let services_fees: ServiceFees = ServiceFees::builder()
+            .cadence(Cadence::Monthly)
+            .rate(Rate::daily(UnitPrice::ONE))
+            .name("Sample Service Fees".to_owned())
+            .build()
+            .unwrap();
+        save_to_disk(&services_fees, service_fees_path(tempdir.path())).unwrap();
+        let result = record_expenses_with_base_path(
+            &period,
+            &[Item::sample_expense_breakfast()],
+            tempdir.path().to_path_buf(),
+        );
+        assert!(
+            result.is_err(),
+            "Expected error when recording expenses for fortnight with monthly cadence, got: {:?}",
+            result
+        );
+    }
+
+    #[test]
+    fn test_record_expenses_with_base_path_throws_when_cadence_is_bi_weekly_and_period_is_year_and_month()
+     {
+        let tempdir = tempfile::tempdir().expect("Failed to create temp dir");
+        let period = YearAndMonth::may(2025);
+        let services_fees: ServiceFees = ServiceFees::builder()
+            .cadence(Cadence::BiWeekly)
+            .rate(Rate::daily(UnitPrice::ONE))
+            .name("Sample Service Fees".to_owned())
+            .build()
+            .unwrap();
+        save_to_disk(&services_fees, service_fees_path(tempdir.path())).unwrap();
+        let result = record_expenses_with_base_path(
+            &period,
+            &[Item::sample_expense_breakfast()],
+            tempdir.path().to_path_buf(),
+        );
+        assert!(
+            result.is_err(),
+            "Expected error when recording expenses for year and month with bi-weekly cadence, got: {:?}",
+            result
+        );
+    }
 }

crates/core/src/models/data/data.rs

@@ -374,4 +374,42 @@ mod tests {
             );
         }
     }
+
+    #[test]
+    fn test_to_partial_when_offset_is_year_month_and_fortnight() {
+        let offset_period = YearMonthAndFortnight::builder()
+            .year(2025.into())
+            .month(Month::May)
+            .half(MonthHalf::First)
+            .build();
+        let sut = Data::<YearMonthAndFortnight>::builder()
+            .information(
+                ProtoInvoiceInfo::builder()
+                    .offset(
+                        TimestampedInvoiceNumber::<YearMonthAndFortnight>::builder()
+                            .offset(100.into())
+                            .period(offset_period)
+                            .build(),
+                    )
+                    .build(),
+            )
+            .vendor(CompanyInformation::sample_vendor())
+            .client(CompanyInformation::sample_client())
+            .payment_info(PaymentInformation::sample())
+            .service_fees(ServiceFees::sample())
+            .expensed_periods(ExpensedPeriods::sample())
+            .build();
+        let target_period = YearMonthAndFortnight::builder()
+            .year(2025.into())
+            .month(Month::May)
+            .half(MonthHalf::First)
+            .build();
+        let input = ValidInput::builder()
+            .items(InvoicedItems::Service { time_off: None })
+            .period(target_period)
+            .build();
+        let partial = sut.to_partial(input).unwrap();
+        let invoice_date = partial.information().invoice_date();
+        assert_eq!(*invoice_date, target_period.to_date_end_of_period());
+    }
 }