EncryCredentialAuthenticationHandlerTest.java

package com.github.ebnew.ki4so.core.authentication.handlers;

import junit.framework.Assert;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.mockito.Mockito;

import com.github.ebnew.ki4so.core.authentication.Credential;
import com.github.ebnew.ki4so.core.authentication.EncryCredential;
import com.github.ebnew.ki4so.core.authentication.EncryCredentialManager;
import com.github.ebnew.ki4so.core.exception.InvalidEncryCredentialException;
import com.github.ebnew.ki4so.core.model.EncryCredentialInfo;

/**
 * 测试加密后凭据认证处理器对象。
 * @author burgess yang
 *
 */
public class EncryCredentialAuthenticationHandlerTest {
	
	private EncryCredentialAuthenticationHandler handler;

	@Before
	public void setUp() throws Exception {
		handler = new EncryCredentialAuthenticationHandler();
	}

	@After
	public void tearDown() throws Exception {
	}

	@Test
	public void testDoAuthentication() {
		
		/**
		 * 测试异常输入情况。
		 */
		Assert.assertFalse(handler.authenticate(null));
		Credential credential = Mockito.mock(Credential.class);
		Assert.assertFalse(handler.authenticate(credential));
		
		/**
		 * 测试解密失败抛出异常的情况。
		 */
		EncryCredential encryCredential = new EncryCredential("sssddaf");
		EncryCredentialManager encryCredentialManager = Mockito.mock(EncryCredentialManager.class);
		this.handler.setEncryCredentialManager(encryCredentialManager);
		Mockito.when(encryCredentialManager.decrypt(encryCredential)).thenThrow(InvalidEncryCredentialException.INSTANCE);
		Assert.assertFalse(handler.authenticate(encryCredential));
		
		/**
		 * 测试解密返回null的情况。
		 */
		Mockito.reset(encryCredentialManager);
		Mockito.when(encryCredentialManager.decrypt(encryCredential)).thenReturn(null);
		Assert.assertFalse(handler.authenticate(encryCredential));
		
		/**
		 * 测试解密成功，但是凭据不合法的情况。
		 */
		Mockito.reset(encryCredentialManager);
		EncryCredentialInfo encryCredentialInfo = new EncryCredentialInfo();
		Mockito.when(encryCredentialManager.decrypt(encryCredential)).thenReturn(encryCredentialInfo);
		Assert.assertFalse(handler.authenticate(encryCredential));
		
		
		/**
		 * 测试解密成功，但是凭据合法的情况。
		 */
		Mockito.reset(encryCredentialManager);
		encryCredentialInfo = new EncryCredentialInfo();
		Mockito.when(encryCredentialManager.decrypt(encryCredential)).thenReturn(encryCredentialInfo);
		Mockito.when(encryCredentialManager.checkEncryCredentialInfo(encryCredentialInfo)).thenReturn(true);
		Assert.assertTrue(handler.authenticate(encryCredential));
	}

}