From 724d12b762df0e82bf8d530de5b8f773f427c2a4 Mon Sep 17 00:00:00 2001
From: Pepe Rivera <joriv@microsoft.com>
Date: Thu, 6 Feb 2020 15:15:29 -0800
Subject: [PATCH] Update mocks (#967)

---
 src/GraphingImpl/Mocks/Graph.h      |  8 ++++++
 src/GraphingImpl/Mocks/MathSolver.h | 44 ++++++++++++++++++++++++++++-
 2 files changed, 51 insertions(+), 1 deletion(-)

diff --git a/src/GraphingImpl/Mocks/Graph.h b/src/GraphingImpl/Mocks/Graph.h
index e28f5e74..ae10c35f 100644
--- a/src/GraphingImpl/Mocks/Graph.h
+++ b/src/GraphingImpl/Mocks/Graph.h
@@ -18,6 +18,14 @@ namespace MockGraphingImpl
         }
         virtual std::optional<std::vector<std::shared_ptr<Graphing::IEquation>>> TryInitialize(const Graphing::IExpression* graphingExp = nullptr)
         {
+            if (graphingExp != nullptr)
+            {
+                std::vector<std::shared_ptr<Graphing::IEquation>> equations;
+                equations.push_back(nullptr);
+                m_variables.push_back(std::make_shared<MockVariable>(MockVariable{}));
+                return std::optional<std::vector<std::shared_ptr<Graphing::IEquation>>>(equations);
+            }
+
             return std::nullopt;
         }
 
diff --git a/src/GraphingImpl/Mocks/MathSolver.h b/src/GraphingImpl/Mocks/MathSolver.h
index 165959b9..7f9a0fca 100644
--- a/src/GraphingImpl/Mocks/MathSolver.h
+++ b/src/GraphingImpl/Mocks/MathSolver.h
@@ -48,6 +48,43 @@ namespace MockGraphingImpl
         }
     };
 
+    class MockExpression : public Graphing::IExpression
+    {
+    public:
+        MockExpression()
+        {
+        }
+
+        unsigned int GetExpressionID() const override
+        {
+            return 0;
+        }
+        bool IsEmptySet() const
+        {
+            return false;
+        }
+    };
+
+    class MockVariable : public Graphing::IVariable
+    {
+    public:
+        MockVariable()
+        {
+        }
+
+        int GetVariableID() const override
+        {
+            return 0;
+        }
+        const std::wstring& GetVariableName() override
+        {
+            return varName;
+        }
+
+    private:
+        const std::wstring& varName = L"m";
+    };
+
     class MathSolver : public Graphing::IMathSolver
     {
     public:
@@ -72,7 +109,12 @@ namespace MockGraphingImpl
 
         std::unique_ptr<Graphing::IExpression> ParseInput(const std::wstring& input) override
         {
-            return nullptr;
+            if (input.empty())
+            {
+                return nullptr;
+            }
+
+            return std::make_unique<MockExpression>(MockExpression{});
         }
 
         std::shared_ptr<Graphing::IGraph> CreateGrapher(const Graphing::IExpression* expression) override;