nearby/internal/platform/implementation/apple/Mediums/WiFiCommon/Tests/GNCFakeNWConnection.m

// Copyright 2025 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     https://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 "internal/platform/implementation/apple/Mediums/WiFiCommon/Tests/GNCFakeNWConnection.h"

#import <Foundation/Foundation.h>
#import <Network/Network.h>

NS_ASSUME_NONNULL_BEGIN

@implementation GNCFakeNWConnection {
  nw_connection_t _fakeNWConnection;
}

- (instancetype)initWithNWConnection:(nw_connection_t)connection {
  self = [super init];
  if (self) {
    _fakeNWConnection = connection;
  }
  return self;
}

- (nw_connection_t)createConnectionWithEndpoint:(nw_endpoint_t)endpoint
                                     parameters:(nw_parameters_t)parameters {
  // Return a non-nil, unique-ish value to represent the connection.
  _fakeNWConnection = (nw_connection_t)[[NSObject alloc] init];
  return _fakeNWConnection;
}

- (void)setQueue:(dispatch_queue_t)queue {
  // No-op for fake.
}

- (void)setStateChangedHandler:(nullable nw_connection_state_changed_handler_t)handler {
  // Capture the handler passed to the method.
  _stateChangedHandler = handler;
}

- (void)start {
  // Simulate immediate transition to ready or failed.
  if (self.stateChangedHandler) {
    nw_connection_state_t state =
        self.simulateStartFailure ? nw_connection_state_failed : nw_connection_state_ready;
    dispatch_async(dispatch_get_global_queue(QOS_CLASS_DEFAULT, 0), ^{
      self.stateChangedHandler(state, nil);
    });
  }
}

- (void)cancel {
  self.cancelCalled = YES;
  if (self.stateChangedHandler) {
    self.stateChangedHandler(nw_connection_state_cancelled, nil);
  }
}

- (void)receiveMessageWithMinLength:(uint32_t)minIncompleteLength
                          maxLength:(uint32_t)maxLength
                  completionHandler:(void (^)(dispatch_data_t _Nullable content,
                                              nw_content_context_t _Nullable context,
                                              bool isComplete, nw_error_t _Nullable error))handler {
  dispatch_async(dispatch_get_global_queue(QOS_CLASS_DEFAULT, 0), ^{
    if (self.simulateReceiveFailure) {
      // We cannot create a realistic nw_error_t, so use nil content to signal failure.
      handler(nil, nil, YES, nil);
    } else {
      handler(self.dataToReceive, nil, YES, nil);
    }
  });
}

- (void)sendData:(dispatch_data_t)content
              context:(nw_content_context_t)context
           isComplete:(BOOL)isComplete
    completionHandler:(void (^)(nw_error_t _Nullable error))handler {
  dispatch_async(dispatch_get_global_queue(QOS_CLASS_DEFAULT, 0), ^{
    if (self.simulateSendFailure) {
      // Simulate an error by timing out, as creating a realistic nw_error_t is not possible.
    } else {
      handler(nil);
    }
  });
}

- (nullable nw_connection_t)nwConnection {
  return _fakeNWConnection;
}

@end

NS_ASSUME_NONNULL_END