haiku.git

/*
 * Copyright 2020, Andrew Lindesay <apl@lindesay.co.nz>.
 * All rights reserved. Distributed under the terms of the MIT License.
 */
#include "TarArchiveServiceTest.h"

#include <stdlib.h>

#include <AutoDeleter.h>
#include <File.h>
#include <String.h>

#include <cppunit/TestCaller.h>
#include <cppunit/TestSuite.h>

#include "TarArchiveHeader.h"
#include "TarArchiveService.h"

#include "TarArchiveServiceTestSample.h"


TarArchiveServiceTest::TarArchiveServiceTest()
{
}


TarArchiveServiceTest::~TarArchiveServiceTest()
{
}


void
TarArchiveServiceTest::TestReadHeader()
{
	BString tarPath;
	if (CreateSampleFile(tarPath) != B_OK) {
		printf("! unable to setup the sample tar --> exit");
		exit(1);
	}

	BFile *tarFile = new BFile(tarPath, O_RDONLY);
	tarFile->Seek(2048, SEEK_SET);
		// known offset in the same data

	TarArchiveHeader header;

// ----------------------
	status_t result = TarArchiveService::GetEntry(*tarFile, header);
// ----------------------

	CPPUNIT_ASSERT_EQUAL(B_OK, result);
	CPPUNIT_ASSERT_EQUAL(BString("hicn/somepkg/16.png"), header.FileName());
	CPPUNIT_ASSERT_EQUAL(657, header.Length());
	CPPUNIT_ASSERT_EQUAL(TAR_FILE_TYPE_NORMAL, header.FileType());

	delete tarFile;

	DeleteSampleFile(tarPath);
}


/*!	This method will store the tar file sample into a temporary file and will
	return the pathname to that file.
*/

status_t
TarArchiveServiceTest::CreateSampleFile(BString& path)
{
	path.SetTo(tmpnam(NULL));
	BFile* file = new BFile(path, O_WRONLY | O_CREAT);
	ObjectDeleter<BFile> tarIoDeleter(file);
	status_t result = file->WriteExactly(kSampleTarPayload, kSampleTarLength);

	if (result != B_OK) {
		printf("! unable to write the sample data to [%s]\n", path.String());
	}

	return result;
}


status_t
TarArchiveServiceTest::DeleteSampleFile(const BString& path)
{
	if (remove(path.String()) != 0)
		return B_ERROR;
	return B_OK;
}


/*static*/ void
TarArchiveServiceTest::AddTests(BTestSuite& parent)
{
	CppUnit::TestSuite& suite = *new CppUnit::TestSuite("TarArchiveServiceTest");

	suite.addTest(
		new CppUnit::TestCaller<TarArchiveServiceTest>(
			"TarArchiveServiceTest::TestReadHeader",
			&TarArchiveServiceTest::TestReadHeader));

	parent.addTest("TarArchiveServiceTest", &suite);
}