Compiler: Add createVariableWrite() method

Author: adazem009

include/scratchcpp/dev/compiler.h

@@ -78,6 +78,8 @@ class LIBSCRATCHCPP_EXPORT Compiler
         void createExp();
         void createExp10();
 
+        void createVariableWrite(Variable *variable);
+
         void moveToIf(std::shared_ptr<Block> substack);
         void moveToIfElse(std::shared_ptr<Block> substack1, std::shared_ptr<Block> substack2);
         void moveToRepeatLoop(std::shared_ptr<Block> substack);

src/dev/engine/compiler.cpp

@@ -256,6 +256,12 @@ void Compiler::createExp10()
     impl->builder->createExp10();
 }
 
+/*! Creates a variable write operation using the last value. */
+void Compiler::createVariableWrite(Variable *variable)
+{
+    impl->builder->createVariableWrite(variable);
+}
+
 /*! Jumps to the given if substack. */
 void Compiler::moveToIf(std::shared_ptr<Block> substack)
 {

test/dev/compiler/compiler_test.cpp

@@ -116,7 +116,7 @@ TEST_F(CompilerTest, AddVariableValue)
     compile(compiler, block);
 }
 
-TEST_F(CompilerTest, AddListValue)
+TEST_F(CompilerTest, AddListContents)
 {
     Compiler compiler(&m_engine, &m_target);
     auto block = std::make_shared<Block>("a", "");
@@ -557,6 +557,20 @@ TEST_F(CompilerTest, CreateExp10)
     compile(compiler, block);
 }
 
+TEST_F(CompilerTest, CreateVariableWrite)
+{
+    Compiler compiler(&m_engine, &m_target);
+    auto block = std::make_shared<Block>("", "");
+
+    block->setCompileFunction([](Compiler *compiler) {
+        auto var = std::make_shared<Variable>("", "");
+        EXPECT_CALL(*m_builder, createVariableWrite(var.get()));
+        compiler->createVariableWrite(var.get());
+    });
+
+    compile(compiler, block);
+}
+
 TEST_F(CompilerTest, MoveToIf)
 {
     Compiler compiler(&m_engine, &m_target);