packages/shared_preferences/shared_preferences_foundation/darwin/Tests/RunnerTests.swift

// Copyright 2013 The Flutter Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

import XCTest

#if os(iOS)
import Flutter
#elseif os(macOS)
import FlutterMacOS
#endif

@testable import shared_preferences_foundation

class RunnerTests: XCTestCase {
  let prefixes: [String] = ["aPrefix", ""]

  func testSetAndGet() throws {
    for aPrefix in prefixes {
      let plugin = SharedPreferencesPlugin()

      plugin.setBool(key: "\(aPrefix)aBool", value: true)
      plugin.setDouble(key: "\(aPrefix)aDouble", value: 3.14)
      plugin.setValue(key: "\(aPrefix)anInt", value: 42)
      plugin.setValue(key: "\(aPrefix)aString", value: "hello world")
      plugin.setValue(key: "\(aPrefix)aStringList", value: ["hello", "world"])

      let storedValues = plugin.getAll(prefix: aPrefix, allowList: nil)
      XCTAssertEqual(storedValues["\(aPrefix)aBool"] as? Bool, true)
      XCTAssertEqual(storedValues["\(aPrefix)aDouble"] as! Double, 3.14, accuracy: 0.0001)
      XCTAssertEqual(storedValues["\(aPrefix)anInt"] as? Int, 42)
      XCTAssertEqual(storedValues["\(aPrefix)aString"] as? String, "hello world")
      XCTAssertEqual(storedValues["\(aPrefix)aStringList"] as? Array<String>, ["hello", "world"])
    }
  }

  func testGetWithAllowList() throws {
    for aPrefix in prefixes {
      let plugin = SharedPreferencesPlugin()

      plugin.setBool(key: "\(aPrefix)aBool", value: true)
      plugin.setDouble(key: "\(aPrefix)aDouble", value: 3.14)
      plugin.setValue(key: "\(aPrefix)anInt", value: 42)
      plugin.setValue(key: "\(aPrefix)aString", value: "hello world")
      plugin.setValue(key: "\(aPrefix)aStringList", value: ["hello", "world"])

      let storedValues = plugin.getAll(prefix: aPrefix, allowList: ["\(aPrefix)aBool"])
      XCTAssertEqual(storedValues["\(aPrefix)aBool"] as? Bool, true)
      XCTAssertNil(storedValues["\(aPrefix)aDouble"] ?? nil)
      XCTAssertNil(storedValues["\(aPrefix)anInt"] ?? nil)
      XCTAssertNil(storedValues["\(aPrefix)aString"] ?? nil)
      XCTAssertNil(storedValues["\(aPrefix)aStringList"] ?? nil)
    }
  }

  func testRemove() throws {
    for aPrefix in prefixes {
      let plugin = SharedPreferencesPlugin()
      let testKey = "\(aPrefix)foo"
      plugin.setValue(key: testKey, value: 42)

      // Make sure there is something to remove, so the test can't pass due to a set failure.
      let preRemovalValues = plugin.getAll(prefix: aPrefix, allowList: nil)
      XCTAssertEqual(preRemovalValues[testKey] as? Int, 42)

      // Then verify that removing it works.
      plugin.remove(key: testKey)

      let finalValues = plugin.getAll(prefix: aPrefix, allowList: nil)
      XCTAssertNil(finalValues[testKey] as Any?)
    }
  }
    
    func testClearWithNoAllowlist() throws {
      for aPrefix in prefixes {
        let plugin = SharedPreferencesPlugin()
        let testKey = "\(aPrefix)foo"
        plugin.setValue(key: testKey, value: 42)

        // Make sure there is something to clear, so the test can't pass due to a set failure.
        let preRemovalValues = plugin.getAll(prefix: aPrefix, allowList: nil)
        XCTAssertEqual(preRemovalValues[testKey] as? Int, 42)

        // Then verify that clearing works.
        plugin.clear(prefix: aPrefix, allowList: nil)

        let finalValues = plugin.getAll(prefix: aPrefix, allowList: nil)
        XCTAssertNil(finalValues[testKey] as Any?)
      }
    }
    
    func testClearWithAllowlist() throws {
      for aPrefix in prefixes {
        let plugin = SharedPreferencesPlugin()
        let testKey = "\(aPrefix)foo"
        plugin.setValue(key: testKey, value: 42)

        // Make sure there is something to clear, so the test can't pass due to a set failure.
        let preRemovalValues = plugin.getAll(prefix: aPrefix, allowList: nil)
        XCTAssertEqual(preRemovalValues[testKey] as? Int, 42)

        plugin.clear(prefix: aPrefix, allowList: ["\(aPrefix)notfoo"])

        let finalValues = plugin.getAll(prefix: aPrefix, allowList: nil)
          XCTAssertEqual(finalValues[testKey] as? Int, 42)
      }
    }
  
}