mock_networking_service.swift

import Foundation
import OpenFeature
@testable import OFREP
@testable import GOFeatureFlag

public class MockNetworkingService: NetworkingService {
    var mockData: Data?
    var mockStatus: Int
    var mockURLResponse: URLResponse?
    var callCounter = 0
    var dataCollectorCallCounter = 0
    var dataCollectorEventCounter = 0
    var requests: [URLRequest] = []

    public init(mockStatus: Int = 200) {
        self.mockStatus = mockStatus
    }

    public func doRequest(for request: URLRequest) async throws -> (Data, URLResponse) {
        self.requests.append(request)
        self.callCounter+=1
        let isDataCollector = request.url?.absoluteString.contains("/v1/data/collector") ?? false
        let isBulkEvaluation = request.url?.absoluteString.contains("/ofrep/v1/evaluate/flags") ?? false
        if isDataCollector {
            self.dataCollectorCallCounter+=1
            let requestBody = try JSONDecoder().decode(DataCollectorRequest.self, from: request.httpBody!)
            if (requestBody.events != nil && requestBody.events!.count > 1){
                self.dataCollectorEventCounter += requestBody.events!.count
                let dataCollectorResponse = DataCollectorResponse(ingestedContentCount: requestBody.events!.count)
                let responseBody = try JSONEncoder().encode(dataCollectorResponse)
                let response = HTTPURLResponse(url: request.url!, statusCode: 200, httpVersion: nil, headerFields: nil)!
                return (responseBody, response)
            }

            if request.allHTTPHeaderFields?["Authorization"] != nil {
                if request.allHTTPHeaderFields?["Authorization"] == "Bearer apiKey1" {
                    let response = HTTPURLResponse(url: request.url!, statusCode: 200, httpVersion: nil, headerFields: nil)!
                    return (validResponse.data(using: .utf8)!, response)
                }

                let response = HTTPURLResponse(url: request.url!, statusCode: 401, httpVersion: nil, headerFields: nil)!
                return ("{}".data(using: .utf8)!, response)
            }

            if self.mockStatus == 200 {
                let response = HTTPURLResponse(url: request.url!, statusCode: self.mockStatus, httpVersion: nil, headerFields: nil)!
                return (validResponse.data(using: .utf8)!, response)
            }

            if self.mockStatus == 401 {
                let response = HTTPURLResponse(url: request.url!, statusCode: self.mockStatus, httpVersion: nil, headerFields: nil)!
                return ("{}".data(using: .utf8)!, response)
            }

            if self.mockStatus == 403 {
                let response = HTTPURLResponse(url: request.url!, statusCode: self.mockStatus, httpVersion: nil, headerFields: nil)!
                return ("{}".data(using: .utf8)!, response)
            }

            if self.mockStatus == 400 {
                // Invalid json response
                let response = HTTPURLResponse(url: request.url!, statusCode: 200, httpVersion: nil, headerFields: nil)!
                return (invalidResponse.data(using: .utf8)!, response)
            }

            if self.mockStatus >= 400 {
                let response = HTTPURLResponse(url: request.url!, statusCode: self.mockStatus, httpVersion: nil, headerFields: nil)!
                return ("{}".data(using: .utf8)!, response)
            }
        }

        if isBulkEvaluation {
            let headers = ["ETag": "33a64df551425fcc55e"]
            let response = HTTPURLResponse(url: request.url!, statusCode: 200, httpVersion: nil, headerFields: headers)!
            return (defaultResponse.data(using: .utf8)!, response)
        }

        let response = HTTPURLResponse(url: request.url!, statusCode: 500, httpVersion: nil, headerFields: nil)!
        return ("{}".data(using: .utf8)!, response)
    }

    let validResponse = "{\"ingestedContentCount\":1}"
    let invalidResponse = "{\"ingestedContentCount\":1"
    private let defaultResponse = """
    {
      "flags": [
        {
          "value": true,
          "key": "my-flag",
          "reason": "STATIC",
          "variant": "variantA",
          "metadata": {
            "additionalProp1": true,
            "additionalProp2": true,
            "additionalProp3": true
          }
        },
        {
          "value": true,
          "key": "bool-flag",
          "reason": "TARGETING_MATCH",
          "variant": "variantA",
          "metadata": {
            "additionalProp1": true,
            "additionalProp2": true,
            "additionalProp3": true
          }
        },
        {
          "value": 1234,
          "key": "int-flag",
          "reason": "TARGETING_MATCH",
          "variant": "variantA",
          "metadata": {
            "additionalProp1": true,
            "additionalProp2": true,
            "additionalProp3": true
          }
        },
        {
          "value": 12.34,
          "key": "double-flag",
          "reason": "TARGETING_MATCH",
          "variant": "variantA",
          "metadata": {
            "additionalProp1": true,
            "additionalProp2": true,
            "additionalProp3": true
          }
        },
        {
          "value": "1234value",
          "key": "string-flag",
          "reason": "TARGETING_MATCH",
          "variant": "variantA",
          "metadata": {
            "additionalProp1": true,
            "additionalProp2": true,
            "additionalProp3": true
          }
        },
        {
          "value": {"testValue":{"toto":1234}},
          "key": "object-flag",
          "reason": "TARGETING_MATCH",
          "variant": "variantA",
          "metadata": {
            "additionalProp1": true,
            "additionalProp2": true,
            "additionalProp3": true
          }
        },
        {
          "value": [1234, 5678],
          "key": "array-flag",
          "reason": "TARGETING_MATCH",
          "variant": "variantA",
          "metadata": {
            "additionalProp1": true,
            "additionalProp2": true,
            "additionalProp3": true
          }
        }
      ]
    }
"""
}