1 files changed, 117 insertions, 0 deletions
diff --git a/shadows/framework/src/main/java/org/robolectric/shadows/AssociationInfoBuilder.java b/shadows/framework/src/main/java/org/robolectric/shadows/AssociationInfoBuilder.java
new file mode 100644
index 000000000..73c36a542
--- /dev/null
+++ b/shadows/framework/src/main/java/org/robolectric/shadows/AssociationInfoBuilder.java
@@ -0,0 +1,117 @@
+package org.robolectric.shadows;
+
+import static android.os.Build.VERSION_CODES.TIRAMISU;
+
+import android.companion.AssociationInfo;
+import android.net.MacAddress;
+import org.robolectric.RuntimeEnvironment;
+import org.robolectric.util.ReflectionHelpers;
+import org.robolectric.util.ReflectionHelpers.ClassParameter;
+
+/** Builder for {@link AssociationInfo}. */
+public class AssociationInfoBuilder {
+  private int id;
+  private int userId;
+  private String packageName;
+  private String deviceMacAddress;
+  private CharSequence displayName;
+  private String deviceProfile;
+  private boolean selfManaged;
+  private boolean notifyOnDeviceNearby;
+  private long approvedMs;
+  private long lastTimeConnectedMs;
+
+  private AssociationInfoBuilder() {}
+
+  public static AssociationInfoBuilder newBuilder() {
+    return new AssociationInfoBuilder();
+  }
+
+  public AssociationInfoBuilder setId(int id) {
+    this.id = id;
+    return this;
+  }
+
+  public AssociationInfoBuilder setUserId(int userId) {
+    this.userId = userId;
+    return this;
+  }
+
+  public AssociationInfoBuilder setPackageName(String packageName) {
+    this.packageName = packageName;
+    return this;
+  }
+
+  public AssociationInfoBuilder setDeviceMacAddress(String deviceMacAddress) {
+    this.deviceMacAddress = deviceMacAddress;
+    return this;
+  }
+
+  public AssociationInfoBuilder setDisplayName(CharSequence displayName) {
+    this.displayName = displayName;
+    return this;
+  }
+
+  public AssociationInfoBuilder setDeviceProfile(String deviceProfile) {
+    this.deviceProfile = deviceProfile;
+    return this;
+  }
+
+  public AssociationInfoBuilder setSelfManaged(boolean selfManaged) {
+    this.selfManaged = selfManaged;
+    return this;
+  }
+
+  public AssociationInfoBuilder setNotifyOnDeviceNearby(boolean notifyOnDeviceNearby) {
+    this.notifyOnDeviceNearby = notifyOnDeviceNearby;
+    return this;
+  }
+
+  public AssociationInfoBuilder setApprovedMs(long approvedMs) {
+    this.approvedMs = approvedMs;
+    return this;
+  }
+
+  public AssociationInfoBuilder setLastTimeConnectedMs(long lastTimeConnectedMs) {
+    this.lastTimeConnectedMs = lastTimeConnectedMs;
+    return this;
+  }
+
+  public AssociationInfo build() {
+    try {
+      if (RuntimeEnvironment.getApiLevel() <= TIRAMISU) {
+        return ReflectionHelpers.callConstructor(
+            AssociationInfo.class,
+            ClassParameter.from(int.class, id),
+            ClassParameter.from(int.class, userId),
+            ClassParameter.from(String.class, packageName),
+            ClassParameter.from(MacAddress.class, MacAddress.fromString(deviceMacAddress)),
+            ClassParameter.from(CharSequence.class, displayName),
+            ClassParameter.from(String.class, deviceProfile),
+            ClassParameter.from(boolean.class, selfManaged),
+            ClassParameter.from(boolean.class, notifyOnDeviceNearby),
+            ClassParameter.from(boolean.class, false /*revoked*/),
+            ClassParameter.from(long.class, approvedMs),
+            ClassParameter.from(long.class, lastTimeConnectedMs));
+      } else {
+        return ReflectionHelpers.callConstructor(
+            AssociationInfo.class,
+            ClassParameter.from(int.class, id),
+            ClassParameter.from(int.class, userId),
+            ClassParameter.from(String.class, packageName),
+            ClassParameter.from(MacAddress.class, MacAddress.fromString(deviceMacAddress)),
+            ClassParameter.from(CharSequence.class, displayName),
+            ClassParameter.from(String.class, deviceProfile),
+            ClassParameter.from(Class.forName("android.companion.AssociatedDevice"), null),
+            ClassParameter.from(boolean.class, selfManaged),
+            ClassParameter.from(boolean.class, notifyOnDeviceNearby),
+            ClassParameter.from(boolean.class, false /*revoked*/),
+            ClassParameter.from(long.class, approvedMs),
+            ClassParameter.from(long.class, lastTimeConnectedMs),
+            ClassParameter.from(int.class, 0 /*systemDataSyncFlags*/));
+      }
+    } catch (ClassNotFoundException e) {
+      throw new RuntimeException(e);
+    }
+  }
+}