firebase-ios-sdk/Example/Storage/Tests/Unit/FIRStorageUpdateMetadataTests.m

// Copyright 2017 Google
//
// 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 "FIRStorageMetadata_Private.h"
#import "FIRStorageTestHelpers.h"
#import "FIRStorageUpdateMetadataTask.h"

@interface FIRStorageUpdateMetadataTests : XCTestCase

@property(strong, nonatomic) GTMSessionFetcherService *fetcherService;
@property(strong, nonatomic) FIRStorageMetadata *metadata;
@property(strong, nonatomic) FIRStorage *storage;
@property(strong, nonatomic) id mockApp;

@end

@implementation FIRStorageUpdateMetadataTests

- (void)setUp {
  [super setUp];

  NSDictionary *metadataDict = @{ @"bucket" : @"bucket", @"name" : @"path/to/object" };
  self.metadata = [[FIRStorageMetadata alloc] initWithDictionary:metadataDict];

  id mockOptions = OCMClassMock([FIROptions class]);
  OCMStub([mockOptions storageBucket]).andReturn(@"bucket.appspot.com");

  self.mockApp = OCMClassMock([FIRApp class]);
  OCMStub([self.mockApp name]).andReturn(kFIRStorageAppName);
  OCMStub([(FIRApp *)self.mockApp options]).andReturn(mockOptions);

  self.fetcherService = [[GTMSessionFetcherService alloc] init];
  self.fetcherService.authorizer =
      [[FIRStorageTokenAuthorizer alloc] initWithApp:self.mockApp
                                      fetcherService:self.fetcherService];

  self.storage = [FIRStorage storageForApp:self.mockApp];
}

- (void)tearDown {
  self.fetcherService = nil;
  self.storage = nil;
  self.mockApp = nil;
  [super tearDown];
}

- (void)testFetcherConfiguration {
  XCTestExpectation *expectation = [self expectationWithDescription:@"testSuccessfulFetch"];

  self.fetcherService.testBlock = ^(GTMSessionFetcher *fetcher,
                                    GTMSessionFetcherTestResponse response) {
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Warc-retain-cycles"
    XCTAssertEqualObjects(fetcher.request.URL, [FIRStorageTestHelpers objectURL]);
    XCTAssertEqualObjects(fetcher.request.HTTPMethod, @"PATCH");
    NSData *bodyData = [NSData frs_dataFromJSONDictionary:[self.metadata updatedMetadata]];
    XCTAssertEqualObjects(fetcher.request.HTTPBody, bodyData);
    NSDictionary *HTTPHeaders = fetcher.request.allHTTPHeaderFields;
    XCTAssertEqualObjects(HTTPHeaders[@"Content-Type"], @"application/json; charset=UTF-8");
    XCTAssertEqualObjects(HTTPHeaders[@"Content-Length"], [@(bodyData.length) stringValue]);
#pragma clang diagnostic pop
    NSHTTPURLResponse *httpResponse =
        [[NSHTTPURLResponse alloc] initWithURL:fetcher.request.URL
                                    statusCode:200
                                   HTTPVersion:kHTTPVersion
                                  headerFields:nil];
    response(httpResponse, nil, nil);
    self.fetcherService.testBlock = nil;
  };

  FIRStoragePath *path = [FIRStorageTestHelpers objectPath];
  FIRStorageReference *ref = [[FIRStorageReference alloc] initWithStorage:self.storage path:path];
  FIRStorageUpdateMetadataTask *task = [[FIRStorageUpdateMetadataTask alloc]
      initWithReference:ref
         fetcherService:self.fetcherService
               metadata:self.metadata
             completion:^(FIRStorageMetadata *_Nullable metadata, NSError *_Nullable error) {
               [expectation fulfill];
             }];
  [task enqueue];

  [FIRStorageTestHelpers waitForExpectation:self];
}

- (void)testSuccessfulFetch {
  XCTestExpectation *expectation = [self expectationWithDescription:@"testSuccessfulFetch"];

  self.fetcherService.testBlock = [FIRStorageTestHelpers successBlockWithMetadata:self.metadata];
  FIRStoragePath *path = [FIRStorageTestHelpers objectPath];
  FIRStorageReference *ref = [[FIRStorageReference alloc] initWithStorage:self.storage path:path];
  FIRStorageUpdateMetadataTask *task = [[FIRStorageUpdateMetadataTask alloc]
      initWithReference:ref
         fetcherService:self.fetcherService
               metadata:self.metadata
             completion:^(FIRStorageMetadata *_Nullable metadata, NSError *_Nullable error) {
               XCTAssertEqualObjects(self.metadata.bucket, metadata.bucket);
               XCTAssertEqualObjects(self.metadata.name, metadata.name);
               XCTAssertNil(error);
               [expectation fulfill];
             }];
  [task enqueue];

  [FIRStorageTestHelpers waitForExpectation:self];
}

- (void)testUnsuccessfulFetchUnauthenticated {
  XCTestExpectation *expectation =
      [self expectationWithDescription:@"testUnsuccessfulFetchUnauthenticated"];

  self.fetcherService.testBlock = [FIRStorageTestHelpers unauthenticatedBlock];
  FIRStoragePath *path = [FIRStorageTestHelpers objectPath];
  FIRStorageReference *ref = [[FIRStorageReference alloc] initWithStorage:self.storage path:path];
  FIRStorageUpdateMetadataTask *task = [[FIRStorageUpdateMetadataTask alloc]
      initWithReference:ref
         fetcherService:self.fetcherService
               metadata:self.metadata
             completion:^(FIRStorageMetadata *_Nullable metadata, NSError *_Nullable error) {
               XCTAssertNil(metadata);
               XCTAssertEqual(error.code, FIRStorageErrorCodeUnauthenticated);
               [expectation fulfill];
             }];
  [task enqueue];

  [FIRStorageTestHelpers waitForExpectation:self];
}

- (void)testUnsuccessfulFetchUnauthorized {
  XCTestExpectation *expectation =
      [self expectationWithDescription:@"testUnsuccessfulFetchUnauthorized"];

  self.fetcherService.testBlock = [FIRStorageTestHelpers unauthorizedBlock];
  FIRStoragePath *path = [FIRStorageTestHelpers objectPath];
  FIRStorageReference *ref = [[FIRStorageReference alloc] initWithStorage:self.storage path:path];
  FIRStorageUpdateMetadataTask *task = [[FIRStorageUpdateMetadataTask alloc]
      initWithReference:ref
         fetcherService:self.fetcherService
               metadata:self.metadata
             completion:^(FIRStorageMetadata *_Nullable metadata, NSError *_Nullable error) {
               XCTAssertNil(metadata);
               XCTAssertEqual(error.code, FIRStorageErrorCodeUnauthorized);
               [expectation fulfill];
             }];
  [task enqueue];

  [FIRStorageTestHelpers waitForExpectation:self];
}

- (void)testUnsuccessfulFetchObjectDoesntExist {
  XCTestExpectation *expectation =
      [self expectationWithDescription:@"testUnsuccessfulFetchObjectDoesntExist"];

  self.fetcherService.testBlock = [FIRStorageTestHelpers notFoundBlock];
  FIRStoragePath *path = [FIRStorageTestHelpers notFoundPath];
  FIRStorageReference *ref = [[FIRStorageReference alloc] initWithStorage:self.storage path:path];
  FIRStorageUpdateMetadataTask *task = [[FIRStorageUpdateMetadataTask alloc]
      initWithReference:ref
         fetcherService:self.fetcherService
               metadata:self.metadata
             completion:^(FIRStorageMetadata *_Nullable metadata, NSError *_Nullable error) {
               XCTAssertNil(metadata);
               XCTAssertEqual(error.code, FIRStorageErrorCodeObjectNotFound);
               [expectation fulfill];
             }];
  [task enqueue];

  [FIRStorageTestHelpers waitForExpectation:self];
}

- (void)testUnsuccessfulFetchBadJSON {
  XCTestExpectation *expectation =
      [self expectationWithDescription:@"testUnsuccessfulFetchBadJSON"];

  self.fetcherService.testBlock = [FIRStorageTestHelpers invalidJSONBlock];
  FIRStoragePath *path = [FIRStorageTestHelpers objectPath];
  FIRStorageReference *ref = [[FIRStorageReference alloc] initWithStorage:self.storage path:path];
  FIRStorageUpdateMetadataTask *task = [[FIRStorageUpdateMetadataTask alloc]
      initWithReference:ref
         fetcherService:self.fetcherService
               metadata:self.metadata
             completion:^(FIRStorageMetadata *_Nullable metadata, NSError *_Nullable error) {
               XCTAssertNil(metadata);
               XCTAssertEqual(error.code, FIRStorageErrorCodeUnknown);
               [expectation fulfill];
             }];
  [task enqueue];

  [FIRStorageTestHelpers waitForExpectation:self];
}

@end