2 files changed, 148 insertions, 0 deletions
diff --git a/tests/nutrient-measurement-model.rkt b/tests/nutrient-measurement-model.rkt
new file mode 100644
index 0000000..b0b053d
--- /dev/null
+++ b/tests/nutrient-measurement-model.rkt
@@ -0,0 +1,72 @@
+#lang racket
+
+(module+ test
+  (require rackunit
+           rackunit/text-ui
+           "../db/conn.rkt"
+           "../db/migrations.rkt"
+           "../models/nutrient.rkt"
+           "../models/nutrient-measurement.rkt")
+
+  (define measured-on "2025-09-01")
+
+  (run-tests
+   (test-suite
+    "Nutrient measurement model"
+    #:before (λ ()
+               (connect! #:path 'memory)
+               ;; (connect! #:path "test.sqlite3")
+               (migrate-all!)
+               (create-nutrient! "Nitrogen" "N")
+               (create-nutrient! "Phosphorus" "P")
+               (create-nutrient! "Potassium" "K"))
+    #:after (λ ()
+              (disconnect!))
+
+    (test-case "Create measurement with values"
+      (define nitrogen (get-nutrient #:name "Nitrogen"))
+      (define phosphorus (get-nutrient #:name "Phosphorus"))
+      (create-nutrient-measurement! measured-on (list
+                                                 (cons nitrogen 12.3)
+                                                 (cons phosphorus 4.5)))
+      (check-equal? (length (get-nutrient-measurements)) 1)
+      (define nm (get-nutrient-measurement #:measured-on measured-on))
+      (check-true (nutrient-measurement? nm))
+      (check-equal? (nutrient-measurement-measured-on nm) measured-on)
+      (define mvs (get-nutrient-measurement-values nm))
+      (check-equal? (length mvs) 2)
+      (check-equal? (cdr (assoc nitrogen mvs)) 12.3)
+      (check-equal? (cdr (assoc phosphorus mvs)) 4.5)
+      )
+
+    #;(test-case "Update a single measurement value"
+        (define nitrogen (get-nutrient #:name "Nitrogen"))
+        (define nm (get-nutrient-measurement #:measured-on measured-on))
+        (update-nutrient-measurement! nm #:nutrient-values (list (cons nitrogen 1.1)))
+        (define mvs (get-nutrient-measurement-values nm))
+        (check-equal? (length mvs) 2)
+        (check-equal? (cdr (assoc nitrogen mvs)) 1.1))
+
+    #;(test-case "Upsert measurement values"
+        (define nitrogen (get-nutrient #:name "Nitrogen"))
+        (define phosphorus (get-nutrient #:name "Phosphorus"))
+        (define potassium (get-nutrient #:name "Potassium"))
+        (define nm (get-nutrient-measurement #:measured-on measured-on))
+        ;; Upsert: set K=8.8 and change N to 10.0, keep P as-is
+        (update-nutrient-measurement! nm
+                                      #:nutrient-values (list
+                                                         (cons nitrogen 10.0)
+                                                         (cons potassium 8.8)))
+        (define mvs (get-nutrient-measurement-values nm))
+        (check-equal? (length mvs) 3)
+        (check-equal? (cdr (assoc nitrogen mvs)) 10.0)
+        (check-equal? (cdr (assoc potassium mvs)) 8.8)
+        ;; P should still be present at 4.5
+        (check-equal? (cdr (assoc phosphorus mvs)) 4.5))
+
+    (test-case "Delete measurement cascades its values"
+      (define nm (get-nutrient-measurement #:measured-on measured-on))
+      (delete-nutrient-measurement! nm)
+      (check-false (get-nutrient-measurement #:id (nutrient-measurement-id nm)))
+      (check-equal? (length (get-nutrient-measurements)) 0)
+      (check-true (null? (get-nutrient-measurement-values nm)))))))
diff --git a/tests/nutrient-model.rkt b/tests/nutrient-model.rkt
new file mode 100644
index 0000000..39a574c
--- /dev/null
+++ b/tests/nutrient-model.rkt
@@ -0,0 +1,76 @@
+#lang racket
+
+(module+ test
+  (require rackunit
+           rackunit/text-ui
+           "../db/conn.rkt"
+           "../db/migrations.rkt"
+           "../models/nutrient.rkt")
+
+  (run-tests
+   (test-suite
+    "Nutrient model"
+    #:before (λ ()
+               (connect! #:path 'memory)
+               (migrate-all!))
+    #:after (λ ()
+              (disconnect!))
+
+    (test-case "Create nutrients"
+      (create-nutrient! "Examplium" "Ex")
+      (check-equal? (length (get-nutrients)) 1)
+      (create-nutrient! "Ignorium" "Ig")
+      (check-equal? (length (get-nutrients)) 2))
+
+    (test-case "Read nutrient"
+      (define examplium (get-nutrient #:id 1))
+      (check-true (nutrient? examplium))
+      (check-equal? (nutrient-id examplium) 1))
+    
+    (test-case "Read nutrient by name"
+      (define examplium (get-nutrient #:name "Examplium"))
+      (check-true (nutrient? examplium))
+      (check-equal? (nutrient-name examplium) "Examplium"))
+
+    (test-case "Read nutrient by formula"
+      (define examplium (get-nutrient #:formula "Ex"))
+      (check-true (nutrient? examplium))
+      (check-equal? (nutrient-formula examplium) "Ex"))
+    
+    (test-case "Read inexisting nutrient"
+      (check-false (get-nutrient #:name "Inexistium")))
+
+    (test-case "Update nutrient name"
+      (define examplium (get-nutrient #:name "Examplium"))
+      (define examplium-nitrate
+        (update-nutrient! examplium #:name "Examplium Nitrate"))
+      (check-equal? (length (get-nutrients)) 2)
+      (check-equal? (nutrient-name examplium-nitrate) "Examplium Nitrate")
+      (check-equal? (nutrient-formula examplium-nitrate) "Ex"))
+
+    (test-case "Update nutrient formula"
+      (define examplium-nitrate (get-nutrient #:name "Examplium Nitrate"))
+      (define examplium-sulfate
+        (update-nutrient! examplium-nitrate #:formula "ExSO4"))
+      (check-equal? (length (get-nutrients)) 2)
+      (check-equal? (nutrient-name examplium-sulfate) "Examplium Nitrate")
+      (check-equal? (nutrient-formula examplium-sulfate) "ExSO4"))
+
+    (test-case "Update nutrient name and formula"
+      (define examplium-nitrate
+        (get-nutrient #:name "Examplium Nitrate"))
+      (define examplium-sulfate
+        (update-nutrient! examplium-nitrate
+                          #:name "Examplium Sulfate"
+                          #:formula "ExNO3"))
+      (check-equal? (length (get-nutrients)) 2)
+      (check-equal? (nutrient-name examplium-sulfate) "Examplium Sulfate")
+      (check-equal? (nutrient-formula examplium-sulfate) "ExNO3"))
+
+    (test-case "Delete nutrient"
+      (define examplium-sulfate (get-nutrient #:name "Examplium Sulfate"))
+      (delete-nutrient! examplium-sulfate)
+      (check-equal? (length (get-nutrients)) 1)
+      (define ignorium (get-nutrient #:name "Ignorium"))
+      (delete-nutrient! ignorium)
+      (check-equal? (length (get-nutrients)) 0)))))