0675dd924d
Reviewed By: passy Differential Revision: D17863711 fbshipit-source-id: 259dc77826fb803ff1b88c88529d7f679d3b74d8
48 lines
1.0 KiB
Objective-C
48 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
|