Separate kernel and partial wakelock parsing

In some bugreports kernel wakelock section is missing. This prevents
parsing of partial wakelock as well. So this fix is to separate kernel
and partial wakelock parsing

Change-Id: I196dcd9438976cb5d3fcd3b0a81839903aebac15

2 files changed, 39 insertions, 11 deletions

diff --git a/tests/src/com/android/loganalysis/parser/BugreportParserTest.java b/tests/src/com/android/loganalysis/parser/BugreportParserTest.java
index ea7537f..b36e834 100644
--- a/tests/src/com/android/loganalysis/parser/BugreportParserTest.java
+++ b/tests/src/com/android/loganalysis/parser/BugreportParserTest.java
@@ -124,7 +124,7 @@ public class BugreportParserTest extends TestCase {
                 "          +45s702ms (2) 001 80080000 volt=4187",
                 "          +1m15s525ms (2) 001 80080000 temp=299 volt=4155",
                 "Statistics since last charged:",
-                "  Time on battery: 1h 5m 2s 4ms (9%) realtime, 1h 5m 2s 4ms (9%) uptime", 
+                "  Time on battery: 1h 5m 2s 4ms (9%) realtime, 1h 5m 2s 4ms (9%) uptime",
                 " Time on battery screen off: 1h 4m 5s 8ms (9%) realtime, 1h 4m 5s 8ms (9%) uptime",
                 " All kernel wake locks:",
                 " Kernel Wake lock PowerManagerService.WakeLocks: 5m 10s 6ms (2 times) realtime",
@@ -133,8 +133,9 @@ public class BugreportParserTest extends TestCase {
                 "  All partial wake locks:",
                 "  Wake lock 1001 ProxyController: 1h 4m 47s 565ms (4 times) realtime",
                 "  Wake lock 1013 AudioMix: 1s 979ms (3 times) realtime",
+                "",
                 "  All wakeup reasons:",
-                "  Wakeup reason 2:bcmsdh_sdmmc:2:qcom,smd:2:msmgio: 1m 5s 4ms (2 times) realtime", 
+                "  Wakeup reason 2:bcmsdh_sdmmc:2:qcom,smd:2:msmgio: 1m 5s 4ms (2 times) realtime",
                 "  Wakeup reason 2:qcom,smd-rpm:2:fc4c.qcom,spmi: 7m 1s 914ms (7 times) realtime",
                 "");
 
diff --git a/tests/src/com/android/loganalysis/parser/WakelockParserTest.java b/tests/src/com/android/loganalysis/parser/WakelockParserTest.java
index bed831b..735362f 100644
--- a/tests/src/com/android/loganalysis/parser/WakelockParserTest.java
+++ b/tests/src/com/android/loganalysis/parser/WakelockParserTest.java
@@ -31,7 +31,7 @@ public class WakelockParserTest extends TestCase {
     /**
      * Test that normal input is parsed.
      */
-    public void testWakelockParser() {
+    public void testKernelWakelockParser() {
         List<String> inputBlock = Arrays.asList(
                " All kernel wake locks:",
                " Kernel Wake lock PowerManagerService.WakeLocks: 1h 3m 50s 5ms (8 times) realtime",
@@ -40,14 +40,8 @@ public class WakelockParserTest extends TestCase {
                " Kernel Wake lock wlan_rx_wake: 3m 19s 887ms (225 times) realtime",
                " Kernel Wake lock wlan_tx_wake: 2m 19s 887ms (225 times) realtime",
                " Kernel Wake lock tx_wake: 1m 19s 887ms (225 times) realtime",
-               " ",
-               " All partial wake locks:",
-               " Wake lock u0a7 NlpWakeLock: 8m 13s 203ms (1479 times) realtime",
-               " Wake lock u0a7 NlpCollectorWakeLock: 6m 29s 18ms (238 times) realtime",
-               " Wake lock u0a7 GCM_CONN_ALARM: 6m 8s 587ms (239 times) realtime",
-               " Wake lock 1000 *alarm*: 5m 11s 316ms (1469 times) realtime",
-               " Wake lock u10 xxx: 4m 11s 316ms (1469 times) realtime",
-               " Wake lock u30 cst: 2m 11s 316ms (1469 times) realtime");
+               " "
+              );
 
         WakelockItem wakelock = new WakelockParser().parse(inputBlock);
 
@@ -59,6 +53,20 @@ public class WakelockParserTest extends TestCase {
                 get(0).getHeldTime());
         assertEquals(2399, wakelock.getWakeLocks(WakeLockCategory.KERNEL_WAKELOCK).
                 get(0).getLockedCount());
+    }
+
+    public void testPartialWakelockParser() {
+        List<String> inputBlock = Arrays.asList(
+                " All partial wake locks:",
+                " Wake lock u0a7 NlpWakeLock: 8m 13s 203ms (1479 times) realtime",
+                " Wake lock u0a7 NlpCollectorWakeLock: 6m 29s 18ms (238 times) realtime",
+                " Wake lock u0a7 GCM_CONN_ALARM: 6m 8s 587ms (239 times) realtime",
+                " Wake lock 1000 *alarm*: 5m 11s 316ms (1469 times) realtime",
+                " Wake lock u10 xxx: 4m 11s 316ms (1469 times) realtime",
+                " Wake lock u30 cst: 2m 11s 316ms (1469 times) realtime",
+                "");
+
+        WakelockItem wakelock = new WakelockParser().parse(inputBlock);
 
         assertEquals(WakelockParser.TOP_WAKELOCK_COUNT,
                 wakelock.getWakeLocks(WakeLockCategory.PARTIAL_WAKELOCK).size());
@@ -71,5 +79,24 @@ public class WakelockParserTest extends TestCase {
         assertEquals(1479, wakelock.getWakeLocks(WakeLockCategory.PARTIAL_WAKELOCK).
                 get(0).getLockedCount());
     }
+
+    public void testInvalidInputWakelockParser() {
+        List<String> inputBlock = Arrays.asList(
+             " lock PowerManagerService.WakeLocks: 1h 3m 50s 5ms (8 times) realtime",
+             " lock event0-2656 : 3m 49s 268ms (2399 times) realtime",
+             " lock wlan_wd_wake: 3m 34s 639ms (1751 times) realtime",
+             " lock wlan_rx_wake: 3m 19s 887ms (225 times) realtime",
+             " wlan_tx_wake: 2m 19s 887ms (225 times) realtime",
+             " tx_wake: 1m 19s 887ms (225 times) realtime",
+             " "
+            );
+
+        WakelockItem wakelock = new WakelockParser().parse(inputBlock);
+
+        assertEquals(0,
+                wakelock.getWakeLocks(WakeLockCategory.KERNEL_WAKELOCK).size());
+        assertEquals(0,
+                wakelock.getWakeLocks(WakeLockCategory.PARTIAL_WAKELOCK).size());
+    }
 }