@@ -79,7 +79,8 @@ class RequestProcessorTests: XCTestCase {
7979 let device = createSampleDevice ( )
8080 deviceDiscovery. devices = [ device]
8181
82- let processor = RequestProcessor ( deviceDiscovery: deviceDiscovery)
82+ let mockDeviceClaimManager = MockDeviceClaimManager ( )
83+ let processor = RequestProcessor ( deviceDiscovery: deviceDiscovery, deviceClaimManager: mockDeviceClaimManager)
8384 let requestData = createDeviceListRequest ( )
8485
8586 // Act
@@ -90,7 +91,7 @@ class RequestProcessorTests: XCTestCase {
9091
9192 // Decode the response to verify it
9293 let response = try USBIPMessageDecoder . decodeDeviceListResponse ( from: responseData)
93- XCTAssertEqual ( response. header. command, . replyDeviceList, " Response should be a device list reply " )
94+ XCTAssertEqual ( response. header. command, USBIPProtocol . Command . replyDeviceList, " Response should be a device list reply " )
9495 XCTAssertEqual ( response. header. status, 0 , " Status should be success (0) " )
9596 XCTAssertEqual ( response. deviceCount, 1 , " Response should contain 1 device " )
9697 XCTAssertEqual ( response. devices. count, 1 , " Response should contain 1 device " )
@@ -108,7 +109,8 @@ class RequestProcessorTests: XCTestCase {
108109 let deviceDiscovery = MockDeviceDiscovery ( )
109110 deviceDiscovery. devices = [ ]
110111
111- let processor = RequestProcessor ( deviceDiscovery: deviceDiscovery)
112+ let mockDeviceClaimManager = MockDeviceClaimManager ( )
113+ let processor = RequestProcessor ( deviceDiscovery: deviceDiscovery, deviceClaimManager: mockDeviceClaimManager)
112114 let requestData = createDeviceListRequest ( )
113115
114116 // Act
@@ -119,7 +121,7 @@ class RequestProcessorTests: XCTestCase {
119121
120122 // Decode the response to verify it
121123 let response = try USBIPMessageDecoder . decodeDeviceListResponse ( from: responseData)
122- XCTAssertEqual ( response. header. command, . replyDeviceList, " Response should be a device list reply " )
124+ XCTAssertEqual ( response. header. command, USBIPProtocol . Command . replyDeviceList, " Response should be a device list reply " )
123125 XCTAssertEqual ( response. header. status, 0 , " Status should be success (0) " )
124126 XCTAssertEqual ( response. deviceCount, 0 , " Response should contain 0 devices " )
125127 XCTAssertEqual ( response. devices. count, 0 , " Response should contain 0 devices " )
@@ -131,7 +133,8 @@ class RequestProcessorTests: XCTestCase {
131133 let device = createSampleDevice ( )
132134 deviceDiscovery. devices = [ device]
133135
134- let processor = RequestProcessor ( deviceDiscovery: deviceDiscovery)
136+ let mockDeviceClaimManager = MockDeviceClaimManager ( )
137+ let processor = RequestProcessor ( deviceDiscovery: deviceDiscovery, deviceClaimManager: mockDeviceClaimManager)
135138 let requestData = createDeviceImportRequest ( busID: " 1-1:1.0 " )
136139
137140 // Act
@@ -144,7 +147,7 @@ class RequestProcessorTests: XCTestCase {
144147
145148 // Decode the response to verify it
146149 let response = try USBIPMessageDecoder . decodeDeviceImportResponse ( from: responseData)
147- XCTAssertEqual ( response. header. command, . replyDeviceImport, " Response should be a device import reply " )
150+ XCTAssertEqual ( response. header. command, USBIPProtocol . Command . replyDeviceImport, " Response should be a device import reply " )
148151 XCTAssertEqual ( response. header. status, 0 , " Status should be success (0) " )
149152 XCTAssertEqual ( response. status, 0 , " Status should be success (0) " )
150153 XCTAssertNotNil ( response. deviceInfo, " Device info should be present " )
@@ -162,7 +165,8 @@ class RequestProcessorTests: XCTestCase {
162165 let deviceDiscovery = MockDeviceDiscovery ( )
163166 deviceDiscovery. devices = [ ] // No devices available
164167
165- let processor = RequestProcessor ( deviceDiscovery: deviceDiscovery)
168+ let mockDeviceClaimManager = MockDeviceClaimManager ( )
169+ let processor = RequestProcessor ( deviceDiscovery: deviceDiscovery, deviceClaimManager: mockDeviceClaimManager)
166170 let requestData = createDeviceImportRequest ( busID: " 1-1:1.0 " )
167171
168172 // Act
@@ -173,15 +177,16 @@ class RequestProcessorTests: XCTestCase {
173177
174178 // Decode the response to verify it
175179 let response = try USBIPMessageDecoder . decodeDeviceImportResponse ( from: responseData)
176- XCTAssertEqual ( response. header. command, . replyDeviceImport, " Response should be a device import reply " )
180+ XCTAssertEqual ( response. header. command, USBIPProtocol . Command . replyDeviceImport, " Response should be a device import reply " )
177181 XCTAssertEqual ( response. status, 1 , " Status should be error (1) " )
178182 XCTAssertNil ( response. deviceInfo, " Device info should not be present for error response " )
179183 }
180184
181185 func testProcessInvalidRequest( ) throws {
182186 // Arrange
183187 let deviceDiscovery = MockDeviceDiscovery ( )
184- let processor = RequestProcessor ( deviceDiscovery: deviceDiscovery)
188+ let mockDeviceClaimManager = MockDeviceClaimManager ( )
189+ let processor = RequestProcessor ( deviceDiscovery: deviceDiscovery, deviceClaimManager: mockDeviceClaimManager)
185190
186191 // Create an invalid request with incorrect data
187192 let invalidData = Data ( [ 0x01 , 0x02 , 0x03 , 0x04 ] )
@@ -193,7 +198,8 @@ class RequestProcessorTests: XCTestCase {
193198 func testProcessUnsupportedCommand( ) throws {
194199 // Arrange
195200 let deviceDiscovery = MockDeviceDiscovery ( )
196- let processor = RequestProcessor ( deviceDiscovery: deviceDiscovery)
201+ let mockDeviceClaimManager = MockDeviceClaimManager ( )
202+ let processor = RequestProcessor ( deviceDiscovery: deviceDiscovery, deviceClaimManager: mockDeviceClaimManager)
197203
198204 // Create a reply message (which should not be processed as a request)
199205 let header = USBIPHeader ( command: . replyDeviceList)
0 commit comments