/*
 * Copyright (c) 2014 tabletoptool.com team.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Public License v3.0
 * which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/gpl.html
 * 
 * Contributors:
 *     rptools.com team - initial implementation
 *     tabletoptool.com team - further development
 */
package com.t3.transfer;

import org.testng.annotations.Test;
import org.testng.AssertJUnit;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;

public class AssetTransferManagerTest {

	@Test
	public void testBasicTransfer() throws Exception {
		
		byte[] data = new byte[1024];
		for (int i = 0; i < 1024; i++) {
			data[i] = (byte)i;
		}
		
		File tmpFile = createTempFile(data);
		
		// PRODUCER
		AssetProducer producer = new AssetProducer("Testing", "onetwo", tmpFile);
		AssetHeader header = producer.getHeader();
		
		AssertJUnit.assertNotNull(header);
		AssertJUnit.assertEquals(data.length, header.getSize());
		AssertJUnit.assertFalse(producer.isComplete());
		
		// CONSUMER
		AssetConsumer consumer = new AssetConsumer(new File("."), header);

		AssertJUnit.assertFalse(consumer.isComplete());
		
		// TEST
		while (!producer.isComplete()) {
			AssetChunk chunk = producer.nextChunk(10);
			
			consumer.update(chunk);
		}
		
		// CHECK
		AssertJUnit.assertTrue(consumer.isComplete());
		AssertJUnit.assertTrue(consumer.getFilename().exists());
		AssertJUnit.assertEquals(header.getSize(), consumer.getFilename().length());

		int count = 0;
		int val;
		try(FileInputStream in = new FileInputStream(consumer.getFilename())) {
			while ((val = in.read()) != -1) {
				AssertJUnit.assertEquals(data[count], (byte)val);
				count ++;
			}
		}
		AssertJUnit.assertEquals(data.length, count);
		
		// CLEANUP
		tmpFile.delete();
		consumer.getFilename().delete();
		
	}
	
	private File createTempFile(byte[] data) throws IOException {
		
		File file = new File("tmp.dat");
		try (FileOutputStream out = new FileOutputStream(file)) {
			out.write(data);
		}
		
		return file;
	}
}