C++程序  |  71行  |  1.98 KB

//===- FragmentRefTest ----------------------------------------------------===//
//
//                     The MCLinker Project
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#include "FragmentRefTest.h"

#include <mcld/LD/FragmentRef.h>
#include <mcld/LD/RegionFragment.h>
#include <mcld/Support/MemoryAreaFactory.h>
#include <mcld/Support/FileHandle.h>
#include <mcld/Support/Path.h>

using namespace mcld;
using namespace mcld::sys::fs;
using namespace mcld::sys::fs::detail;
using namespace mcldtest;

// Constructor can do set-up work for all test here.
FragmentRefTest::FragmentRefTest()
{
}

// Destructor can do clean-up work that doesn't throw exceptions here.
FragmentRefTest::~FragmentRefTest()
{
}

// SetUp() will be called immediately before each test.
void FragmentRefTest::SetUp()
{
}

// TearDown() will be called immediately after each test.
void FragmentRefTest::TearDown()
{
}

//==========================================================================//
// Testcases
//
TEST_F( FragmentRefTest, ) {
  Path path(TOPDIR);
  path.append("unittests/test3.txt");
  MemoryAreaFactory* areaFactory = new MemoryAreaFactory(1);
  MemoryArea* area = areaFactory->produce(path, FileHandle::ReadWrite);

  MemoryRegion* region = area->request(0, 4096);
  RegionFragment *frag = new RegionFragment(*region);
  FragmentRef *ref = new FragmentRef(*frag);

  ASSERT_EQ('H', region->getBuffer()[0]);
  ASSERT_EQ(4096, region->size());
  ASSERT_EQ('H', frag->getRegion().getBuffer()[0]);
  ASSERT_EQ(4096, frag->getRegion().size());
  ASSERT_EQ(frag, ref->frag());
  ASSERT_EQ('H', static_cast<RegionFragment*>(ref->frag())->getRegion().getBuffer()[0]);
  ASSERT_EQ(4096, static_cast<RegionFragment*>(ref->frag())->getRegion().size());
  ASSERT_EQ('H', ref->deref()[0]);

  ASSERT_TRUE(RegionFragment::classof(frag));

  delete ref;
  delete frag;
  delete areaFactory;
}