From 43a7154c12c77569b0dc8893eb31c78a9ec13ffc Mon Sep 17 00:00:00 2001
From: jinyus <starboymc4@gmail.com>
Date: Thu, 14 Mar 2024 20:44:22 -0500
Subject: [PATCH] Refactor family_test.dart to include getKeys() method

---
 .../test/src/beacons/family_test.dart             | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/packages/state_beacon_core/test/src/beacons/family_test.dart b/packages/state_beacon_core/test/src/beacons/family_test.dart
index 2c4a94d..8aefcc1 100644
--- a/packages/state_beacon_core/test/src/beacons/family_test.dart
+++ b/packages/state_beacon_core/test/src/beacons/family_test.dart
@@ -143,16 +143,23 @@ void main() {
     final beacon1 = family(1);
     final beacon2 = family(2);
 
-    List<WritableBeacon<String>> getBeacons() =>
-        family.entries.value.map((e) => e.value).toList();
+    List<WritableBeacon<String>> getBeacons() {
+      return family.entries.value.map((e) => e.value).toList();
+    }
+
+    List<int> getKeys() {
+      return family.entries.value.map((e) => e.key).toList();
+    }
 
     expect(getBeacons(), [beacon1, beacon2]);
+    expect(getKeys(), [1, 2]);
 
     family.entries.subscribe((_) => ran++, synchronous: true, startNow: false);
 
     final beacon3 = family(3);
 
     expect(getBeacons(), [beacon1, beacon2, beacon3]);
+    expect(getKeys(), [1, 2, 3]);
 
     expect(ran, 1);
 
@@ -163,18 +170,21 @@ void main() {
     beacon1.dispose();
 
     expect(getBeacons(), [beacon2, beacon3]);
+    expect(getKeys(), [2, 3]);
 
     expect(ran, 2);
 
     family.remove(2);
 
     expect(getBeacons(), [beacon3]);
+    expect(getKeys(), [3]);
 
     expect(ran, 3);
 
     final beacon4 = family(4);
 
     expect(getBeacons(), [beacon3, beacon4]);
+    expect(getKeys(), [3, 4]);
 
     expect(ran, 4);
 
@@ -184,6 +194,7 @@ void main() {
     family.clear();
 
     expect(getBeacons(), isEmpty);
+    expect(getKeys(), isEmpty);
 
     expect(ran, 5);
   });