/*
    * Copyright 2002-2013 the original author or authors.
    *
    * Licensed under the Apache License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    *      http://www.apache.org/licenses/LICENSE-2.0
    *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package org.springframework.jdbc.core.support;
  
  import java.io.ByteArrayInputStream;
  import java.io.InputStream;
  import java.io.InputStreamReader;
  import java.sql.PreparedStatement;
  import java.sql.SQLException;
  import java.sql.Types;
  
  import org.junit.Before;
  import org.junit.Rule;
  import org.junit.Test;
  import org.junit.rules.ExpectedException;
  import org.mockito.ArgumentCaptor;
  import org.mockito.Captor;
  import org.mockito.MockitoAnnotations;
  
  import org.springframework.jdbc.support.lob.LobCreator;
  import org.springframework.jdbc.support.lob.LobHandler;
  
  import static org.hamcrest.Matchers.*;
  import static org.junit.Assert.*;
  import static org.mockito.BDDMockito.*;
  
  /**
   * Test cases for the sql lob value:
   *
   * BLOB:
   *   1. Types.BLOB: setBlobAsBytes (byte[])
   *   2. String: setBlobAsBytes (byte[])
   *   3. else: IllegalArgumentException
   *
   * CLOB:
   *   4. String or NULL: setClobAsString (String)
   *   5. InputStream: setClobAsAsciiStream (InputStream)
   *   6. Reader: setClobAsCharacterStream (Reader)
   *   7. else: IllegalArgumentException
   *
   * @author Alef Arendsen
   */
  public class SqlLobValueTests  {
  
  	@Rule
  	public ExpectedException thrown = ExpectedException.none();
  
  	private PreparedStatement preparedStatement;
  	private LobHandler handler;
  	private LobCreator creator;
  
  	@Captor
  	private ArgumentCaptor<InputStream> inputStreamCaptor;
  
  	@Before
  	public void setUp() {
  		MockitoAnnotations.initMocks(this);
  		preparedStatement = mock(PreparedStatement.class);
  		handler = mock(LobHandler.class);
  		creator = mock(LobCreator.class);
  		given(handler.getLobCreator()).willReturn(creator);
  	}
  
  	@Test
  	public void test1() throws SQLException {
  		byte[] testBytes = "Bla".getBytes();
  		SqlLobValue lob = new SqlLobValue(testBytes, handler);
  		lob.setTypeValue(preparedStatement, 1, Types.BLOB, "test");
  		verify(creator).setBlobAsBytes(preparedStatement, 1, testBytes);
  	}
  
  	@Test
  	public void test2() throws SQLException {
  		String testString = "Bla";
  		SqlLobValue lob = new SqlLobValue(testString, handler);
  		lob.setTypeValue(preparedStatement, 1, Types.BLOB, "test");
  		verify(creator).setBlobAsBytes(preparedStatement, 1, testString.getBytes());
  	}
  
  	@Test
  	public void test3() throws SQLException {
  		SqlLobValue lob = new SqlLobValue(new InputStreamReader(new ByteArrayInputStream("Bla".getBytes())), 12);
  		thrown.expect(IllegalArgumentException.class);
  		lob.setTypeValue(preparedStatement, 1, Types.BLOB, "test");
  	}
  
 	@Test
 	public void test4() throws SQLException {
 		String testContent = "Bla";
 		SqlLobValue lob = new SqlLobValue(testContent, handler);
 		lob.setTypeValue(preparedStatement, 1, Types.CLOB, "test");
 		verify(creator).setClobAsString(preparedStatement, 1, testContent);
 	}
 
 	@Test
 	public void test5() throws Exception {
 		byte[] testContent = "Bla".getBytes();
 		SqlLobValue lob = new SqlLobValue(new ByteArrayInputStream(testContent), 3, handler);
 		lob.setTypeValue(preparedStatement, 1, Types.CLOB, "test");
 		verify(creator).setClobAsAsciiStream(eq(preparedStatement), eq(1), inputStreamCaptor.capture(), eq(3));
 		byte[] bytes = new byte[3];
 		inputStreamCaptor.getValue().read(bytes );
 		assertThat(bytes, equalTo(testContent));
 	}
 
 	@Test
 	public void test6() throws SQLException {
 		byte[] testContent = "Bla".getBytes();
 		ByteArrayInputStream bais = new ByteArrayInputStream(testContent);
 		InputStreamReader reader = new InputStreamReader(bais);
 		SqlLobValue lob = new SqlLobValue(reader, 3, handler);
 		lob.setTypeValue(preparedStatement, 1, Types.CLOB, "test");
 		verify(creator).setClobAsCharacterStream(eq(preparedStatement), eq(1), eq(reader), eq(3));
 	}
 
 	@Test
 	public void test7() throws SQLException {
 		SqlLobValue lob = new SqlLobValue("bla".getBytes());
 		thrown.expect(IllegalArgumentException.class);
 		lob.setTypeValue(preparedStatement, 1, Types.CLOB, "test");
 	}
 
 	@Test
 	public void testOtherConstructors() throws SQLException {
 		// a bit BS, but we need to test them, as long as they don't throw exceptions
 
 		SqlLobValue lob = new SqlLobValue("bla");
 		lob.setTypeValue(preparedStatement, 1, Types.CLOB, "test");
 
 		try {
 			lob = new SqlLobValue("bla".getBytes());
 			lob.setTypeValue(preparedStatement, 1, Types.CLOB, "test");
 			fail("IllegalArgumentException should have been thrown");
 		}
 		catch (IllegalArgumentException e) {
 			// expected
 		}
 
 		lob = new SqlLobValue(new ByteArrayInputStream("bla".getBytes()), 3);
 		lob.setTypeValue(preparedStatement, 1, Types.CLOB, "test");
 
 		lob = new SqlLobValue(new InputStreamReader(new ByteArrayInputStream(
 				"bla".getBytes())), 3);
 		lob.setTypeValue(preparedStatement, 1, Types.CLOB, "test");
 
 		// same for BLOB
 		lob = new SqlLobValue("bla");
 		lob.setTypeValue(preparedStatement, 1, Types.BLOB, "test");
 
 		lob = new SqlLobValue("bla".getBytes());
 		lob.setTypeValue(preparedStatement, 1, Types.BLOB, "test");
 
 		lob = new SqlLobValue(new ByteArrayInputStream("bla".getBytes()), 3);
 		lob.setTypeValue(preparedStatement, 1, Types.BLOB, "test");
 
 		lob = new SqlLobValue(new InputStreamReader(new ByteArrayInputStream(
 				"bla".getBytes())), 3);
 
 		try {
 			lob.setTypeValue(preparedStatement, 1, Types.BLOB, "test");
 			fail("IllegalArgumentException should have been thrown");
 		}
 		catch (IllegalArgumentException e) {
 			// expected
 		}
 	}
 
 	@Test
 	public void testCorrectCleanup() throws SQLException {
 		SqlLobValue lob = new SqlLobValue("Bla", handler);
 		lob.setTypeValue(preparedStatement, 1, Types.CLOB, "test");
 		lob.cleanup();
 		verify(creator).setClobAsString(preparedStatement, 1, "Bla");
 		verify(creator).close();
 	}
 
 	@Test
 	public void testOtherSqlType() throws SQLException {
 		SqlLobValue lob = new SqlLobValue("Bla", handler);
 		thrown.expect(IllegalArgumentException.class);
 		lob.setTypeValue(preparedStatement, 1, Types.SMALLINT, "test");
 	}
 
 }