Bundle+Tests.swift 912 B

123456789101112131415161718192021222324
  1. // Copyright 2021 Google LLC
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. import Foundation
  15. extension Bundle {
  16. class func mock(with cls: Bundle.Type) {
  17. guard let original = class_getClassMethod(Bundle.self, #selector(getter: main)),
  18. let swizzled = class_getClassMethod(cls, #selector(getter: main))
  19. else { return }
  20. method_exchangeImplementations(original, swizzled)
  21. }
  22. }