CopyCommandTest.java

/**
 * Copyright (c) 2012, Thilo Planz. All rights reserved.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
package v7db.files;

import java.io.ByteArrayInputStream;
import java.io.IOException;

import jmockmongo.MockMongoTestCaseSupport;

import org.apache.commons.io.IOUtils;
import org.bson.types.ObjectId;

import v7db.files.mongodb.MongoContentStorage;
import v7db.files.mongodb.V7File;
import v7db.files.mongodb.V7GridFS;
import v7db.files.spi.ContentSHA;

public class CopyCommandTest extends MockMongoTestCaseSupport {

	@Override
	protected void setUp() throws Exception {
		super.setUp();
		UnitTestSupport.initConfiguration(UnitTestSupport
				.getDefaultProperties());

	}

	public void testSHA() throws IOException {
		MongoContentStorage storage = new MongoContentStorage(getMongo().getDB(
				"test"));
		ContentSHA sha = storage.storeContent(new ByteArrayInputStream("test"
				.getBytes()));

		CopyCommand.main(new String[] { "copy", "-sha", sha.getDigest(), "x",
				"copy.txt" });

		V7GridFS fs = new V7GridFS(getMongo().getDB("test"));
		V7File file = fs.getFile("x", "copy.txt");
		assertEquals("test", IOUtils.toString(file.getInputStream()));

	}

	public void testCopyFile() throws IOException {
		V7GridFS fs = new V7GridFS(getMongo().getDB("test"));
		ObjectId dir = fs.addFolder("test", "folder");
		fs.addFile("test".getBytes(), dir, "test.txt", "text/plain");

		CopyCommand.main(new String[] { "copy", "test", "folder/test.txt", "x",
				"copy.txt" });

		V7File file = fs.getFile("x", "copy.txt");
		assertEquals("test", IOUtils.toString(file.getInputStream()));
		assertEquals("text/plain", file.getContentType());
	}

	public void testCopyDirectory() throws IOException {
		V7GridFS fs = new V7GridFS(getMongo().getDB("test"));
		ObjectId dir = fs.addFolder("test", "folder");
		fs.addFile("test".getBytes(), dir, "test.txt", "text/plain");

		CopyCommand
				.main(new String[] { "copy", "test", "folder", "x", "copy" });

		V7File file = fs.getFile("x", "copy", "test.txt");
		assertEquals("test", IOUtils.toString(file.getInputStream()));
		assertEquals("text/plain", file.getContentType());
	}

}