Internal Change.

PiperOrigin-RevId: 327533597
diff --git a/Device/Tests/EDODeviceConnectorTest.m b/Device/Tests/EDODeviceConnectorTest.m
index 6553c89..cedae75 100644
--- a/Device/Tests/EDODeviceConnectorTest.m
+++ b/Device/Tests/EDODeviceConnectorTest.m
@@ -13,13 +13,22 @@
 // limitations under the License.
 //
 
-#import <OCMock/OCMock.h>
 #import <XCTest/XCTest.h>
 
 #import "Device/Sources/EDODeviceConnector.h"
 #import "Device/Sources/EDODeviceDetector.h"
 #import "Device/Sources/EDOUSBMuxUtil.h"
 
+#import <OCMock/OCMock.h>
+
+// IWYU pragma: no_include "OCMArg.h"
+// IWYU pragma: no_include "OCMFunctions.h"
+// IWYU pragma: no_include "OCMLocation.h"
+// IWYU pragma: no_include "OCMMacroState.h"
+// IWYU pragma: no_include "OCMRecorder.h"
+// IWYU pragma: no_include "OCMStubRecorder.h"
+// IWYU pragma: no_include "OCMockObject.h"
+
 static NSString *const kFakeSerialNumber = @"fake_serial";
 
 // Exposes the internal property for test purpose.
diff --git a/DeviceForwarder/Tests/EDODeviceForwardersManagerTest.m b/DeviceForwarder/Tests/EDODeviceForwardersManagerTest.m
index 985b1c8..9838772 100644
--- a/DeviceForwarder/Tests/EDODeviceForwardersManagerTest.m
+++ b/DeviceForwarder/Tests/EDODeviceForwardersManagerTest.m
@@ -16,7 +16,6 @@
 
 #import "DeviceForwarder/Sources/EDODeviceForwardersManager.h"
 
-#import <OCMock/OCMock.h>
 #import <XCTest/XCTest.h>
 
 #import "Channel/Sources/EDOChannel.h"
@@ -27,6 +26,8 @@
 #import "Channel/Sources/EDOSocketPort.h"
 #import "Device/Sources/EDODeviceConnector.h"
 
+#import <OCMock/OCMock.h>
+
 // IWYU pragma: no_include "OCMArg.h"
 // IWYU pragma: no_include "OCMFunctions.h"
 // IWYU pragma: no_include "OCMLocation.h"
diff --git a/Service/Tests/PerfTests/EDOServicePerfTest.m b/Service/Tests/PerfTests/EDOServicePerfTest.m
index bee7361..43a6ec0 100644
--- a/Service/Tests/PerfTests/EDOServicePerfTest.m
+++ b/Service/Tests/PerfTests/EDOServicePerfTest.m
@@ -14,7 +14,6 @@
 // limitations under the License.
 //
 
-#import <OCMock/OCMock.h>
 #import <XCTest/XCTest.h>
 
 #import "Channel/Sources/EDOHostPort.h"
@@ -25,6 +24,8 @@
 #import "Service/Sources/NSObject+EDOValueObject.h"
 #import "Service/Tests/TestsBundle/EDOTestDummy.h"
 
+#import <OCMock/OCMock.h>
+
 // IWYU pragma: no_include "OCMArg.h"
 // IWYU pragma: no_include "OCMFunctions.h"
 // IWYU pragma: no_include "OCMLocation.h"
diff --git a/Service/Tests/TestsBundle/EDOTestProtocol.m b/Service/Tests/TestsBundle/EDOTestProtocol.m
index 618c046..2171848 100644
--- a/Service/Tests/TestsBundle/EDOTestProtocol.m
+++ b/Service/Tests/TestsBundle/EDOTestProtocol.m
@@ -18,6 +18,14 @@
 
 #import <OCMock/OCMock.h>
 
+// IWYU pragma: no_include "OCMArg.h"
+// IWYU pragma: no_include "OCMFunctions.h"
+// IWYU pragma: no_include "OCMLocation.h"
+// IWYU pragma: no_include "OCMMacroState.h"
+// IWYU pragma: no_include "OCMRecorder.h"
+// IWYU pragma: no_include "OCMStubRecorder.h"
+// IWYU pragma: no_include "OCMockObject.h"
+
 @implementation EDOProtocolMockTestHelper
 
 + (id<EDOTestProtocol>)createTestProtocol {
diff --git a/Service/Tests/UnitTests/EDOServiceTest.m b/Service/Tests/UnitTests/EDOServiceTest.m
index 3378c75..8d98f58 100644
--- a/Service/Tests/UnitTests/EDOServiceTest.m
+++ b/Service/Tests/UnitTests/EDOServiceTest.m
@@ -14,7 +14,6 @@
 // limitations under the License.
 //
 
-#import <OCMock/OCMock.h>
 #import <XCTest/XCTest.h>
 
 #import "Channel/Sources/EDOHostPort.h"
@@ -34,6 +33,8 @@
 #import "Service/Tests/TestsBundle/EDOTestProtocol.h"
 #import "Service/Tests/TestsBundle/EDOTestValueType.h"
 
+#import <OCMock/OCMock.h>
+
 // IWYU pragma: no_include "OCMArg.h"
 // IWYU pragma: no_include "OCMFunctions.h"
 // IWYU pragma: no_include "OCMLocation.h"
diff --git a/Service/Tests/UnitTests/EDOWeakReferenceTest.m b/Service/Tests/UnitTests/EDOWeakReferenceTest.m
index 60a15c6..f215bde 100644
--- a/Service/Tests/UnitTests/EDOWeakReferenceTest.m
+++ b/Service/Tests/UnitTests/EDOWeakReferenceTest.m
@@ -14,7 +14,6 @@
 // limitations under the License.
 //
 
-#import <OCMock/OCMock.h>
 #import <XCTest/XCTest.h>
 
 #import "Service/Sources/EDOClientService+Private.h"
@@ -32,6 +31,8 @@
 #import "Service/Tests/TestsBundle/EDOTestDummy.h"
 #import "Service/Tests/TestsBundle/EDOTestValueType.h"
 
+#import <OCMock/OCMock.h>
+
 // IWYU pragma: no_include "OCMArg.h"
 // IWYU pragma: no_include "OCMFunctions.h"
 // IWYU pragma: no_include "OCMLocation.h"