LLVMExecutableCode: Implement promises

Author: adazem009

src/dev/engine/internal/llvm/llvmexecutablecode.cpp

@@ -4,6 +4,7 @@
 #include <scratchcpp/value_functions.h>
 #include <scratchcpp/stage.h>
 #include <scratchcpp/iengine.h>
+#include <scratchcpp/dev/promise.h>
 #include <llvm/Support/Error.h>
 #include <iostream>
 
@@ -46,6 +47,15 @@ void LLVMExecutableCode::run(ExecutionContext *context)
     if (ctx->finished())
         return;
 
+    auto promise = ctx->promise();
+
+    if (promise) {
+        if (promise->isResolved())
+            ctx->setPromise(nullptr);
+        else
+            return;
+    }
+
     if (ctx->coroutineHandle()) {
         bool done = m_resumeFunction(ctx->coroutineHandle());
 
@@ -69,13 +79,15 @@ void LLVMExecutableCode::kill(ExecutionContext *context)
     LLVMExecutionContext *ctx = getContext(context);
     ctx->setCoroutineHandle(nullptr);
     ctx->setFinished(true);
+    ctx->setPromise(nullptr);
 }
 
 void LLVMExecutableCode::reset(ExecutionContext *context)
 {
     LLVMExecutionContext *ctx = getContext(context);
     ctx->setCoroutineHandle(nullptr);
     ctx->setFinished(false);
+    ctx->setPromise(nullptr);
 }
 
 bool LLVMExecutableCode::isFinished(ExecutionContext *context) const

test/dev/llvm/llvmexecutablecode_test.cpp

@@ -2,6 +2,7 @@
 #include <scratchcpp/value_functions.h>
 #include <scratchcpp/variable.h>
 #include <scratchcpp/list.h>
+#include <scratchcpp/dev/promise.h>
 #include <dev/engine/internal/llvm/llvmexecutablecode.h>
 #include <dev/engine/internal/llvm/llvmexecutioncontext.h>
 #include <llvm/Support/TargetSelect.h>
@@ -144,3 +145,94 @@ TEST_F(LLVMExecutableCodeTest, MainFunction)
     ASSERT_TRUE(code.isFinished(anotherCtx.get()));
     ASSERT_FALSE(code.isFinished(ctx.get()));
 }
+
+TEST_F(LLVMExecutableCodeTest, Promise)
+{
+    auto f = beginMainFunction();
+    addTestFunction(f);
+    endFunction(nullPointer());
+
+    beginResumeFunction();
+    endFunction(m_builder->getInt1(true));
+
+    LLVMExecutableCode code(std::move(m_module));
+    auto ctx = code.createExecutionContext(&m_target);
+    ASSERT_FALSE(code.isFinished(ctx.get()));
+
+    // run()
+    auto promise = std::make_shared<Promise>();
+    ctx->setPromise(promise);
+    EXPECT_CALL(m_mock, f).Times(0);
+
+    for (int i = 0; i < 10; i++) {
+        code.run(ctx.get());
+        ASSERT_FALSE(code.isFinished(ctx.get()));
+    }
+
+    promise->resolve();
+
+    EXPECT_CALL(m_mock, f);
+    code.run(ctx.get());
+    ASSERT_TRUE(code.isFinished(ctx.get()));
+    ASSERT_EQ(ctx->promise(), nullptr);
+    code.reset(ctx.get());
+
+    // kill()
+    promise = std::make_shared<Promise>();
+    ctx->setPromise(promise);
+    EXPECT_CALL(m_mock, f).Times(0);
+
+    for (int i = 0; i < 10; i++) {
+        code.run(ctx.get());
+        ASSERT_FALSE(code.isFinished(ctx.get()));
+    }
+
+    code.kill(ctx.get());
+    ASSERT_TRUE(code.isFinished(ctx.get()));
+    ASSERT_EQ(ctx->promise(), nullptr);
+    code.reset(ctx.get());
+
+    // reset()
+    promise = std::make_shared<Promise>();
+    ctx->setPromise(promise);
+    EXPECT_CALL(m_mock, f).Times(0);
+
+    for (int i = 0; i < 10; i++) {
+        code.run(ctx.get());
+        ASSERT_FALSE(code.isFinished(ctx.get()));
+    }
+
+    code.reset(ctx.get());
+    ASSERT_FALSE(code.isFinished(ctx.get()));
+    ASSERT_EQ(ctx->promise(), nullptr);
+
+    EXPECT_CALL(m_mock, f);
+    code.run(ctx.get());
+    ASSERT_TRUE(code.isFinished(ctx.get()));
+
+    // Test with another context
+    Target anotherTarget;
+    auto anotherCtx = code.createExecutionContext(&anotherTarget);
+    ASSERT_FALSE(code.isFinished(anotherCtx.get()));
+    ASSERT_TRUE(code.isFinished(ctx.get()));
+
+    promise = std::make_shared<Promise>();
+    anotherCtx->setPromise(promise);
+    EXPECT_CALL(m_mock, f).Times(0);
+
+    for (int i = 0; i < 10; i++) {
+        code.run(anotherCtx.get());
+        ASSERT_FALSE(code.isFinished(anotherCtx.get()));
+    }
+
+    promise->resolve();
+
+    EXPECT_CALL(m_mock, f);
+    code.run(anotherCtx.get());
+    ASSERT_TRUE(code.isFinished(anotherCtx.get()));
+    ASSERT_TRUE(code.isFinished(ctx.get()));
+
+    code.reset(ctx.get());
+    ASSERT_TRUE(code.isFinished(anotherCtx.get()));
+    ASSERT_FALSE(code.isFinished(ctx.get()));
+}