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 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150
|
//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift open source project
//
// Copyright (c) 2020-2022 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
//
//===----------------------------------------------------------------------===//
@testable import Basics
import SPMTestSupport
import Workspace
import XCTest
import class TSCBasic.InMemoryFileSystem
final class AuthorizationProviderTests: XCTestCase {
func testNetrcAuthorizationProviders() throws {
let observability = ObservabilitySystem.makeForTesting()
// custom .netrc file
do {
let fileSystem: FileSystem = InMemoryFileSystem()
let customPath = try fileSystem.homeDirectory.appending(components: UUID().uuidString, "custom-netrc-file")
try fileSystem.createDirectory(customPath.parentDirectory, recursive: true)
try fileSystem.writeFileContents(
customPath,
string: "machine mymachine.labkey.org login custom@labkey.org password custom"
)
let configuration = Workspace.Configuration.Authorization(netrc: .custom(customPath), keychain: .disabled)
let authorizationProvider = try configuration.makeAuthorizationProvider(fileSystem: fileSystem, observabilityScope: observability.topScope) as? CompositeAuthorizationProvider
let netrcProviders = authorizationProvider?.providers.compactMap { $0 as? NetrcAuthorizationProvider }
XCTAssertEqual(netrcProviders?.count, 1)
XCTAssertEqual(try netrcProviders?.first.map { try resolveSymlinks($0.path) }, try resolveSymlinks(customPath))
let auth = authorizationProvider?.authentication(for: "https://mymachine.labkey.org")
XCTAssertEqual(auth?.user, "custom@labkey.org")
XCTAssertEqual(auth?.password, "custom")
// delete it
try fileSystem.removeFileTree(customPath)
XCTAssertThrowsError(try configuration.makeAuthorizationProvider(fileSystem: fileSystem, observabilityScope: observability.topScope), "error expected") { error in
XCTAssertEqual(error as? StringError, StringError("Did not find netrc file at \(customPath)."))
}
}
// user .netrc file
do {
let fileSystem = InMemoryFileSystem()
let userPath = try fileSystem.homeDirectory.appending(".netrc")
try fileSystem.createDirectory(userPath.parentDirectory, recursive: true)
try fileSystem.writeFileContents(
userPath,
string: "machine mymachine.labkey.org login user@labkey.org password user"
)
let configuration = Workspace.Configuration.Authorization(netrc: .user, keychain: .disabled)
let authorizationProvider = try configuration.makeAuthorizationProvider(fileSystem: fileSystem, observabilityScope: observability.topScope) as? CompositeAuthorizationProvider
let netrcProviders = authorizationProvider?.providers.compactMap { $0 as? NetrcAuthorizationProvider }
XCTAssertEqual(netrcProviders?.count, 1)
XCTAssertEqual(try netrcProviders?.first.map { try resolveSymlinks($0.path) }, try resolveSymlinks(userPath))
let auth = authorizationProvider?.authentication(for: "https://mymachine.labkey.org")
XCTAssertEqual(auth?.user, "user@labkey.org")
XCTAssertEqual(auth?.password, "user")
// delete it
do {
try fileSystem.removeFileTree(userPath)
let authorizationProvider = try configuration.makeAuthorizationProvider(fileSystem: fileSystem, observabilityScope: observability.topScope) as? CompositeAuthorizationProvider
XCTAssertNil(authorizationProvider)
}
}
}
func testRegistryNetrcAuthorizationProviders() throws {
let observability = ObservabilitySystem.makeForTesting()
// custom .netrc file
do {
let fileSystem: FileSystem = InMemoryFileSystem()
let customPath = try fileSystem.homeDirectory.appending(components: UUID().uuidString, "custom-netrc-file")
try fileSystem.createDirectory(customPath.parentDirectory, recursive: true)
try fileSystem.writeFileContents(
customPath,
string: "machine mymachine.labkey.org login custom@labkey.org password custom"
)
let configuration = Workspace.Configuration.Authorization(netrc: .custom(customPath), keychain: .disabled)
let netrcProvider = try configuration.makeRegistryAuthorizationProvider(fileSystem: fileSystem, observabilityScope: observability.topScope) as? NetrcAuthorizationProvider
XCTAssertNotNil(netrcProvider)
XCTAssertEqual(try netrcProvider.map { try resolveSymlinks($0.path) }, try resolveSymlinks(customPath))
let auth = netrcProvider?.authentication(for: "https://mymachine.labkey.org")
XCTAssertEqual(auth?.user, "custom@labkey.org")
XCTAssertEqual(auth?.password, "custom")
// delete it
try fileSystem.removeFileTree(customPath)
XCTAssertThrowsError(try configuration.makeRegistryAuthorizationProvider(fileSystem: fileSystem, observabilityScope: observability.topScope), "error expected") { error in
XCTAssertEqual(error as? StringError, StringError("did not find netrc file at \(customPath)"))
}
}
// user .netrc file
do {
let fileSystem = InMemoryFileSystem()
let userPath = try fileSystem.homeDirectory.appending(".netrc")
try fileSystem.createDirectory(userPath.parentDirectory, recursive: true)
try fileSystem.writeFileContents(
userPath,
string: "machine mymachine.labkey.org login user@labkey.org password user"
)
let configuration = Workspace.Configuration.Authorization(netrc: .user, keychain: .disabled)
let netrcProvider = try configuration.makeRegistryAuthorizationProvider(fileSystem: fileSystem, observabilityScope: observability.topScope) as? NetrcAuthorizationProvider
XCTAssertNotNil(netrcProvider)
XCTAssertEqual(try netrcProvider.map { try resolveSymlinks($0.path) }, try resolveSymlinks(userPath))
let auth = netrcProvider?.authentication(for: "https://mymachine.labkey.org")
XCTAssertEqual(auth?.user, "user@labkey.org")
XCTAssertEqual(auth?.password, "user")
// delete it
do {
try fileSystem.removeFileTree(userPath)
let authorizationProvider = try configuration.makeRegistryAuthorizationProvider(fileSystem: fileSystem, observabilityScope: observability.topScope) as? NetrcAuthorizationProvider
// Even if user .netrc file doesn't exist, the provider will be non-nil but contain no data.
XCTAssertNotNil(authorizationProvider)
XCTAssertEqual(try authorizationProvider.map { try resolveSymlinks($0.path) }, try resolveSymlinks(userPath))
XCTAssertTrue(authorizationProvider!.machines.isEmpty)
}
}
}
}
|