c4c5d453c2
Reviewed By: scottrice Differential Revision: D17673481 fbshipit-source-id: 1794e0a6e92e6894b5634b24c6e7910847616261
47 lines
1.0 KiB
Objective-C
47 lines
1.0 KiB
Objective-C
/*
|
|
* Copyright (c) Facebook, Inc. and its affiliates.
|
|
*
|
|
* 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
|