mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
76 lines
2.4 KiB
Swift
76 lines
2.4 KiB
Swift
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2016 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See http://swift.org/LICENSE.txt for license information
|
|
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
import ObjectiveC
|
|
import Foundation
|
|
|
|
internal var _temporaryNSLocaleCurrentLocale: NSLocale? = nil
|
|
|
|
extension NSLocale {
|
|
@objc
|
|
public class func _swiftUnittest_currentLocale() -> NSLocale {
|
|
return _temporaryNSLocaleCurrentLocale!
|
|
}
|
|
}
|
|
|
|
public func withOverriddenNSLocaleCurrentLocale<Result>(
|
|
temporaryLocale: NSLocale,
|
|
@noescape _ body: () -> Result
|
|
) -> Result {
|
|
let oldMethod = class_getClassMethod(
|
|
NSLocale.self, #selector(NSLocale.current))
|
|
precondition(oldMethod != nil, "could not find +[NSLocale currentLocale]")
|
|
|
|
let newMethod = class_getClassMethod(
|
|
NSLocale.self, #selector(NSLocale._swiftUnittest_currentLocale))
|
|
precondition(newMethod != nil, "could not find +[NSLocale _swiftUnittest_currentLocale]")
|
|
|
|
precondition(_temporaryNSLocaleCurrentLocale == nil,
|
|
"nested calls to withOverriddenNSLocaleCurrentLocale are not supported")
|
|
|
|
_temporaryNSLocaleCurrentLocale = temporaryLocale
|
|
method_exchangeImplementations(oldMethod, newMethod)
|
|
let result = body()
|
|
method_exchangeImplementations(newMethod, oldMethod)
|
|
_temporaryNSLocaleCurrentLocale = nil
|
|
|
|
return result
|
|
}
|
|
|
|
public func withOverriddenNSLocaleCurrentLocale<Result>(
|
|
temporaryLocaleIdentifier: String,
|
|
@noescape _ body: () -> Result
|
|
) -> Result {
|
|
precondition(
|
|
NSLocale.availableLocaleIdentifiers().contains(temporaryLocaleIdentifier),
|
|
"requested locale \(temporaryLocaleIdentifier) is not available")
|
|
|
|
return withOverriddenNSLocaleCurrentLocale(
|
|
NSLocale(localeIdentifier: temporaryLocaleIdentifier), body)
|
|
}
|
|
|
|
/// Executes the `body` in an autorelease pool if the platform does not
|
|
/// implement the return-autoreleased optimization.
|
|
///
|
|
/// (Currently, only the i386 iOS and watchOS simulators don't implement the
|
|
/// return-autoreleased optimization.)
|
|
@inline(never)
|
|
public func autoreleasepoolIfUnoptimizedReturnAutoreleased(
|
|
@noescape body: () -> Void
|
|
) {
|
|
#if arch(i386) && (os(iOS) || os(watchOS))
|
|
autoreleasepool(body)
|
|
#else
|
|
body()
|
|
#endif
|
|
}
|