fbbf8cf16b
fbshipit-source-id: b6fc29740c6875d2e78953b8a7123890a67930f2 Co-authored-by: Sebastian McKenzie <sebmck@fb.com> Co-authored-by: John Knox <jknox@fb.com> Co-authored-by: Emil Sjölander <emilsj@fb.com> Co-authored-by: Pritesh Nandgaonkar <prit91@fb.com>
48 lines
1.0 KiB
Objective-C
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 "SonarConnectionMock.h"
|
|
|
|
@implementation SonarConnectionMock
|
|
|
|
- (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
|