bookdaoimpltest.java

来自「在线商店」· Java 代码 · 共 82 行

package dao.impl;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.util.List;



import junit.framework.TestCase;
import domain.BookInfo;

public class BookDAOImplTest extends TestCase {
	public void testAddBook() {
		try {
			
			
			
			BookInfo book = new BookInfo();
			book.setId("book000001");
			book.setBrief("goodbookgoodbook");

			ByteArrayOutputStream baos = new ByteArrayOutputStream();
			BufferedInputStream bis = new BufferedInputStream(
					new FileInputStream("e:/water.jpg"));

			int b = -1;
			while ((b = bis.read()) != -1) {
				baos.write(b);
			}
			byte[] cover = baos.toByteArray();

			bis.close();
			baos.close();

			book.setCover(cover);

			BookDAOImpl dao = new BookDAOImpl();
			dao.add(book);

		} catch (Exception e) {
			e.printStackTrace();
			TestCase.fail();
		}
	}

	public void testFindAll() {
		try {
			BookDAOImpl dao = new BookDAOImpl();
			List list = dao.findAll();
			System.out.println(list);
		} catch (Exception e) {
			e.printStackTrace();
			TestCase.fail();
		}
	}

	public void testGetCover() {
		try {
			BookDAOImpl dao = new BookDAOImpl();
			BookInfo book = new BookInfo();
			book.setId("book000001");
			byte[] cover = dao.getCover(book);
			BufferedOutputStream bos = new BufferedOutputStream(
					new FileOutputStream("d:/img.jpg"));
			ByteArrayInputStream bais = new ByteArrayInputStream(cover);
			int b = -1;
			while ((b = bais.read()) != -1) {
				bos.write(b);
			}
			bais.close();
			bos.close();
		} catch (Exception e) {
			e.printStackTrace();
			TestCase.fail();
		}
	}
}