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/Support/StreamingMemoryObject.h"
11 #include "gtest/gtest.h"
17 class NullDataStreamer : public DataStreamer {
18 size_t GetBytes(unsigned char *buf, size_t len) override {
25 TEST(StreamingMemoryObject, Test) {
26 auto *DS = new NullDataStreamer();
27 StreamingMemoryObject O(DS);
28 EXPECT_TRUE(O.isValidAddress(32 * 1024));
31 TEST(StreamingMemoryObject, TestSetKnownObjectSize) {
32 auto *DS = new NullDataStreamer();
33 StreamingMemoryObject O(DS);
35 EXPECT_EQ((uint64_t) 16, O.readBytes(Buf, 16, 0));
36 O.setKnownObjectSize(24);
37 EXPECT_EQ((uint64_t) 8, O.readBytes(Buf, 16, 16));