Metacat

import gov.loc.repository.bagit.Bag;

import gov.loc.repository.bagit.BagFactory;

import java.io.File;

import java.io.FileOutputStream;

    // MN packaging tests

    suite.addTest(new MNodeServiceTest("testGetPackage"));

	/**

	 * Test getting a known object

	 */

	public void testGetPackage() {

		printTestHeader("testGetPackage");

		try {

			Session session = getTestSession();

			Identifier guid = new Identifier();

			guid.setValue("testGetPackage." + System.currentTimeMillis());

			InputStream object = new ByteArrayInputStream("test".getBytes("UTF-8"));

			SystemMetadata sysmeta = createSystemMetadata(guid, session.getSubject(), object);

			Identifier pid = MNodeService.getInstance(request).create(session, guid, object, sysmeta);

			InputStream bagStream = MNodeService.getInstance(request).getPackage(session, pid);

			File bagFile = File.createTempFile("bagit", ".zip");

			IOUtils.copy(bagStream, new FileOutputStream(bagFile));

			BagFactory bagFactory = new BagFactory();

			Bag bag = bagFactory.createBag(bagFile);

			InputStream result = bag.getPayload().iterator().next().newInputStream();

			// go back to beginning of original stream

			object.reset();

			// check

			assertTrue(object.available() > 0);

			assertTrue(result.available() > 0);

			assertTrue(IOUtils.contentEquals(result, object));

			// clean up

			bagFile.delete();

		} catch (Exception e) {

			e.printStackTrace();

			fail("Unexpected error: " + e.getMessage());

		}

	}