Test update OCMock

third_party/ocmock/OCMock/NSNotificationCenter+OCMAdditions.m

-//---------------------------------------------------------------------------------------
-//  $Id$
-//  Copyright (c) 2009 by Mulle Kybernetik. See License file for details.
-//---------------------------------------------------------------------------------------
+/*
+ *  Copyright (c) 2009-2015 Erik Doernenburg and contributors
+ *
+ *  Licensed under the Apache License, Version 2.0 (the "License"); you may
+ *  not use these files 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 "NSNotificationCenter+OCMAdditions.h"
 #import "OCObserverMockObject.h"
 
 
 @implementation NSNotificationCenter(OCMAdditions)
 
-- (void)addMockObserver:(OCMockObserver *)notificationObserver name:(NSString *)notificationName object:(id)notificationSender
-{
-        [self addObserver:notificationObserver selector:@selector(handleNotification:) name:notificationName object:notificationSender];
+- (void)addMockObserver:(OCObserverMockObject*)notificationObserver
+                   name:(NSString*)notificationName
+                 object:(id)notificationSender {
+  [notificationObserver autoRemoveFromCenter:self];
+  [self addObserver:notificationObserver
+           selector:@selector(handleNotification:)
+               name:notificationName
+             object:notificationSender];
 }
 
 @end