LocalFileServiceIntegrationTests.java

package org.springframework.batch.admin.service;

import java.io.File;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.core.io.Resource;
import org.springframework.integration.MessageRejectedException;
import org.springframework.messaging.Message;
import org.springframework.messaging.MessageDeliveryException;
import org.springframework.messaging.MessageHandler;
import org.springframework.messaging.MessageHandlingException;
import org.springframework.messaging.SubscribableChannel;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;

@ContextConfiguration
@RunWith(SpringJUnit4ClassRunner.class)
public class LocalFileServiceIntegrationTests {
	
	private static Log logger = LogFactory.getLog(LocalFileServiceIntegrationTests.class);

	@Autowired
	private LocalFileService service;

	@Autowired
	@Qualifier("input-files")
	private SubscribableChannel files;
	
	private static File trigger;

	private static MessageHandler handler = new MessageHandler() {
		public void handleMessage(Message<?> message) throws MessageRejectedException, MessageHandlingException,
				MessageDeliveryException {
			logger.debug("Handled " + message);
			trigger = (File) message.getPayload();
		}
	};

	@Before
	public void setUp() throws Exception {
		File bucket = new File(service.getUploadDirectory(), "spam/bucket");
		if(bucket.exists()) {
			bucket.delete();
		}
		File crap = new File(service.getUploadDirectory(), "spam/bucket/crap");
		if(crap.exists()) {
			crap.delete();
		}
		files.unsubscribe(handler);
		files.subscribe(handler);
	}

	@Test
	public void testUpload() throws Exception {
		FileInfo info = service.createFile("spam/bucket");
		Resource file = service.getResource(info.getPath());
		assertTrue(file.exists());
		assertTrue(file.getFile().getParentFile().exists());
	}

	@Test
	public void testTrigger() throws Exception {
		FileInfo info = service.createFile("spam/bucket/crap");
		Resource file = service.getResource(info.getPath());
		assertTrue(file.exists());
		service.publish(info);
		assertNotNull(trigger);
	}

}