1 //===- llvm/unittest/Support/StreamingMemoryObject.cpp - unit tests -------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 10 #include "llvm/ADT/STLExtras.h" 11 #include "llvm/Support/StreamingMemoryObject.h" 12 #include "gtest/gtest.h" 13 #include <string.h> 14 15 using namespace llvm; 16 17 namespace { 18 class NullDataStreamer : public DataStreamer { 19 size_t GetBytes(unsigned char *buf, size_t len) override { 20 memset(buf, 0, len); 21 return len; 22 } 23 }; 24 } 25 26 TEST(StreamingMemoryObject, Test) { 27 auto DS = make_unique<NullDataStreamer>(); 28 StreamingMemoryObject O(std::move(DS)); 29 EXPECT_TRUE(O.isValidAddress(32 * 1024)); 30 } 31 32 TEST(StreamingMemoryObject, TestSetKnownObjectSize) { 33 auto DS = make_unique<NullDataStreamer>(); 34 StreamingMemoryObject O(std::move(DS)); 35 uint8_t Buf[32]; 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)); 39 } 40
