From 985f7728a52865c16fc613bca5cf16934ede09af Mon Sep 17 00:00:00 2001
From: ModZero <modzero@modzero.xyz>
Date: Thu, 8 Feb 2024 18:58:20 +0100
Subject: [PATCH] Extra test for a case that worried me.

---
 ServerTests/CalculatorTests.cs | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/ServerTests/CalculatorTests.cs b/ServerTests/CalculatorTests.cs
index a972c50..8773b95 100644
--- a/ServerTests/CalculatorTests.cs
+++ b/ServerTests/CalculatorTests.cs
@@ -55,4 +55,18 @@ public class CalculatorTests
         calculator = calculator.EqualsPressed();
         Assert.Equal("12", calculator.Display);
     }
+
+    [Fact]
+    public void EqualsAfterOperator()
+    {
+        var calculator = new Calculator()
+            .DigitPressed("3");
+        Assert.Equal("3", calculator.Display);
+        calculator = calculator.OperatorPressed(CalculatorOperator.Addition);
+        Assert.Equal("3", calculator.Display);
+        calculator = calculator.OperatorPressed(CalculatorOperator.Multiplication);
+        Assert.Equal("3", calculator.Display);
+        calculator = calculator.EqualsPressed();
+        Assert.Equal("9", calculator.Display);
+    }
 }
\ No newline at end of file