diff --git a/libs/contextualcards.aar b/libs/contextualcards.aar index a44a26c38b1..5e855fa0dec 100755 Binary files a/libs/contextualcards.aar and b/libs/contextualcards.aar differ diff --git a/src/com/android/settings/homepage/contextualcards/SettingsContextualCardProvider.java b/src/com/android/settings/homepage/contextualcards/SettingsContextualCardProvider.java index 44705c59522..2dede95e758 100644 --- a/src/com/android/settings/homepage/contextualcards/SettingsContextualCardProvider.java +++ b/src/com/android/settings/homepage/contextualcards/SettingsContextualCardProvider.java @@ -21,14 +21,12 @@ import static android.provider.SettingsSlicesContract.KEY_WIFI; import android.annotation.Nullable; import com.android.settings.homepage.deviceinfo.DataUsageSlice; +import com.android.settings.intelligence.ContextualCardProto.ContextualCard; +import com.android.settings.intelligence.ContextualCardProto.ContextualCardList; import com.android.settings.wifi.WifiSlice; -import com.google.android.settings.intelligence.libs.contextualcards.ContextualCard; import com.google.android.settings.intelligence.libs.contextualcards.ContextualCardProvider; -import java.util.ArrayList; -import java.util.List; - /** Provides dynamic card for SettingsIntelligence. */ public class SettingsContextualCardProvider extends ContextualCardProvider { @@ -36,21 +34,22 @@ public class SettingsContextualCardProvider extends ContextualCardProvider { @Override @Nullable - public List getContextualCards() { - final List cards = new ArrayList<>(); + public ContextualCardList getContextualCards() { final ContextualCard wifiCard = - new ContextualCard.Builder() + ContextualCard.newBuilder() .setSliceUri(WifiSlice.WIFI_URI.toString()) - .setName(KEY_WIFI) + .setCardName(KEY_WIFI) .build(); final ContextualCard dataUsageCard = - new ContextualCard.Builder() + ContextualCard.newBuilder() .setSliceUri(DataUsageSlice.DATA_USAGE_CARD_URI.toString()) - .setName(DataUsageSlice.PATH_DATA_USAGE_CARD) + .setCardName(DataUsageSlice.PATH_DATA_USAGE_CARD) .build(); + final ContextualCardList cards = ContextualCardList.newBuilder() + .addCard(wifiCard) + .addCard(dataUsageCard) + .build(); - cards.add(wifiCard); - cards.add(dataUsageCard); return cards; } } diff --git a/tests/robotests/src/com/android/settings/homepage/contextualcards/SettingsContextualCardProviderTest.java b/tests/robotests/src/com/android/settings/homepage/contextualcards/SettingsContextualCardProviderTest.java index 80d94977607..a80b963c594 100644 --- a/tests/robotests/src/com/android/settings/homepage/contextualcards/SettingsContextualCardProviderTest.java +++ b/tests/robotests/src/com/android/settings/homepage/contextualcards/SettingsContextualCardProviderTest.java @@ -24,9 +24,9 @@ import android.content.Context; import android.net.Uri; import android.os.Bundle; +import com.android.settings.intelligence.ContextualCardProto.ContextualCardList; import com.android.settings.testutils.SettingsRobolectricTestRunner; -import com.google.android.settings.intelligence.libs.contextualcards.ContextualCard; import com.google.android.settings.intelligence.libs.contextualcards.ContextualCardProvider; import org.junit.Before; @@ -34,8 +34,6 @@ import org.junit.Test; import org.junit.runner.RunWith; import org.robolectric.RuntimeEnvironment; -import java.util.ArrayList; - @RunWith(SettingsRobolectricTestRunner.class) public class SettingsContextualCardProviderTest { @@ -54,11 +52,12 @@ public class SettingsContextualCardProviderTest { } @Test - public void contentProviderCall_returnCorrectSize() { + public void contentProviderCall_returnCorrectSize() throws Exception { final Bundle returnValue = mResolver.call(mUri, ContextualCardProvider.METHOD_GET_CARD_LIST, "", null); - final ArrayList cards = - returnValue.getParcelableArrayList(ContextualCardProvider.BUNDLE_CARD_LIST); - assertThat(cards.size()).isEqualTo(2); + final ContextualCardList cards = + ContextualCardList.parseFrom( + returnValue.getByteArray(ContextualCardProvider.BUNDLE_CARD_LIST)); + assertThat(cards.getCardCount()).isEqualTo(2); } } \ No newline at end of file