WeakMapTable/Tests/WeakMapTableTests/WeakMapTableTests.swift at 98e7f0930d8fd7a1487f19e8b77fafa1437b10c6 · Woollim/WeakMapTable · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
import XCTest
import WeakMapTable

final class WeakMapTableTests: XCTestCase {
  func testSetValueForKey() {
    let map = WeakMapTable<KeyObject, ValueObject>()

    let (key1, value1) = (KeyObject(), ValueObject())
    let (key2, value2) = (KeyObject(), ValueObject())
    map.setValue(value1, forKey: key1)
    map.setValue(value2, forKey: key2)

    XCTAssert(map.value(forKey: key1) === value1)
    XCTAssert(map.value(forKey: key2) === value2)
  }

  func testSetAnotherValue() {
    let map = WeakMapTable<KeyObject, ValueObject>()

    let key = KeyObject()
    weak var weakOldValue: ValueObject?
    weak var weakNewValue: ValueObject?

    _ = {
      let oldValue = ValueObject()
      map.setValue(oldValue, forKey: key)
      weakOldValue = oldValue

      let newValue = ValueObject()
      map.setValue(newValue, forKey: key)
      weakNewValue = newValue
    }()

    XCTAssertNil(weakOldValue)
    XCTAssertNotNil(weakNewValue)
    XCTAssert(map.value(forKey: key) === weakNewValue)
  }

  func testSetNil() {
    let map = WeakMapTable<KeyObject, ValueObject>()

    let key = KeyObject()
    weak var weakValue: ValueObject?

    _ = {
      let value = ValueObject()
      map.setValue(value, forKey: key)
      weakValue = value

      map.setValue(nil, forKey: key)
    }()

    XCTAssertNil(map.value(forKey: key))
    XCTAssertNil(weakValue)
  }

  func testDefaultValue() {
    let map = WeakMapTable<KeyObject, ValueObject>()

    let key = KeyObject()
    let expectedValue = ValueObject()
    let actualValue = map.value(forKey: key, default: expectedValue)

    XCTAssert(actualValue === expectedValue)
  }

  func testReleaseKeyAndValue() {
    let map = WeakMapTable<KeyObject, ValueObject>()

    weak var weakKey: KeyObject?
    weak var weakValue: ValueObject?

    _ = {
      let key = KeyObject()
      let value = ValueObject()
      map.setValue(value, forKey: key)
      weakKey = key
      weakValue = value
    }()

    XCTAssertNil(weakKey)
    XCTAssertNil(weakValue)
  }

  func testRemoveAll() {
    let map = WeakMapTable<KeyObject, ValueObject>()
    let key = KeyObject()
    weak var weakValue: ValueObject?

    _ = {
      let value = ValueObject()
      map.setValue(value, forKey: key)
      weakValue = value

      map.removeAll()
    }()

    XCTAssertNil(map.value(forKey: key))
    XCTAssertNil(weakValue)
  }
}

private final class KeyObject {
}

private final class ValueObject {
}