feat(TransactionArchive): Implement TODO countDistinctWeeks() and cor…

…responding tests.
danieskj · Mar 25, 2026 · d1ae6ae · d1ae6ae
1 parent 4c6750c
commit d1ae6ae
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/src/main/java/edu/ntnu/idi/idatt/transaction/TransactionArchive.java b/src/main/java/edu/ntnu/idi/idatt/transaction/TransactionArchive.java
@@ -73,8 +73,11 @@ public List<Sale> getSales(int week) {
    *
    * @return
    */
-  public int countDistinctWeeks() { // TODO: HERE
-    return -1;
+  public int countDistinctWeeks() {
+    return (int) transactions.stream()
+        .map(Transaction::getWeek)
+        .distinct()
+        .count();
   }
 
 }
diff --git a/src/test/java/edu/ntnu/idi/idatt/transaction/TransactionArchiveTest.java b/src/test/java/edu/ntnu/idi/idatt/transaction/TransactionArchiveTest.java
@@ -87,4 +87,18 @@ void PTgetTransactions() {
 
   }
 
+  /**
+   * Tests for countDistinctWeeks().
+   */
+  @Test
+  void PTcountDistinctWeeks() {
+    assertEquals(2, transactionArchive.countDistinctWeeks());
+  }
+
+  @Test
+  void NTcountDistinctWeeks() {
+    TransactionArchive tArchive = new TransactionArchive();
+    assertEquals(0, tArchive.countDistinctWeeks());
+  }
+
 }