Osiris/Tests/OsirisTests/HTTPRequestTests.swift

//
//  HTTPRequestTests.swift
//  OsirisTests
//
//  Created by Sami Samhuri on 2025-06-15.
//

@testable import Osiris
import XCTest

class HTTPRequestTests: XCTestCase {
    let baseURL = URL(string: "https://api.example.net")!
    
    func testHTTPRequestInitialization() {
        let request = HTTPRequest(method: .get, url: baseURL)
        XCTAssertEqual(request.method, .get)
        XCTAssertEqual(request.url, baseURL)
        XCTAssertEqual(request.contentType, .none)
        XCTAssertNil(request.parameters)
        XCTAssertTrue(request.headers.isEmpty)
        XCTAssertTrue(request.parts.isEmpty)
    }
    
    func testHTTPRequestWithParameters() {
        let params = ["key": "value", "number": 42] as [String: any Sendable]
        let request = HTTPRequest(method: .post, url: baseURL, contentType: .json, parameters: params)
        
        XCTAssertEqual(request.method, .post)
        XCTAssertEqual(request.contentType, .json)
        XCTAssertNotNil(request.parameters)
    }
    
    func testGETConvenience() {
        let request = HTTPRequest.get(baseURL)
        XCTAssertEqual(request.method, .get)
        XCTAssertEqual(request.url, baseURL)
        XCTAssertEqual(request.contentType, .none)
    }
    
    func testPOSTConvenience() {
        let params = ["name": "Jane"]
        let request = HTTPRequest.post(baseURL, contentType: .json, parameters: params)
        
        XCTAssertEqual(request.method, .post)
        XCTAssertEqual(request.contentType, .json)
        XCTAssertNotNil(request.parameters)
    }
    
    func testPUTConvenience() {
        let params = ["name": "Jane"]
        let request = HTTPRequest.put(baseURL, contentType: .formEncoded, parameters: params)
        
        XCTAssertEqual(request.method, .put)
        XCTAssertEqual(request.contentType, .formEncoded)
        XCTAssertNotNil(request.parameters)
    }
    
    func testDELETEConvenience() {
        let request = HTTPRequest.delete(baseURL)
        XCTAssertEqual(request.method, .delete)
        XCTAssertEqual(request.url, baseURL)
        XCTAssertEqual(request.contentType, .none)
    }
    
    func testMultipartPartsAutomaticallySetContentType() {
        var request = HTTPRequest.post(baseURL)
        XCTAssertEqual(request.contentType, .none)
        
        request.parts = [.text("value", name: "field")]
        XCTAssertEqual(request.contentType, .multipart)
    }
    
    #if canImport(UIKit)
    func testAddMultipartJPEG() {
        var request = HTTPRequest.post(baseURL)
        
        // Create a simple 1x1 pixel image
        let size = CGSize(width: 1, height: 1)
        UIGraphicsBeginImageContext(size)
        let image = UIGraphicsGetImageFromCurrentImageContext()!
        UIGraphicsEndImageContext()
        
        request.addMultipartJPEG(name: "avatar", image: image, quality: 0.8, filename: "test.jpg")
        
        XCTAssertEqual(request.parts.count, 1)
        XCTAssertEqual(request.contentType, .multipart)
        
        let part = request.parts.first!
        XCTAssertEqual(part.name, "avatar")
        
        if case let .binaryData(_, type, filename) = part.content {
            XCTAssertEqual(type, "image/jpeg")
            XCTAssertEqual(filename, "test.jpg")
        } else {
            XCTFail("Expected binary data content")
        }
    }
    
    func testAddMultipartJPEGWithInvalidQuality() {
        var request = HTTPRequest.post(baseURL)
        
        // Create a valid image
        let size = CGSize(width: 1, height: 1)
        UIGraphicsBeginImageContext(size)
        let image = UIGraphicsGetImageFromCurrentImageContext()!
        UIGraphicsEndImageContext()
        
        // Test with extreme quality values that might cause issues
        request.addMultipartJPEG(name: "avatar1", image: image, quality: -1.0)
        request.addMultipartJPEG(name: "avatar2", image: image, quality: 2.0)
        
        // The method should handle extreme quality values gracefully
        // Either by clamping them or by having jpegData handle them
        XCTAssertTrue(request.parts.count >= 0) // Should not crash
    }
    #endif
    
    func testHTTPRequestPATCHConvenience() {
        let params = ["status": "active"]
        let request = HTTPRequest(method: .patch, url: baseURL, contentType: .json, parameters: params)
        
        XCTAssertEqual(request.method, .patch)
        XCTAssertEqual(request.contentType, .json)
        XCTAssertNotNil(request.parameters)
    }
    
    func testHTTPRequestWithMultipleHeaders() {
        var request = HTTPRequest.get(baseURL)
        request.addHeader(name: "Authorization", value: "Bearer token123")
        request.addHeader(name: "User-Agent", value: "Osiris/2.0")
        request.addHeader(name: "Accept", value: "application/json")
        
        XCTAssertEqual(request.headers["Authorization"], "Bearer token123")
        XCTAssertEqual(request.headers["User-Agent"], "Osiris/2.0")
        XCTAssertEqual(request.headers["Accept"], "application/json")
        XCTAssertEqual(request.headers.count, 3)
    }
    
    func testHTTPRequestOverwriteHeaders() {
        var request = HTTPRequest.get(baseURL)
        request.addHeader(name: "Accept", value: "application/xml")
        request.addHeader(name: "Accept", value: "application/json") // Should overwrite
        
        XCTAssertEqual(request.headers["Accept"], "application/json")
        XCTAssertEqual(request.headers.count, 1)
    }
    
    func testHTTPRequestWithEmptyMultipartParts() {
        var request = HTTPRequest.post(baseURL)
        request.parts = [] // Empty parts array
        
        XCTAssertEqual(request.contentType, .none) // Should not be set to multipart
        XCTAssertTrue(request.parts.isEmpty)
    }
    
    func testHTTPRequestMultipartPartsResetContentType() {
        var request = HTTPRequest.post(baseURL, contentType: .json)
        XCTAssertEqual(request.contentType, .json)
        
        request.parts = [.text("test", name: "field")]
        XCTAssertEqual(request.contentType, .multipart) // Should be automatically changed
        
        request.parts = [] // Clear parts
        XCTAssertEqual(request.contentType, .multipart) // Should remain multipart
    }
}