Add code property to Script

Author: adazem009

include/scratchcpp/script.h

@@ -14,6 +14,9 @@ namespace libscratchcpp
 class Target;
 class Block;
 class IEngine;
+#ifdef USE_LLVM
+class ExecutableCode;
+#endif
 class Value;
 class Thread;
 class Variable;
@@ -34,6 +37,11 @@ class LIBSCRATCHCPP_EXPORT Script
         const std::vector<unsigned int> &bytecodeVector() const;
         void setBytecode(const std::vector<unsigned int> &code);
 
+#ifdef USE_LLVM
+        ExecutableCode *code() const;
+        void setCode(std::shared_ptr<ExecutableCode> code);
+#endif
+
         void setHatPredicateBytecode(const std::vector<unsigned int> &code);
         bool runHatPredicate(Target *target);
 

src/engine/script.cpp

@@ -50,6 +50,20 @@ void Script::setBytecode(const std::vector<unsigned int> &code)
     impl->bytecode = impl->bytecodeVector.data();
 }
 
+#ifdef USE_LLVM
+/*! Returns the executable code of the script. */
+ExecutableCode *Script::code() const
+{
+    return impl->code.get();
+}
+
+/*! Sets the executable code of the script. */
+void Script::setCode(std::shared_ptr<ExecutableCode> code)
+{
+    impl->code = code;
+}
+#endif // USE_LLVM
+
 /*! Sets the edge-activated hat predicate bytecode. */
 void Script::setHatPredicateBytecode(const std::vector<unsigned int> &code)
 {

src/engine/script_p.h

@@ -12,6 +12,7 @@ namespace libscratchcpp
 class Target;
 class Block;
 class IEngine;
+class ExecutableCode;
 class Variable;
 class List;
 
@@ -24,6 +25,10 @@ struct ScriptPrivate
         std::vector<unsigned int> bytecodeVector;
         std::vector<unsigned int> hatPredicateBytecodeVector;
 
+#ifdef USE_LLVM
+        std::shared_ptr<ExecutableCode> code;
+#endif
+
         Target *target = nullptr;
         std::shared_ptr<Block> topBlock;
         IEngine *engine = nullptr;

test/mocks/executablecodemock.h

@@ -0,0 +1,21 @@
+#pragma once
+
+#include <scratchcpp/dev/executablecode.h>
+#include <gmock/gmock.h>
+
+using namespace libscratchcpp;
+
+class ExecutableCodeMock : public ExecutableCode
+{
+    public:
+        MOCK_METHOD(void, run, (ExecutionContext *), (override));
+        MOCK_METHOD(void, kill, (ExecutionContext *), (override));
+        MOCK_METHOD(void, reset, (ExecutionContext *), (override));
+
+        MOCK_METHOD(bool, isFinished, (ExecutionContext *), (const, override));
+
+        MOCK_METHOD(void, promise, (), (override));
+        MOCK_METHOD(void, resolvePromise, (), (override));
+
+        MOCK_METHOD(std::shared_ptr<ExecutionContext>, createExecutionContext, (Target *), (const, override));
+};

test/script/script_test.cpp

@@ -7,6 +7,7 @@
 #include <scratchcpp/variable.h>
 #include <scratchcpp/list.h>
 #include <enginemock.h>
+#include <executablecodemock.h>
 
 #include "../common.h"
 
@@ -31,6 +32,17 @@ TEST_F(ScriptTest, Constructors)
     ASSERT_EQ(script.topBlock(), block);
 }
 
+#ifdef USE_LLVM
+TEST_F(ScriptTest, Code)
+{
+    Script script(nullptr, nullptr, nullptr);
+    ASSERT_EQ(script.code(), nullptr);
+
+    auto code = std::make_shared<ExecutableCodeMock>();
+    script.setCode(code);
+    ASSERT_EQ(script.code(), code.get());
+}
+#else
 TEST_F(ScriptTest, Bytecode)
 {
     Script script(nullptr, nullptr, nullptr);
@@ -248,3 +260,4 @@ TEST_F(ScriptTest, Start)
     EXPECT_CALL(m_engine, deinitClone(clone));
     clone->deleteClone();
 }
+#endif