1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
//===--- TaskStatus.cpp - Unit tests for the task-status API --------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
#include "swift/Basic/STLExtras.h"
#include "swift/Runtime/Concurrency.h"
#include "gtest/gtest.h"
using namespace swift;
namespace {
template <class T> struct ValueContext;
template <class T>
using InvokeFunctionRef =
llvm::function_ref<void(ValueContext<T> *context)>;
using BodyFunctionRef =
llvm::function_ref<void(AsyncTask *task)>;
template <class Storage> struct ValueContext : AsyncContext {
Storage Value;
InvokeFunctionRef<Storage> StoredInvokeFn;
};
// Disable template argument deduction.
template <class T>
using undeduced =
typename std::enable_if<std::is_same<T, T>::value, T>::type;
template <class T>
SWIFT_CC(swiftasync)
static void simpleTaskInvokeFunction(SWIFT_ASYNC_CONTEXT AsyncContext *context,
SWIFT_CONTEXT HeapObject *) {
auto valueContext = static_cast<ValueContext<T>*>(context);
valueContext->StoredInvokeFn(valueContext);
// Destroy the stored value.
valueContext->Value.T::~T();
// Return to finish off the task.
// In a normal situation we'd need to free the context, but here
// we know we're at the top level.
return valueContext->ResumeParent(valueContext);
}
template <class T>
static void withSimpleTask(TaskCreateFlags flags, T &&value,
undeduced<InvokeFunctionRef<T>> invokeFn,
BodyFunctionRef body) {
auto taskAndContext =
swift_task_create_common(flags.getOpaqueValue(),
nullptr,
nullptr,
reinterpret_cast<TaskContinuationFunction *>(
&simpleTaskInvokeFunction<T>),
nullptr,
sizeof(ValueContext<T>));
auto valueContext =
static_cast<ValueContext<T>*>(taskAndContext.InitialContext);
new (&valueContext->Value) T(std::forward<T>(value));
valueContext->StoredInvokeFn = invokeFn;
// Forward our owning reference to the task into its execution,
// causing it to be destroyed when it completes.
body(taskAndContext.Task);
}
template <class T>
static void withSimpleTask(T &&value,
undeduced<InvokeFunctionRef<T>> invokeFn,
BodyFunctionRef bodyFn) {
withSimpleTask(TaskCreateFlags(), std::forward<T>(value), invokeFn, bodyFn);
}
static SerialExecutorRef createFakeExecutor(uintptr_t value) {
return SerialExecutorRef::forDefaultActor(reinterpret_cast<DefaultActor*>(value));
}
} // end anonymous namespace
TEST(TaskStatusTest, basicTasks) {
AsyncTask *createdTask = nullptr;
auto createdExecutor = createFakeExecutor(1234);
bool hasRun = false;
struct Storage { int value; };
withSimpleTask(Storage{47},
[&](ValueContext<Storage> *context) {
// The task passed in should be the task we created.
EXPECT_EQ(createdTask, swift_task_getCurrent());
// The executor passed in should be the executor we created.
//EXPECT_EQ(createdExecutor, executor);
// We shouldn't have run yet.
EXPECT_FALSE(hasRun);
// The context should've been initialized correctly.
// (This is really just testing our harness, not the runtime.)
EXPECT_EQ(47, context->Value.value);
hasRun = true;
}, [&](AsyncTask *task) {
createdTask = task;
EXPECT_FALSE(hasRun);
swift_job_run(task, createdExecutor);
EXPECT_TRUE(hasRun);
createdTask = nullptr;
});
EXPECT_TRUE(hasRun);
EXPECT_EQ(nullptr, createdTask);
}
TEST(TaskStatusTest, cancellation_simple) {
struct Storage { int value; };
withSimpleTask(Storage{47},
[&](ValueContext<Storage> *context) {
auto task = swift_task_getCurrent();
EXPECT_FALSE(swift_task_isCancelled(task));
swift_task_cancel(task);
EXPECT_TRUE(swift_task_isCancelled(task));
swift_task_cancel(task);
EXPECT_TRUE(swift_task_isCancelled(task));
}, [&](AsyncTask *task) {
swift_job_run(task, createFakeExecutor(1234));
});
}
|
