spring boot security + DB認証のテストコード

はじめに

• 作成したテストコードは【spring boot security + DB認証を試した時のポイント】用です

コード

• 下記のコードはDBへの接続を行わなずにモックを使用した場合のテストコード

DemoControllerTest.java

@RunWith(SpringJUnit4ClassRunner.class)
@SpringBootTest
@ContextConfiguration
@WebAppConfiguration
public class DemoControllerTest {
	private MockMvc mockMvc;

	@Autowired
    private WebApplicationContext context;

	@Autowired
    private FilterChainProxy springSecurityFilterChain;

	@InjectMocks
	private DemoController demoController;

	@MockBean
	private UserRepository userRepository;

	@Before
	public void setUp() {
		mockMvc = MockMvcBuilders
					.webAppContextSetup(context)
					.apply(springSecurity(springSecurityFilterChain))
					.build();
	}

	@Test
	public void testLoginOk() throws Exception {
		// 正常系
		String user = "user01";
		UserModel userModel = new UserModel("user01", "ゆーざ０１", "pass1", true);
		doReturn(userModel).when(userRepository).selectByUser(user);

		mockMvc.perform(formLogin("/login").user(user).password("pass1"))
			.andExpect(status().isOk());
	}

	@Test
	public void testLoginNg1() throws Exception {
		// DBにユーザが存在しない場合テスト
		String user = "user01";
		UserModel userModel = null;
		doReturn(userModel).when(userRepository).selectByUser(user);

		mockMvc.perform(formLogin("/login").user(user).password("pass1"))
			.andExpect(status().is3xxRedirection());
	}

	@Test
	public void testLoginNg2() throws Exception {
		// 有効フラグが無効になっている場合のテスト
		String user = "user01";
		UserModel userModel = new UserModel("user01", "ゆーざ０１", "pass1", false);

		doReturn(userModel).when(userRepository).selectByUser(user);

		mockMvc.perform(formLogin("/login").user(user).password("pass1"))
			.andExpect(status().is3xxRedirection());
	}

}

• 下記のコードはDBへの接続を行い一連の動作確認する場合のテストコード

DemoControllerITTest.java

@RunWith(SpringJUnit4ClassRunner.class)
@SpringBootTest
@ContextConfiguration
@WebAppConfiguration
public class DemoControllerITTest {
	private MockMvc mockMvc;

	@Autowired
    private WebApplicationContext context;

	@Autowired
    private FilterChainProxy springSecurityFilterChain;

	@Before
	public void setUp() {
		mockMvc = MockMvcBuilders
					.webAppContextSetup(context)
					.apply(springSecurity(springSecurityFilterChain))
					.build();
	}

	@Test
	public void test1() throws Exception {
		// ルートへのアクセス時のテスト
		mockMvc.perform(get("/"))
			.andExpect(status().isOk())
			.andExpect(view().name("home"));
	}

	@Test
	public void testLoginOk1() throws Exception {
		// ログイン画面の正常系テスト
		mockMvc.perform(formLogin("/login").user("user01").password("pass"))
			.andExpect(status().isOk());
	}

	@Test
	public void testLoginNg1() throws Exception {
		// ログイン画面の異常系テスト
		mockMvc.perform(formLogin("/login").user("user01").password("pass1"))
			.andExpect(status().is3xxRedirection());
	}

}