Add tests for new BatteryManager API.

Bug: 309792384
Test: CtsBatteryHealthTestCases
Change-Id: Idc071555682536ddd6e0b4f8b803d5327a9aaa9b
diff --git a/tests/tests/batteryhealth/src/android/os/cts/batteryhealth/BatteryHealthTest.java b/tests/tests/batteryhealth/src/android/os/cts/batteryhealth/BatteryHealthTest.java
index 411bd8a..8e59a17 100644
--- a/tests/tests/batteryhealth/src/android/os/cts/batteryhealth/BatteryHealthTest.java
+++ b/tests/tests/batteryhealth/src/android/os/cts/batteryhealth/BatteryHealthTest.java
@@ -16,6 +16,7 @@
 package android.os.cts.batteryhealth;
 
 import static android.os.Flags.stateOfHealthPublic;
+import static android.os.Flags.batteryPartStatusApi;
 import static androidx.test.platform.app.InstrumentationRegistry.getInstrumentation;
 import static com.google.common.truth.Truth.assertThat;
 import static org.junit.Assert.fail;
@@ -125,6 +126,39 @@
     }
 
     @Test
+    @ApiTest(apis = {"android.os.BatteryManager#BATTERY_PROPERTY_SERIAL_NUMBER"})
+    public void testBatterySerialNumber_dataValid() {
+        if (!batteryPartStatusApi()) {
+            return;
+        }
+        mAutomation = getInstrumentation().getUiAutomation();
+        mAutomation.adoptShellPermissionIdentity(android.Manifest.permission.BATTERY_STATS);
+        final String serialNumber = mBatteryManager.getStringProperty(BatteryManager
+                .BATTERY_PROPERTY_SERIAL_NUMBER);
+
+        if (serialNumber != null) {
+            assertThat(serialNumber.length()).isAtLeast(6);
+        }
+        mAutomation.dropShellPermissionIdentity();
+    }
+
+    @Test
+    @ApiTest(apis = {"android.os.BatteryManager#BATTERY_PROPERTY_PART_STATUS"})
+    public void testBatteryPartStatus_dataInRange() {
+        if (!batteryPartStatusApi()) {
+            return;
+        }
+        mAutomation = getInstrumentation().getUiAutomation();
+        mAutomation.adoptShellPermissionIdentity(android.Manifest.permission.BATTERY_STATS);
+        final int partStatus = mBatteryManager.getIntProperty(BatteryManager
+                .BATTERY_PROPERTY_PART_STATUS);
+
+        assertThat(partStatus).isAtLeast(BatteryManager.PART_STATUS_UNSUPPORTED);
+        assertThat(partStatus).isAtLeast(BatteryManager.PART_STATUS_REPLACED + 1);
+        mAutomation.dropShellPermissionIdentity();
+    }
+
+    @Test
     @ApiTest(apis = {"android.os.BatteryManager#EXTRA_CYCLE_COUNT"})
     public void testBatteryCycleCount_dataInRange() {
         final Context context = InstrumentationRegistry.getContext();