From 02b4582c8d2c425b1bcf83fd0956d17425d5c7e0 Mon Sep 17 00:00:00 2001
From: Jonathan Almeida <jalmeida@mozilla.com>
Date: Sun, 19 Jan 2020 21:36:53 -0500
Subject: [PATCH] No issue: Update Migration UI with new API

---
 .../mozilla/fenix/migration/MigrationProgressActivity.kt   | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/app/src/main/java/org/mozilla/fenix/migration/MigrationProgressActivity.kt b/app/src/main/java/org/mozilla/fenix/migration/MigrationProgressActivity.kt
index 075cf0870..76819d6b2 100644
--- a/app/src/main/java/org/mozilla/fenix/migration/MigrationProgressActivity.kt
+++ b/app/src/main/java/org/mozilla/fenix/migration/MigrationProgressActivity.kt
@@ -63,7 +63,7 @@ class MigrationProgressActivity : AbstractMigrationProgressActivity() {
         }
     }
 
-    override fun onMigrationCompleted() {
+    override fun onMigrationCompleted(results: MigrationResults) {
         // Enable clicking the finish button
         migration_button.apply {
             isEnabled = true
@@ -71,9 +71,8 @@ class MigrationProgressActivity : AbstractMigrationProgressActivity() {
             setBackgroundColor(ContextCompat.getColor(context, R.color.button_text_color))
             setTextColor(ContextCompat.getColor(context, R.color.white_color))
         }
-        store.state.results?.let {
-            statusAdapter.submitList(it.toItemList())
-        }
+        // Keep the results list up-to-date.
+        statusAdapter.submitList(results.toItemList())
     }
 
     override fun onMigrationStateChanged(progress: MigrationProgress, results: MigrationResults) {