fix unit toggle

Author: ipax77

src/dsstats.maui/dsstats.builder/dsstats.builder.tests/BuildAreaTests.cs

@@ -6,15 +6,15 @@ public sealed class BuildAreaTests
     [TestMethod]
     public void CanPlaceUnit()
     {
-        var buildArea = new BuildArea();
+        var buildArea = new BuildArea(2);
         var result = buildArea.PlaceUnit("Zergling", new(90, 75));
         Assert.IsTrue(result);
     }
 
     [TestMethod]
     public void CannotPlaceUnit()
     {
-        var buildArea = new BuildArea();
+        var buildArea = new BuildArea(1);
         var result = buildArea.PlaceUnit("Zergling", new(0, 0));
         Assert.IsFalse(result);
     }

src/dsstats.maui/dsstats.builder/dsstats.builder.tests/ScreenAreaTests.cs

@@ -31,10 +31,13 @@ public virtual List<InputEvent> GetBuildEvents(string unitName, RlPoint pos, Scr
                     if (toggleEvent != null)
                     {
                         events.AddRange(toggleEvent);
-                        var otherToggleUnit = UnitMap.FirstOrDefault(f => f.Value.Key == buildOption.Key && f.Key != unitName);
-                        if (!string.IsNullOrEmpty(otherToggleUnit.Key))
+                        var otherToggleUnits = UnitMap.Where(f => f.Value.Key == buildOption.Key && f.Key != unitName);
+                        foreach (var otherUnit in otherToggleUnits)
                         {
-                            activeUnits[otherToggleUnit.Key] = false;
+                            if (activeUnits.ContainsKey(otherUnit.Key))
+                            {
+                                activeUnits[otherUnit.Key] = false;
+                            }
                         }
                         activeUnits[unitName] = true;
                     }