Add back iob clamping tests that were lost in meal detection merge.

Author: ps2

LoopTests/Managers/LoopDataManagerDosingTests.swift

@@ -27,8 +27,8 @@ class MockDelegate: LoopDataManagerDelegate {
         self.recommendation = automaticDose.recommendation
         completion(error)
     }
-    func roundBasalRate(unitsPerHour: Double) -> Double { unitsPerHour }
-    func roundBolusVolume(units: Double) -> Double { units }
+    func roundBasalRate(unitsPerHour: Double) -> Double { Double(Int(unitsPerHour / 0.05)) * 0.05 }
+    func roundBolusVolume(units: Double) -> Double { Double(Int(units / 0.05)) * 0.05 }
     var pumpManagerStatus: PumpManagerStatus?
     var cgmManagerStatus: CGMManagerStatus?
     var pumpStatusHighlight: DeviceStatusHighlight?
@@ -331,7 +331,7 @@ class LoopDataManagerDosingTests: LoopDataManagerTests {
         }
         loopDataManager.loop()
         wait(for: [exp], timeout: 1.0)
-        let expectedAutomaticDoseRecommendation = AutomaticDoseRecommendation(basalAdjustment: TempBasalRecommendation(unitsPerHour: 4.577747629410191, duration: .minutes(30)))
+        let expectedAutomaticDoseRecommendation = AutomaticDoseRecommendation(basalAdjustment: TempBasalRecommendation(unitsPerHour: 4.55, duration: .minutes(30)))
         XCTAssertEqual(delegate.recommendation, expectedAutomaticDoseRecommendation)
         XCTAssertEqual(dosingDecisionStore.dosingDecisions.count, 1)
         if dosingDecisionStore.dosingDecisions.count == 1 {
@@ -355,7 +355,7 @@ class LoopDataManagerDosingTests: LoopDataManagerTests {
         }
         loopDataManager.loop()
         wait(for: [exp], timeout: 1.0)
-        let expectedAutomaticDoseRecommendation = AutomaticDoseRecommendation(basalAdjustment: TempBasalRecommendation(unitsPerHour: 4.577747629410191, duration: .minutes(30)))
+        let expectedAutomaticDoseRecommendation = AutomaticDoseRecommendation(basalAdjustment: TempBasalRecommendation(unitsPerHour: 4.55, duration: .minutes(30)))
         XCTAssertNil(delegate.recommendation)
         XCTAssertEqual(dosingDecisionStore.dosingDecisions.count, 1)
         XCTAssertEqual(dosingDecisionStore.dosingDecisions[0].reason, "loop")
@@ -456,4 +456,81 @@ class LoopDataManagerDosingTests: LoopDataManagerTests {
         waitOnMain(timeout: 5)
         XCTAssertNil(mockDelegate.recommendation)
     }
+
+    func testAutoBolusMaxIOBClamping() {
+        /// `maximumBolus` is set to clamp the automatic dose
+        /// Autobolus without clamping: 0.65 U. Clamped recommendation: 0.2 U.
+        setUp(for: .autoBolusIOBClamping, maxBolus: 5)
+
+        // This sets up dose rounding
+        let delegate = MockDelegate()
+        loopDataManager.delegate = delegate
+
+        let updateGroup = DispatchGroup()
+        updateGroup.enter()
+
+        var insulinOnBoard: InsulinValue?
+        var recommendedBolus: Double?
+        self.loopDataManager.getLoopState { _, state in
+            insulinOnBoard = state.insulinOnBoard
+            recommendedBolus = state.recommendedAutomaticDose?.recommendation.bolusUnits
+            updateGroup.leave()
+        }
+        updateGroup.wait()
+
+        XCTAssertEqual(recommendedBolus!, 0.5, accuracy: 0.01)
+        XCTAssertEqual(insulinOnBoard?.value, 9.47)
+
+        /// Set the `maximumBolus` to 10U so there's no clamping
+        updateGroup.enter()
+        self.loopDataManager.mutateSettings { settings in settings.maximumBolus = 10 }
+        self.loopDataManager.getLoopState { _, state in
+            insulinOnBoard = state.insulinOnBoard
+            recommendedBolus = state.recommendedAutomaticDose?.recommendation.bolusUnits
+            updateGroup.leave()
+        }
+        updateGroup.wait()
+
+        XCTAssertEqual(recommendedBolus!, 0.65, accuracy: 0.01)
+        XCTAssertEqual(insulinOnBoard?.value, 9.47)
+    }
+
+    func testTempBasalMaxIOBClamping() {
+        /// `maximumBolus` is set to 5U to clamp max IOB at 10U
+        /// Without clamping: 4.25 U/hr. Clamped recommendation: 1.25 U/hr.
+        setUp(for: .tempBasalIOBClamping, maxBolus: 5)
+
+        // This sets up dose rounding
+        let delegate = MockDelegate()
+        loopDataManager.delegate = delegate
+
+        let updateGroup = DispatchGroup()
+        updateGroup.enter()
+
+        var insulinOnBoard: InsulinValue?
+        var recommendedBasal: TempBasalRecommendation?
+        self.loopDataManager.getLoopState { _, state in
+            insulinOnBoard = state.insulinOnBoard
+            recommendedBasal = state.recommendedAutomaticDose?.recommendation.basalAdjustment
+            updateGroup.leave()
+        }
+        updateGroup.wait()
+
+        XCTAssertEqual(recommendedBasal!.unitsPerHour, 1.25, accuracy: 0.01)
+        XCTAssertEqual(insulinOnBoard?.value, 9.87)
+
+        /// Set the `maximumBolus` to 10U so there's no clamping
+        updateGroup.enter()
+        self.loopDataManager.mutateSettings { settings in settings.maximumBolus = 10 }
+        self.loopDataManager.getLoopState { _, state in
+            insulinOnBoard = state.insulinOnBoard
+            recommendedBasal = state.recommendedAutomaticDose?.recommendation.basalAdjustment
+            updateGroup.leave()
+        }
+        updateGroup.wait()
+
+        XCTAssertEqual(recommendedBasal!.unitsPerHour, 4.25, accuracy: 0.01)
+        XCTAssertEqual(insulinOnBoard?.value, 9.87)
+    }
+
 }