firebase-ios-sdk/FirebaseAuth/Tests/SampleSwift/SwiftApiTests/GoogleTests.swift

/*
 * Copyright 2020 Google LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import FirebaseAuth
import Foundation
import GTMSessionFetcherCore
import XCTest

class GoogleTests: TestsBase {
  func testSignInWithGoogle() throws {
    let auth = Auth.auth()
    let userInfoDict = getGoogleAccessToken()
    let googleAccessToken: String = try XCTUnwrap(userInfoDict["access_token"] as? String)
    let googleIDToken: String = try XCTUnwrap(userInfoDict["id_token"] as? String)
    let credential = GoogleAuthProvider.credential(withIDToken: googleIDToken,
                                                   accessToken: googleAccessToken)
    let expectation = self.expectation(description: "Signing in with Google finished.")
    auth.signIn(with: credential) { result, error in
      if let error = error {
        XCTFail("Signing in with Google had error: \(error)")
      }
      expectation.fulfill()
    }
    waitForExpectations(timeout: TestsBase.kExpectationsTimeout)
  }

  @available(iOS 13, tvOS 13, macOS 10.15, macCatalyst 13, watchOS 7, *)
  func testSignInWithGoogleAsync() async throws {
    let auth = Auth.auth()
    let userInfoDict = try await getGoogleAccessTokenAsync()
    let googleAccessToken: String = try XCTUnwrap(userInfoDict["access_token"] as? String)
    let googleIDToken: String = try XCTUnwrap(userInfoDict["id_token"] as? String)
    let credential = GoogleAuthProvider.credential(withIDToken: googleIDToken,
                                                   accessToken: googleAccessToken)
    _ = try await auth.signIn(with: credential)
    let displayName = try XCTUnwrap(auth.currentUser?.displayName)
    XCTAssertEqual(displayName, Credentials.kGoogleUserName)
  }

  /// Sends http request to Google OAuth2 token server to use refresh token to exchange for Google
  /// access token.
  /// Returns a dictionary that constains "access_token", "token_type", "expires_in" and sometimes
  /// the "id_token". (The id_token is not guaranteed to be returned during a refresh exchange; see
  /// https://openid.net/specs/openid-connect-core-1_0.html#RefreshTokenResponse)
  func getGoogleAccessToken() -> [String: Any] {
    var returnValue: [String: Any] = [:]
    let googleOauth2TokenServerUrl = "https://www.googleapis.com/oauth2/v4/token"
    let bodyString = "client_id=\(Credentials.kGoogleClientID)&grant_type=refresh_token" +
      "&refresh_token=\(Credentials.kGoogleTestAccountRefreshToken)"
    let postData = bodyString.data(using: .utf8)
    let service = GTMSessionFetcherService()
    let fetcher = service.fetcher(withURLString: googleOauth2TokenServerUrl)
    fetcher.bodyData = postData
    fetcher.setRequestValue("application/x-www-form-urlencoded", forHTTPHeaderField: "Content-Type")
    let expectation = self.expectation(description: "Exchanging Google account tokens finished.")
    fetcher.beginFetch { data, error in
      if let error {
        XCTFail("Exchanging Google account tokens finished with error: \(error)")
      } else {
        do {
          let data = try XCTUnwrap(data)
          returnValue = try JSONSerialization.jsonObject(with: data, options: [])
            as! [String: Any]
        } catch {
          XCTFail("Failed to unwrap data \(error)")
        }
      }
      expectation.fulfill()
    }
    waitForExpectations(timeout: TestsBase.kExpectationsTimeout)
    return returnValue
  }

  /// Sends http request to Google OAuth2 token server to use refresh token to exchange for Google
  /// access token.
  /// Returns a dictionary that constains "access_token", "token_type", "expires_in" and sometimes
  /// the "id_token". (The id_token is not guaranteed to be returned during a refresh exchange;
  /// see https://openid.net/specs/openid-connect-core-1_0.html#RefreshTokenResponse)
  @available(iOS 13, tvOS 13, macOS 10.15, macCatalyst 13, watchOS 7, *)
  func getGoogleAccessTokenAsync() async throws -> [String: Any] {
    let googleOauth2TokenServerUrl = "https://www.googleapis.com/oauth2/v4/token"
    let bodyString = "client_id=\(Credentials.kGoogleClientID)&grant_type=refresh_token" +
      "&refresh_token=\(Credentials.kGoogleTestAccountRefreshToken)"
    let postData = bodyString.data(using: .utf8)
    let service = GTMSessionFetcherService()
    let fetcher = service.fetcher(withURLString: googleOauth2TokenServerUrl)
    fetcher.bodyData = postData
    fetcher.setRequestValue(
      "application/x-www-form-urlencoded",
      forHTTPHeaderField: "Content-Type"
    )
    let data = try await fetcher.beginFetch()
    guard let returnValue = try JSONSerialization.jsonObject(with: data, options: [])
      as? [String: Any] else {
      XCTFail("Failed to serialize userInfo as a Dictionary")
      fatalError()
    }
    return returnValue
  }
}