Files
flipper/iOS/FlipperKitTestUtils/FlipperConnectionMock.m
Pritesh Nandgaonkar f7aef49d8a Rename SonarConnectionMock
Summary: Renames SonarConnectionMock

Reviewed By: jknoxville

Differential Revision: D10026707

fbshipit-source-id: eb1be0ad5fa53ab1fc6ca0c7e8482bb61521daeb
2018-09-25 13:08:13 -07:00

48 lines
1.0 KiB
Objective-C

/*
* Copyright (c) 2018-present, Facebook, Inc.
*
* This source code is licensed under the MIT license found in the LICENSE
* file in the root directory of this source tree.
*
*/
#import "FlipperConnectionMock.h"
@implementation FlipperConnectionMock
- (instancetype)init
{
if (self = [super init]) {
_connected = YES;
_receivers = @{};
_sent = @{};
}
return self;
}
- (void)send:(NSString *)method withParams:(NSDictionary *)params
{
if (_connected) {
NSMutableDictionary *newSent = [NSMutableDictionary new];
[newSent addEntriesFromDictionary:_sent];
if (newSent[method]) {
newSent[method] = [_sent[method] arrayByAddingObject:params];
} else {
newSent[method] = @[params];
}
_sent = newSent;
}
}
- (void)receive:(NSString *)method withBlock:(SonarReceiver)receiver
{
if (_connected) {
NSMutableDictionary *newReceivers = [NSMutableDictionary new];
[newReceivers addEntriesFromDictionary:_receivers];
newReceivers[method] = receiver;
_receivers = newReceivers;
}
}
@end