Trio/TrioTests/CoreDataTests/TempTargetStorageTests.swift

import CoreData
import Foundation
import Swinject
import Testing

@testable import Trio

@Suite("TempTargetStorage Tests", .serialized) struct TempTargetsStorageTests: Injectable {
    @Injected() var storage: TempTargetsStorage!
    let resolver: Resolver
    var coreDataStack: CoreDataStack!
    var testContext: NSManagedObjectContext!

    init() async throws {
        // Create test context
        coreDataStack = try await CoreDataStack.createForTests()
        testContext = coreDataStack.newTaskContext()

        // Create assembler with test assembly
        let assembler = Assembler([
            StorageAssembly(),
            ServiceAssembly(),
            APSAssembly(),
            NetworkAssembly(),
            UIAssembly(),
            SecurityAssembly(),
            TestAssembly(testContext: testContext) // Add our test assembly last to override TempTargetStorage
        ])

        resolver = assembler.resolver
        injectServices(resolver)
    }

    @Test("Storage is correctly initialized") func testStorageInitialization() {
        // Verify storage exists
        #expect(storage != nil, "TempTargetsStorage should be injected")

        // Verify it's the correct type
        #expect(
            storage is BaseTempTargetsStorage, "Storage should be of type BaseTempTargetsStorage"
        )
    }

    @Test("Store and retrieve temp target") func testStoreAndRetrieveTempTarget() async throws {
        // Given
        let testTarget = TempTarget(
            name: "Test Target",
            createdAt: Date(),
            targetTop: 120,
            targetBottom: 120,
            duration: 60,
            enteredBy: "Test",
            reason: "Testing",
            isPreset: false,
            halfBasalTarget: 160
        )

        // When
        try await storage.storeTempTarget(tempTarget: testTarget)

        // Then verify stored entries
        let storedEntries = try await coreDataStack.fetchEntitiesAsync(
            ofType: TempTargetStored.self,
            onContext: testContext,
            predicate: NSPredicate(format: "name == %@", "Test Target"),
            key: "date",
            ascending: false
        ) as? [TempTargetStored]

        #expect(storedEntries?.isEmpty == false, "Should have stored entries")
        #expect(storedEntries?.count == 1, "Should have exactly one entry")
        let storedTarget = storedEntries?.first
        #expect(storedTarget?.name == "Test Target", "Name should match")
        #expect(storedTarget?.target == 120, "Target should match")
        #expect(storedTarget?.duration == 60, "Duration should match")
    }

    @Test("Delete temp target Preset") func testDeleteTempTarget() async throws {
        // Given
        let testTarget = TempTarget(
            name: "Delete Test",
            createdAt: Date(),
            targetTop: 120,
            targetBottom: 120,
            duration: 60,
            enteredBy: "Test",
            reason: "Testing deletion of a preset",
            isPreset: true,
            halfBasalTarget: 160
        )
        // Store the target
        try await storage.storeTempTarget(tempTarget: testTarget)

        // Get the stored target's ObjectID
        let storedEntries = try await coreDataStack.fetchEntitiesAsync(
            ofType: TempTargetStored.self,
            onContext: testContext,
            predicate: NSPredicate(format: "name == %@", "Delete Test"),
            key: "date",
            ascending: false
        ) as? [TempTargetStored]

        guard let objectID = storedEntries?.first?.objectID else {
            throw TestError("Failed to get stored target's ObjectID")
        }

        // When
        await storage.deleteTempTargetPreset(objectID)

        // Then verify deletion
        let remainingEntries = try await coreDataStack.fetchEntitiesAsync(
            ofType: TempTargetStored.self,
            onContext: testContext,
            predicate: NSPredicate(format: "name == %@", "Delete Test"),
            key: "date",
            ascending: false
        ) as? [TempTargetStored]

        #expect(remainingEntries?.isEmpty == true, "Should have no entries after deletion")
    }

    @Test("Get temp targets not yet uploaded to Nightscout") func testGetTempTargetsNotYetUploadedToNightscout() async throws {
        // Given
        let testTarget = TempTarget(
            name: "NS Test",
            createdAt: Date(),
            targetTop: 120,
            targetBottom: 120,
            duration: 45,
            enteredBy: "Test",
            reason: "Testing NS Upload",
            isPreset: true,
            enabled: true,
            halfBasalTarget: 160
        )

        // When
        try await storage.storeTempTarget(tempTarget: testTarget)
        let notUploadedTargets = try await storage.getTempTargetsNotYetUploadedToNightscout()

        // Then
        #expect(!notUploadedTargets.isEmpty, "Should have targets not uploaded to NS")
        let target = notUploadedTargets[0]
        #expect(target.eventType == .nsTempTarget, "Event type should be NS temp target")
        #expect(target.duration == 45, "Duration should match")
        #expect(target.targetTop == 120, "Target top should match target")
        #expect(target.targetBottom == 120, "Target bottom should match target")
    }
}