File: AtomicCounterTests.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 (42 lines) | stat: -rw-r--r-- 1,444 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
/*
 This source file is part of the Swift.org open source project

 Copyright (c) 2021 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 Swift project authors
*/

import XCTest
@testable import Markdown

final class AtomicCounterTests: XCTestCase {
    func testIncremental() {
        XCTAssertEqual(AtomicCounter.current, AtomicCounter.current)
        XCTAssertNotEqual(AtomicCounter.next(), AtomicCounter.next())
    }

    func testSimultaneousFetch() {
        var counters = Set<UInt64>()
        let group = DispatchGroup()
        let fetchQueue = DispatchQueue(label: "AtomicCounterTests.testSimultaneousFetch.fetch", attributes: [.concurrent])
        let collectQueue = DispatchQueue(label: "AtomicCounterTests.testSimultaneousFetch.collect")
        let numTasks = 4
        let idsPerQueue = 200000
        for _ in 0..<numTasks {
            group.enter()
            fetchQueue.async {
                let ids = (0..<idsPerQueue).map { _ in AtomicCounter.next() }
                collectQueue.sync {
                    for id in ids {
                        counters.insert(id)
                    }
                }
                group.leave()
            }
        }
        group.wait()
        XCTAssertEqual(numTasks * idsPerQueue, counters.count)
    }
}