ConfigControllerTests.m

/**
 * Copyright (c) 2015-present, Parse, LLC.
 * All rights reserved.
 *
 * This source code is licensed under the BSD-style license found in the
 * LICENSE file in the root directory of this source tree. An additional grant
 * of patent rights can be found in the PATENTS file in the same directory.
 */

#import <OCMock/OCMock.h>

@import Bolts.BFTask;

#import "OCMock+Parse.h"
#import "PFCommandResult.h"
#import "PFCommandRunning.h"
#import "PFConfig.h"
#import "PFConfigController.h"
#import "PFTestCase.h"
#import "PFPersistenceController.h"
#import "PFPersistenceGroup.h"

@protocol ConfigControllerDataSource <PFCommandRunnerProvider, PFPersistenceControllerProvider>

@end

@interface ConfigControllerTests : PFTestCase

@end

@implementation ConfigControllerTests

///--------------------------------------
#pragma mark - Helpers
///--------------------------------------

- (PFPersistenceController *)mockedPersistenceController {
    id controller = PFStrictClassMock([PFPersistenceController class]);
    id group = PFProtocolMock(@protocol(PFPersistenceGroup));
    OCMStub([controller getPersistenceGroupAsync]).andReturn([BFTask taskWithResult:group]);
    return controller;
}

///--------------------------------------
#pragma mark - Tests
///--------------------------------------

- (void)testConstructor {
    id dataSource = PFStrictProtocolMock(@protocol(PFCommandRunnerProvider));

    PFConfigController *controller = [[PFConfigController alloc] initWithDataSource:dataSource];
    XCTAssertNotNil(controller);
    XCTAssertEqual(dataSource, (id)controller.dataSource);
}

- (void)testCurrentConfigController {
    id dataSource = PFStrictProtocolMock(@protocol(PFCommandRunnerProvider));

    PFConfigController *configController = [[PFConfigController alloc] initWithDataSource:dataSource];

    XCTAssertNotNil([configController currentConfigController]);
}

- (void)testFetch {
    id commandRunner = PFStrictProtocolMock(@protocol(PFCommandRunning));
    [commandRunner mockCommandResult:@{ @"params" : @{@"testKey" : @"testValue"} }
              forCommandsPassingTest:^BOOL(id obj) {
                  return YES;
              }];
    id dataSource = PFStrictProtocolMock(@protocol(ConfigControllerDataSource));
    OCMStub([dataSource commandRunner]).andReturn(commandRunner);
    OCMStub([dataSource persistenceController]).andReturn([self mockedPersistenceController]);

    PFConfigController *configController = [[PFConfigController alloc] initWithDataSource:dataSource];

    XCTestExpectation *expectation = [self currentSelectorTestExpectation];

    [[configController fetchConfigAsyncWithSessionToken:@"token"] continueWithBlock:^id(BFTask *task) {
        XCTAssertNotNil(task.result);

        XCTAssertTrue([task.result isKindOfClass:[PFConfig class]]);
        XCTAssertEqualObjects(task.result[@"testKey"], @"testValue");

        [expectation fulfill];

        return nil;
    }];

    [self waitForTestExpectations];
}

@end