From 83a6ae5ad32f91021c325b3791d90a779fb86c67 Mon Sep 17 00:00:00 2001
From: Nicky Bondarenko <nickyb@spotify.com>
Date: Mon, 17 Feb 2025 14:15:59 +0100
Subject: [PATCH] fix: add cleaning context button to demo app

---
 .../java/com/example/confidencedemoapp/MainActivity.kt    | 3 +++
 .../src/main/java/com/example/confidencedemoapp/MainVm.kt | 8 ++++++++
 2 files changed, 11 insertions(+)

diff --git a/ConfidenceDemoApp/src/main/java/com/example/confidencedemoapp/MainActivity.kt b/ConfidenceDemoApp/src/main/java/com/example/confidencedemoapp/MainActivity.kt
index 80dcf22..274381a 100644
--- a/ConfidenceDemoApp/src/main/java/com/example/confidencedemoapp/MainActivity.kt
+++ b/ConfidenceDemoApp/src/main/java/com/example/confidencedemoapp/MainActivity.kt
@@ -52,6 +52,9 @@ class MainActivity : ComponentActivity() {
                         Button(onClick = { vm.multiput() }) {
                             Text("MultiPut")
                         }
+                        Button(onClick = { vm.clear() }) {
+                            Text(text = "Clear out for re-fetching")
+                        }
                         Surface(
                             modifier = Modifier
                                 .padding(20.dp)
diff --git a/ConfidenceDemoApp/src/main/java/com/example/confidencedemoapp/MainVm.kt b/ConfidenceDemoApp/src/main/java/com/example/confidencedemoapp/MainVm.kt
index d55f712..a5ba446 100644
--- a/ConfidenceDemoApp/src/main/java/com/example/confidencedemoapp/MainVm.kt
+++ b/ConfidenceDemoApp/src/main/java/com/example/confidencedemoapp/MainVm.kt
@@ -136,6 +136,14 @@ class MainVm(app: Application) : AndroidViewModel(app) {
             refreshUi()
         }
     }
+
+    fun clear() {
+        Log.d(TAG, "clearing confidence")
+        for (key in confidence.getContext()) {
+            confidence.removeContext(key.key)
+        }
+        Log.d(TAG, "confidence context: ${confidence.getContext()}")
+    }
 }
 
 private fun <T> Evaluation<T>.toComposeColor(): Color {