Trio/TrioTests/OpenAPSSwiftTests/DetermineBasalJsonTests.swift

import Foundation
import Testing
@testable import Trio

@Suite("DetermineBasal testing using JSON inputs", .serialized) struct DetermineBasalJsonTests {
    let timeZoneForTests = TimeZoneForTests()

    @Test(
        "DetermineBasal should produce same results for fixed JS",
        .enabled(if: ReplayTests.enabled)
    ) func replayErrorInputs() async throws {
        // Note: This test case can only test one timezone per invocation
        // so you need to manually change this to try out errors from
        // different timezones
        let testingTimezone = ReplayTests.timezone
        let files = try await HttpFiles.listFiles()
        for filePath in files {
            let algorithmComparison = try await HttpFiles.downloadFile(at: filePath)
            print("Checking \(filePath) @ \(algorithmComparison.createdAt)")
            guard algorithmComparison.timezone == testingTimezone else {
                continue
            }
            guard let determineBasalInput = algorithmComparison.determineBasalInput else {
                print("Skipping, no determineBasalInput found")
                if let str = algorithmComparison.comparisonError {
                    print(str)
                }
                if let str = algorithmComparison.swiftException {
                    print(str)
                }
                continue
            }

            timeZoneForTests.setTimezone(identifier: algorithmComparison.timezone)

            try await checkFixedJsAgainstSwift(determineBasalInput: determineBasalInput)
            print("Checked \(filePath) \(algorithmComparison.timezone)")
            timeZoneForTests.resetTimezone()
        }
    }

    func checkFixedJsAgainstSwift(determineBasalInput: DetermineBasalInputs) async throws {
        let openAps = OpenAPSFixed()
        let (determineBasalResultSwift, _) = OpenAPSSwift.determineBasal(
            glucose: determineBasalInput.glucose,
            currentTemp: determineBasalInput.currentTemp,
            iob: try JSONBridge.to(determineBasalInput.iob),
            profile: try JSONBridge.to(determineBasalInput.profile),
            autosens: try JSONBridge.to(determineBasalInput.autosens),
            meal: try JSONBridge.to(determineBasalInput.meal),
            microBolusAllowed: determineBasalInput.microBolusAllowed,
            reservoir: determineBasalInput.reservoir ?? 0,
            pumpHistory: determineBasalInput.pumpHistory,
            preferences: determineBasalInput.preferences,
            basalProfile: determineBasalInput.basalProfile,
            trioCustomOrefVariables: determineBasalInput.trioCustomOrefVariables,
            clock: determineBasalInput.clock
        )

        let determineBasalResultJavascript = try await openAps.determineBasalJavascript(
            glucose: determineBasalInput.glucose,
            currentTemp: determineBasalInput.currentTemp,
            iob: try JSONBridge.to(determineBasalInput.iob),
            profile: try JSONBridge.to(determineBasalInput.profile),
            autosens: try JSONBridge.to(determineBasalInput.autosens),
            meal: try JSONBridge.to(determineBasalInput.meal),
            microBolusAllowed: determineBasalInput.microBolusAllowed,
            reservoir: determineBasalInput.reservoir ?? 0,
            pumpHistory: determineBasalInput.pumpHistory,
            preferences: determineBasalInput.preferences,
            basalProfile: determineBasalInput.basalProfile,
            trioCustomOrefVariables: determineBasalInput.trioCustomOrefVariables,
            clock: determineBasalInput.clock
        )

        let comparison = JSONCompare.createComparison(
            function: .determineBasal,
            swift: determineBasalResultSwift,
            swiftDuration: 0.1,
            javascript: determineBasalResultJavascript,
            javascriptDuration: 0.1,
            iobInputs: nil,
            mealInputs: nil,
            autosensInputs: nil,
            determineBasalInputs: nil
        )

        if comparison.resultType == .valueDifference {
            print(comparison.differences!.prettyPrintedJSON!)
        }

        if comparison.resultType != .matching {
            print("REPLAY ERROR: Fixed JS didn't match")
        }

        #expect(comparison.resultType == .matching)
    }

    @Test("Format determineBasal inputs for running in JS", .enabled(if: false)) func formatInputs() async throws {
        let openAps = OpenAPSFixed()

        // this test is meant for one-off analysis so it's ok to hard code
        // a file, just make sure to _not_ check in updates to this to
        // avoid polluting our change logs
        let algorithmComparison = try await HttpFiles.downloadFile(at: "/files/f1d04efa-c39b-4f0a-9955-65ab663ff9fb.0.json")
        let determineBasalInput = algorithmComparison.determineBasalInput!

        let encoder = JSONCoding.encoder
        let output = try encoder.encode(determineBasalInput)

        let sharedDir = FileManager.default.temporaryDirectory
        let outputURL = sharedDir.appendingPathComponent("determine_basal_error_inputs.json")
        // Print the path so you can find it
        print("Writing to: \(outputURL.path)")
        try output.write(to: outputURL)

        timeZoneForTests.setTimezone(identifier: algorithmComparison.timezone)

        let (determineBasalResultSwift, _) = OpenAPSSwift.determineBasal(
            glucose: determineBasalInput.glucose,
            currentTemp: determineBasalInput.currentTemp,
            iob: try JSONBridge.to(determineBasalInput.iob),
            profile: try JSONBridge.to(determineBasalInput.profile),
            autosens: try JSONBridge.to(determineBasalInput.autosens),
            meal: try JSONBridge.to(determineBasalInput.meal),
            microBolusAllowed: determineBasalInput.microBolusAllowed,
            reservoir: determineBasalInput.reservoir ?? 0,
            pumpHistory: determineBasalInput.pumpHistory,
            preferences: determineBasalInput.preferences,
            basalProfile: determineBasalInput.basalProfile,
            trioCustomOrefVariables: determineBasalInput.trioCustomOrefVariables,
            clock: determineBasalInput.clock
        )

        print("Swift result")
        switch determineBasalResultSwift {
        case let .success(rawJson):
            print(rawJson)
        case let .failure(error):
            print(error.localizedDescription)
        }

        let determineBasalResultJavascript = try await openAps.determineBasalJavascript(
            glucose: determineBasalInput.glucose,
            currentTemp: determineBasalInput.currentTemp,
            iob: try JSONBridge.to(determineBasalInput.iob),
            profile: try JSONBridge.to(determineBasalInput.profile),
            autosens: try JSONBridge.to(determineBasalInput.autosens),
            meal: try JSONBridge.to(determineBasalInput.meal),
            microBolusAllowed: determineBasalInput.microBolusAllowed,
            reservoir: determineBasalInput.reservoir ?? 0,
            pumpHistory: determineBasalInput.pumpHistory,
            preferences: determineBasalInput.preferences,
            basalProfile: determineBasalInput.basalProfile,
            trioCustomOrefVariables: determineBasalInput.trioCustomOrefVariables,
            clock: determineBasalInput.clock
        )

        print("Fixed JS result")
        switch determineBasalResultJavascript {
        case let .success(rawJson):
            print(rawJson)
        case let .failure(error):
            print(error.localizedDescription)
        }

        let comparison = JSONCompare.createComparison(
            function: .determineBasal,
            swift: determineBasalResultSwift,
            swiftDuration: 0.1,
            javascript: determineBasalResultJavascript,
            javascriptDuration: 0.1,
            iobInputs: nil,
            mealInputs: nil,
            autosensInputs: nil,
            determineBasalInputs: nil
        )

        if comparison.resultType == .valueDifference {
            print(comparison.differences!.prettyPrintedJSON!)
            printForecasts(comparison.differences)
        }

        #expect(comparison.resultType == .matching)

        timeZoneForTests.resetTimezone()
    }

    func printForecasts(_ values: [String: Any]?) {
        guard let values = values else { return }
        guard let forecasts = values["predBGs"] as? Trio.ValueDifference else { return }
        let js = forecasts.js.toDictionary()
        let swift = forecasts.swift.toDictionary()

        for forecastType in ["IOB", "ZT", "UAM", "COB"] {
            print("")
            guard let swiftForecast = swift[forecastType]?.toIntArray(),
                  let jsForecast = js[forecastType]?.toIntArray()
            else {
                print("missing \(forecastType) forecast, skipping")
                continue
            }
            if swiftForecast.count == jsForecast.count {
                print(forecastType)
            } else {
                print("\(forecastType) has length mismatch ❌")
            }
            print("Row\tSft\tJS\tMatch")
            print("--------------")
            for (row, values) in zip(swiftForecast, jsForecast).enumerated() {
                let pass: String
                if abs(values.0 - values.1) <= 1 {
                    pass = "✅"
                } else {
                    pass = "❌"
                }
                print("\(row)\t\(values.0)\t\(values.1)\t\(pass)")
            }
        }
    }
}

extension JSONValue {
    func toDictionary() -> [String: Trio.JSONValue] {
        switch self {
        case let .object(dict):
            return dict
        default:
            fatalError()
        }
    }

    func toIntArray() -> [Int] {
        switch self {
        case let .array(array):
            return array.map { $0.toInt() }
        default:
            fatalError()
        }
    }

    func toInt() -> Int {
        switch self {
        case let .number(number):
            return Int(number)
        default:
            fatalError()
        }
    }
}