nearby/internal/platform/implementation/apple/Tests/GNCBLEL2CAPServerTest.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/BLEv2/GNCBLEL2CAPServer.h"

#import <CoreBluetooth/CoreBluetooth.h>
#import <Foundation/Foundation.h>
#import <XCTest/XCTest.h>

#import "internal/platform/implementation/apple/Tests/GNCFakePeripheralManager.h"

@interface GNCBLEL2CAPServerTest : XCTestCase
@end

@implementation GNCBLEL2CAPServerTest

#pragma mark Tests

- (void)testPublishL2CAPChannelWhenStartListeningChannel {
  GNCFakePeripheralManager *fakePeripheralManager = [[GNCFakePeripheralManager alloc] init];
  GNCBLEL2CAPServer *l2capServer =
      [[GNCBLEL2CAPServer alloc] initWithPeripheralManager:fakePeripheralManager
                                                     queue:dispatch_get_main_queue()];
  [fakePeripheralManager simulatePeripheralManagerDidUpdateState:CBManagerStatePoweredOn];

  [l2capServer startListeningChannelWithCompletionHandler:^(NSError *error) {
    XCTAssertEqual(error, nil);
    XCTAssertEqual([l2capServer PSM], fakePeripheralManager.PSM);
  }];
}

- (void)testFailedToPublishL2CAPChannel {
  GNCFakePeripheralManager *fakePeripheralManager = [[GNCFakePeripheralManager alloc] init];
  fakePeripheralManager.didPublishL2CAPChannelError = [NSError errorWithDomain:@"fake"
                                                                          code:0
                                                                      userInfo:nil];
  GNCBLEL2CAPServer *l2capServer =
      [[GNCBLEL2CAPServer alloc] initWithPeripheralManager:fakePeripheralManager
                                                     queue:dispatch_get_main_queue()];
  [fakePeripheralManager simulatePeripheralManagerDidUpdateState:CBManagerStatePoweredOn];

  [l2capServer startListeningChannelWithCompletionHandler:^(NSError *error) {
    XCTAssertEqual(error, fakePeripheralManager.didPublishL2CAPChannelError);
    XCTAssertEqual([l2capServer PSM], 0);
  }];
}

- (void)testPoweredOffUnpublishesChannel {
  GNCFakePeripheralManager *fakePeripheralManager = [[GNCFakePeripheralManager alloc] init];
  GNCBLEL2CAPServer *l2capServer =
      [[GNCBLEL2CAPServer alloc] initWithPeripheralManager:fakePeripheralManager
                                                     queue:dispatch_get_main_queue()];
  [fakePeripheralManager simulatePeripheralManagerDidUpdateState:CBManagerStatePoweredOn];

  [l2capServer startListeningChannelWithCompletionHandler:^(NSError *error) {
    XCTAssertEqual(error, nil);
    XCTAssertEqual([l2capServer PSM], fakePeripheralManager.PSM);
  }];

  [fakePeripheralManager simulatePeripheralManagerDidUpdateState:CBManagerStatePoweredOff];

  [self waitForExpectations:@[ fakePeripheralManager.unpublishExpectation ] timeout:0.0];
}

- (void)testStartPeripheralManagerInitiallyOff {
  GNCFakePeripheralManager *fakePeripheralManager = [[GNCFakePeripheralManager alloc] init];
  GNCBLEL2CAPServer *l2capServer =
      [[GNCBLEL2CAPServer alloc] initWithPeripheralManager:fakePeripheralManager
                                                     queue:dispatch_get_main_queue()];

  [l2capServer startListeningChannelWithCompletionHandler:^(NSError *error) {
    XCTAssertEqual(error, nil);
    XCTAssertEqual([l2capServer PSM], fakePeripheralManager.PSM);
  }];

  [fakePeripheralManager simulatePeripheralManagerDidUpdateState:CBManagerStatePoweredOn];
}

- (void)testClose {
  GNCFakePeripheralManager *fakePeripheralManager = [[GNCFakePeripheralManager alloc] init];
  GNCBLEL2CAPServer *l2capServer =
      [[GNCBLEL2CAPServer alloc] initWithPeripheralManager:fakePeripheralManager
                                                     queue:dispatch_get_main_queue()];
  [fakePeripheralManager simulatePeripheralManagerDidUpdateState:CBManagerStatePoweredOn];

  [l2capServer startListeningChannelWithCompletionHandler:^(NSError *error) {
    XCTAssertEqual(error, nil);
    XCTAssertEqual([l2capServer PSM], fakePeripheralManager.PSM);
  }];

  [l2capServer close];

  XCTAssertEqual([l2capServer PSM], 0);
}

- (void)testCloseDoesNotUnpublishesChannelIfNotConnected {
  GNCFakePeripheralManager *fakePeripheralManager = [[GNCFakePeripheralManager alloc] init];
  fakePeripheralManager.unpublishExpectation.inverted = YES;
  GNCBLEL2CAPServer *l2capServer =
      [[GNCBLEL2CAPServer alloc] initWithPeripheralManager:fakePeripheralManager
                                                     queue:dispatch_get_main_queue()];

  [l2capServer close];

  [self waitForExpectations:@[ fakePeripheralManager.unpublishExpectation ] timeout:1.0];
}

@end