1 //===- llvm/unittest/Support/StreamingMemoryObject.cpp - unit tests -------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #include "llvm/ADT/STLExtras.h"
11 #include "llvm/Support/StreamingMemoryObject.h"
12 #include "gtest/gtest.h"
18 class NullDataStreamer : public DataStreamer {
19 size_t GetBytes(unsigned char *buf, size_t len) override {
26 TEST(StreamingMemoryObject, Test) {
27 auto DS = make_unique<NullDataStreamer>();
28 StreamingMemoryObject O(std::move(DS));
29 EXPECT_TRUE(O.isValidAddress(32 * 1024));
32 TEST(StreamingMemoryObject, TestSetKnownObjectSize) {
33 auto DS = make_unique<NullDataStreamer>();
34 StreamingMemoryObject O(std::move(DS));
36 EXPECT_EQ((uint64_t) 16, O.readBytes(Buf, 16, 0));
37 O.setKnownObjectSize(24);
38 EXPECT_EQ((uint64_t) 8, O.readBytes(Buf, 16, 16));