File: LockedStateTests.swift

package info (click to toggle)
swiftlang 6.0.3-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 2,519,992 kB
  • sloc: cpp: 9,107,863; ansic: 2,040,022; asm: 1,135,751; python: 296,500; objc: 82,456; f90: 60,502; lisp: 34,951; pascal: 19,946; sh: 18,133; perl: 7,482; ml: 4,937; javascript: 4,117; makefile: 3,840; awk: 3,535; xml: 914; fortran: 619; cs: 573; ruby: 573
file content (81 lines) | stat: -rw-r--r-- 2,764 bytes parent folder | download
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
//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2023 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//

#if canImport(TestSupport)
import TestSupport
#endif

#if FOUNDATION_FRAMEWORK
@testable import Foundation
#elseif canImport(FoundationInternationalization)
@testable import FoundationInternationalization
#endif // FOUNDATION_FRAMEWORK

final class LockedStateTests : XCTestCase {
    final class TestObject {}
    struct TestError: Error {}

    func testWithLockDoesNotExtendLifetimeOfState() {
        weak var state: TestObject?
        let lockedState: LockedState<TestObject>

        (state, lockedState) = {
            let state = TestObject()
            return (state, LockedState(initialState: state))
        }()

        lockedState.withLock { state in
            weak var oldState = state
            state = TestObject()
            XCTAssertNil(oldState, "State object lifetime was extended after reassignment within body")
        }

        XCTAssertNil(state, "State object lifetime was extended beyond end of call")
    }

    func testWithLockExtendingLifespanDoesExtendLifetimeOfState() {
        weak var state: TestObject?
        let lockedState: LockedState<TestObject>

        (state, lockedState) = {
            let state = TestObject()
            return (state, LockedState(initialState: state))
        }()

        lockedState.withLockExtendingLifetimeOfState { state in
            weak var oldState = state
            state = TestObject()
            XCTAssertNotNil(oldState, "State object lifetime was not extended after reassignment within body")
        }

        XCTAssertNil(state, "State object lifetime was extended beyond end of call")
    }

    func testWithLockExtendingLifespanReleasesLockWhenBodyThrows() {
        let lockedState = LockedState(initialState: TestObject())

        XCTAssertThrowsError(
            try lockedState.withLockExtendingLifetimeOfState { _ in
                throw TestError()
            },
            "The body was expected to throw an error, but it did not."
        )

        // ⚠️ This test fails by crashing. If the lock was not properly released by the
        // `withLockExtendingLifetimeOfState()` call above, the following `withLock()`
        // call will abort the program.

        lockedState.withLock { _ in
            // PASS
        }
    }
}