From 7ad97523318e247fd9dce8e3e23b8b5c65c414bc Mon Sep 17 00:00:00 2001
From: "lcrusoveanu@optaros.com" <lcrusoveanu@optaros.com>
Date: Wed, 12 Oct 2016 12:32:19 +0300
Subject: [PATCH] update test with after method

---
 .../web/CustomUserDetailsServiceTest.java     | 40 +++++++++----------
 1 file changed, 18 insertions(+), 22 deletions(-)

diff --git a/spring-security-custom-permission/src/test/java/org/baeldung/web/CustomUserDetailsServiceTest.java b/spring-security-custom-permission/src/test/java/org/baeldung/web/CustomUserDetailsServiceTest.java
index 6917cea99a..0c82a4e98b 100644
--- a/spring-security-custom-permission/src/test/java/org/baeldung/web/CustomUserDetailsServiceTest.java
+++ b/spring-security-custom-permission/src/test/java/org/baeldung/web/CustomUserDetailsServiceTest.java
@@ -39,37 +39,33 @@ public class CustomUserDetailsServiceTest {
 
     @Test
     public void givenExistingUser_whenAuthenticate_thenRetrieveFromDb() {
-        try {
-            User user = new User();
-            user.setUsername(USERNAME);
-            user.setPassword(passwordEncoder.encode(PASSWORD));
+        User user = new User();
+        user.setUsername(USERNAME);
+        user.setPassword(passwordEncoder.encode(PASSWORD));
 
-            myUserRepository.save(user);
+        myUserRepository.save(user);
 
-            UsernamePasswordAuthenticationToken auth = new UsernamePasswordAuthenticationToken(USERNAME, PASSWORD);
-            Authentication authentication = authenticationProvider.authenticate(auth);
+        UsernamePasswordAuthenticationToken auth = new UsernamePasswordAuthenticationToken(USERNAME, PASSWORD);
+        Authentication authentication = authenticationProvider.authenticate(auth);
 
-            assertEquals(authentication.getName(), USERNAME);
-
-        } finally {
-            myUserRepository.removeUserByUsername(USERNAME);
-        }
+        assertEquals(authentication.getName(), USERNAME);
     }
 
     @Test(expected = BadCredentialsException.class)
     public void givenIncorrectUser_whenAuthenticate_thenBadCredentialsException() {
-        try {
-            User user = new User();
-            user.setUsername(USERNAME);
-            user.setPassword(passwordEncoder.encode(PASSWORD));
+        User user = new User();
+        user.setUsername(USERNAME);
+        user.setPassword(passwordEncoder.encode(PASSWORD));
 
-            myUserRepository.save(user);
+        myUserRepository.save(user);
 
-            UsernamePasswordAuthenticationToken auth = new UsernamePasswordAuthenticationToken(USERNAME2, PASSWORD);
-            authenticationProvider.authenticate(auth);
-        } finally {
-            myUserRepository.removeUserByUsername(USERNAME);
-        }
+        UsernamePasswordAuthenticationToken auth = new UsernamePasswordAuthenticationToken(USERNAME2, PASSWORD);
+        authenticationProvider.authenticate(auth);
+    }
+    
+    @After
+    public void tearDown() {
+        myUserRepository.removeUserByUsername(USERNAME);
     }
 
 }