cmsdk: allow custom tiles to set the new qstile's sensitive data flag

Requires: topic:hide-qs-tiles

Change-Id: I71c85a00ae5797f1e142073b4d6a3a4c3274007b
Signed-off-by: Jorge Ruesga <jorge@ruesga.com>

2 files changed, 28 insertions, 0 deletions

diff --git a/tests/src/org/cyanogenmod/tests/customtiles/unit/CustomTileBuilderTest.java b/tests/src/org/cyanogenmod/tests/customtiles/unit/CustomTileBuilderTest.java
index 713d607..128ef77 100644
--- a/tests/src/org/cyanogenmod/tests/customtiles/unit/CustomTileBuilderTest.java
+++ b/tests/src/org/cyanogenmod/tests/customtiles/unit/CustomTileBuilderTest.java
@@ -140,6 +140,15 @@ public class CustomTileBuilderTest extends AndroidTestCase {
         assertEquals(collapsePanel, customTile.collapsePanel);
     }
 
+    @SmallTest
+    public void testCustomTileBuilderSensitiveDataSet() {
+        boolean sensitiveData = true;
+        CustomTile customTile = new CustomTile.Builder(mContext)
+                .hasSensitiveData(sensitiveData)
+                .build();
+        assertEquals(sensitiveData, customTile.sensitiveData);
+    }
+
     @MediumTest
     public void testCustomTileBuilderExpandedListStyleSet() {
         PendingIntent intent = PendingIntent.getActivity(mContext, 0,
diff --git a/tests/src/org/cyanogenmod/tests/customtiles/unit/CustomTileTest.java b/tests/src/org/cyanogenmod/tests/customtiles/unit/CustomTileTest.java
index 188153c..b724550 100644
--- a/tests/src/org/cyanogenmod/tests/customtiles/unit/CustomTileTest.java
+++ b/tests/src/org/cyanogenmod/tests/customtiles/unit/CustomTileTest.java
@@ -189,6 +189,25 @@ public class CustomTileTest extends AndroidTestCase {
         assertEquals(expectedCustomTile.collapsePanel, fromParcel.collapsePanel);
     }
 
+    @SmallTest
+    public void testCustomTileSensitiveDataUnravelFromParcel() {
+        CustomTile expectedCustomTile = new CustomTile.Builder(mContext)
+                .hasSensitiveData(true)
+                .build();
+
+        // Write to parcel
+        Parcel parcel = Parcel.obtain();
+        expectedCustomTile.writeToParcel(parcel, 0);
+
+        // Rewind
+        parcel.setDataPosition(0);
+
+        // Verify data when unraveling
+        CustomTile fromParcel = CustomTile.CREATOR.createFromParcel(parcel);
+
+        assertEquals(expectedCustomTile.sensitiveData, fromParcel.sensitiveData);
+    }
+
     @MediumTest
     public void testCustomTileExpandedListStyleUnravelFromParcel() {
         PendingIntent intent = PendingIntent.getActivity(mContext, 0,