OCMock издевается над UIImagePickerController

Я пытаюсь издеваться над UIImagePickerController для тестирования метода из ViewController (написано на Swift):

var imagePicker: UIImagePickerController!

...

func choosePhoto() {
    imagePicker = UIImagePickerController()
    imagePicker.delegate = self
    imagePicker.sourceType = .PhotoLibrary
    self.presentViewController(imagePicker, animated: true, completion: nil)
}

и тестовый класс (написанный на Objective-C):

Интерфейс:

@property (nonatomic, strong) ViewController *viewController;

Реализация:

- (void)setUp {
    [super setUp];

    UIStoryboard *storyboard = [UIStoryboard storyboardWithName:@"Main" bundle:nil];
    UINavigationController *navigationController = [storyboard instantiateInitialViewController];
    self.viewController = (ViewController *)[navigationController visibleViewController];
    self.viewController.view;
}

....

- (void)testPicker {
    id mockPicker = [OCMockObject mockForClass:[UIImagePickerController class]];
    self.viewController.imagePicker = mockPicker;

    [[mockPicker expect] setSourceType:UIImagePickerControllerSourceTypePhotoLibrary];
    [[(id)self.viewController expect] presentViewController:mockPicker animated:YES completion:nil];

    [self.viewController choosePhoto];

    [mockPicker verify];
}

Тест не проходит, потому что:

OCMockObject(UIImagePickerController): ожидаемый метод не был вызван: setSourceType:0

а также

не удалось: поймано «NSInvalidArgumentException», «-[ожидание MyApp.ViewController]: нераспознанный селектор отправлен в экземпляр 0x....

Кто-нибудь может мне с этим помочь?

Большое спасибо.