chromium/ios/testing/ocmock_complex_type_helper.h

// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef IOS_TESTING_OCMOCK_COMPLEX_TYPE_HELPER_H_
#define IOS_TESTING_OCMOCK_COMPLEX_TYPE_HELPER_H_

#import "third_party/ocmock/OCMock/OCMock.h"

// OCMock cannot check scalar arguments to method as it requires NSObjects to
// work its magic. This class tries to alleviate this restriction in a crude
// way. For an example of use, see the associated unittest class.
@interface OCMockComplexTypeHelper : NSProxy
// Init OCMockComplexTypeHelper with a represented object (retained).
- (instancetype)initWithRepresentedObject:(OCMockObject*)object;
// Registers a block to be called when a selector is called.
- (void)onSelector:(SEL)selector callBlockExpectation:(id)block;
// Returns the block for the given selector. Intended for use by subclasses.
- (id)blockForSelector:(SEL)selector;
@end

#endif  // IOS_TESTING_OCMOCK_COMPLEX_TYPE_HELPER_H_