averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-14 20:36:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

[SourceKit] Add Semaphore to SourceKitSupport's concurrency module

Browse Source 
Includes a libdispatch-based implementation.
This commit is contained in:
Brian Croom
2016-06-20 10:25:51 -04:00
parent 2e9f13e463
commit 479584109c
6 changed files with 89 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
unittests/SourceKit/SwiftLang/CursorInfoTest.cpp
View File

@@ -13,13 +13,11 @@
#include "SourceKit/Core/Context.h"
#include "SourceKit/Core/LangSupport.h"
#include "SourceKit/Core/NotificationCenter.h"
#include "SourceKit/Support/Concurrency.h"
#include "llvm/Support/MemoryBuffer.h"
#include "llvm/Support/Path.h"
#include "gtest/gtest.h"
// FIXME: Portability.
#include <dispatch/dispatch.h>
using namespace SourceKit;
using namespace llvm;
@@ -98,18 +96,12 @@ struct TestCursorInfo {
class CursorInfoTest : public ::testing::Test {
SourceKit::Context Ctx{ getRuntimeLibPath() };
std::atomic<int> NumTasks;
dispatch_semaphore_t TaskSema = nullptr;
public:
LangSupport &getLang() { return Ctx.getSwiftLangSupport(); }
void SetUp() {
NumTasks = 0;
TaskSema = dispatch_semaphore_create(0);
}
void TearDown() {
dispatch_release(TaskSema);
}
void addNotificationReceiver(DocumentUpdateNotificationReceiver Receiver) {
@@ -133,7 +125,7 @@ public:
TestCursorInfo getCursor(const char *DocName, unsigned Offset,
ArrayRef<const char *> CArgs) {
auto Args = makeArgs(DocName, CArgs);
dispatch_semaphore_t sema = dispatch_semaphore_create(0);
Semaphore sema(0);
TestCursorInfo TestInfo;
getLang().getCursorInfo(DocName, Offset, Args,
@@ -142,40 +134,15 @@ public:
TestInfo.Typename = Info.TypeName;
TestInfo.Filename = Info.Filename;
TestInfo.DeclarationLoc = Info.DeclarationLoc;
dispatch_semaphore_signal(sema);
sema.signal();
});
dispatch_time_t when = dispatch_time(DISPATCH_TIME_NOW, NSEC_PER_SEC * 60);
bool expired = dispatch_semaphore_wait(sema, when);
bool expired = sema.wait(60 * 1000);
if (expired)
llvm::report_fatal_error("check took too long");
dispatch_release(sema);
return TestInfo;
}
void checkCursorAsync(const char *DocName, unsigned Offset,
ArrayRef<const char *> CArgs,
StringRef ExpectedName, StringRef ExpectedTypename) {
auto Args = makeArgs(DocName, CArgs);
++NumTasks;
getLang().getCursorInfo(DocName, Offset, Args,
[this, ExpectedName, ExpectedTypename](const CursorInfo &Info) {
EXPECT_EQ(ExpectedName, Info.Name);
EXPECT_EQ(ExpectedTypename, Info.TypeName);
int Num = --NumTasks;
if (Num == 0)
dispatch_semaphore_signal(TaskSema);
});
}
void waitForChecks() {
dispatch_time_t when = dispatch_time(DISPATCH_TIME_NOW, NSEC_PER_SEC * 60);
bool expired = dispatch_semaphore_wait(TaskSema, when);
if (expired)
llvm::report_fatal_error("checks took too long");
}
unsigned findOffset(StringRef Val, StringRef Text) {
auto pos = Text.find(Val);
assert(pos != StringRef::npos);