Files
swift-mirror/unittests/SourceKit/Support/ImmutableTextBufferTest.cpp
Rintaro Ishizaki 930d4342cd [SourceKit] Defensive guard for invalid offset and length in edit request
Otherwise, hits assertion, or crashes in no-assertion build.

Added 'EditableTextBuffer::getSize()' for getting size after previous updates
without actually applying them.

rdar://problem/34206143
2018-08-01 20:02:15 +09:00

52 lines
1.7 KiB
C++

//===----------------------------------------------------------------------===//
//
// 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 "SourceKit/Support/ImmutableTextBuffer.h"
#include "gtest/gtest.h"
using namespace SourceKit;
using namespace llvm;
TEST(EditableTextBuffer, Updates) {
const char *Text = "hello world";
size_t Length = strlen(Text);
EditableTextBufferManager BufMgr;
EditableTextBufferRef EdBuf = BufMgr.getOrCreateBuffer("/a/test", Text);
ImmutableTextBufferRef Buf = EdBuf->getBuffer();
EXPECT_EQ(Buf->getText(), Text);
EXPECT_EQ(EdBuf->getSize(), Length);
Buf = EdBuf->insert(6, "all ")->getBuffer();
EXPECT_EQ(Buf->getText(), "hello all world");
EXPECT_EQ(EdBuf->getSize(), strlen("hello all world"));
Buf = EdBuf->erase(9, 6)->getBuffer();
EXPECT_EQ(Buf->getText(), "hello all");
EXPECT_EQ(EdBuf->getSize(), strlen("hello all"));
Buf = EdBuf->replace(0, 5, "yo")->getBuffer();
EXPECT_EQ(Buf->getText(), "yo all");
EXPECT_EQ(EdBuf->getSize(), strlen("yo all"));
EdBuf = BufMgr.resetBuffer("/a/test", Text);
EdBuf->insert(6, "all ");
EdBuf->erase(9, 6);
EdBuf->replace(0, 5, "yo");
EXPECT_EQ(EdBuf->getSize(), strlen("yo all"));
Buf = EdBuf->getSnapshot()->getBuffer();
EXPECT_EQ(Buf->getText(), "yo all");
EXPECT_EQ(Buf->getFilename(), "/a/test");
}