Feature: Add support for removing an user

Author: KangDroid

src/main/kotlin/com/mptsix/todaydiary/controller/UserController.kt

@@ -56,4 +56,11 @@ class UserController(
         userService.changePassword(userToken, passwordChangeRequest)
         return ResponseEntity.noContent().build()
     }
+
+    @DeleteMapping("/api/v1/user")
+    fun removeUser(@RequestHeader header: HttpHeaders): ResponseEntity<Unit> {
+        val userToken: String = header["X-AUTH-TOKEN"]!![0]
+        userService.removeUser(userToken)
+        return ResponseEntity.noContent().build()
+    }
 }

src/main/kotlin/com/mptsix/todaydiary/data/user/UserRepository.kt

@@ -1,7 +1,9 @@
 package com.mptsix.todaydiary.data.user
 
+import com.mongodb.client.result.DeleteResult
 import com.mptsix.todaydiary.data.user.journal.CategoryCountResult
 import com.mptsix.todaydiary.error.exception.NotFoundException
+import com.mptsix.todaydiary.error.exception.UnknownErrorException
 import org.slf4j.Logger
 import org.slf4j.LoggerFactory
 import org.springframework.data.mongodb.core.MongoTemplate
@@ -86,4 +88,18 @@ class UserRepository(
 
         return aggregationResult.mappedResults[0].categoryCount
     }
+
+    fun removeUser(userId: String) {
+        val removeQuery: Query = Query(
+            Criteria.where(userIdField).`is`(userId)
+        )
+        val removeResult: DeleteResult = mongoTemplate.remove(removeQuery, User::class.java)
+
+        if (!removeResult.wasAcknowledged() || removeResult.deletedCount != 1L) {
+            logger.error("Cannot remove user: ${userId}!")
+            logger.error("Query: $removeQuery")
+            logger.error("Result: $removeResult")
+            throw UnknownErrorException("Remove failed for userId: $userId")
+        }
+    }
 }

src/main/kotlin/com/mptsix/todaydiary/service/UserService.kt

@@ -122,4 +122,8 @@ class UserService(
         }
         userRepository.addUser(user)
     }
+
+    fun removeUser(userToken: String) {
+        userRepository.removeUser(getUserIdFromToken(userToken))
+    }
 }

src/test/kotlin/com/mptsix/todaydiary/controller/UserControllerTest.kt

@@ -188,4 +188,18 @@ internal class UserControllerTest {
 
         assertThat(responseEntity.statusCode).isEqualTo(HttpStatus.NO_CONTENT)
     }
+
+    @Test
+    fun is_removeUser_works_well() {
+        val loginToken: String = loginUser()
+        val url: String = "${serverUrl}/api/v1/user"
+        val httpHeaders: HttpHeaders = HttpHeaders().apply {
+            put("X-AUTH-TOKEN", listOf(loginToken))
+        }
+
+        val responseEntity: ResponseEntity<Unit> =
+            restTemplate.exchange(url, HttpMethod.DELETE, HttpEntity<Unit>(httpHeaders))
+
+        assertThat(responseEntity.statusCode).isEqualTo(HttpStatus.NO_CONTENT)
+    }
 }

src/test/kotlin/com/mptsix/todaydiary/data/UserRepositoryTest.kt

@@ -5,6 +5,7 @@ import com.mptsix.todaydiary.data.user.UserRepository
 import com.mptsix.todaydiary.data.user.journal.Journal
 import com.mptsix.todaydiary.data.user.journal.JournalCategory
 import com.mptsix.todaydiary.error.exception.NotFoundException
+import com.mptsix.todaydiary.error.exception.UnknownErrorException
 import org.assertj.core.api.Assertions.assertThat
 import org.assertj.core.api.Assertions.fail
 import org.junit.jupiter.api.AfterEach
@@ -14,6 +15,7 @@ import org.junit.jupiter.api.extension.ExtendWith
 import org.springframework.beans.factory.annotation.Autowired
 import org.springframework.boot.test.context.SpringBootTest
 import org.springframework.data.mongodb.core.MongoTemplate
+import org.springframework.data.mongodb.core.findAll
 import org.springframework.data.mongodb.core.query.Query
 import org.springframework.data.mongodb.core.remove
 import org.springframework.test.context.junit.jupiter.SpringExtension
@@ -172,4 +174,39 @@ internal class UserRepositoryTest {
         }
     }
 
+    @Test
+    fun is_removeUser_works_well() {
+        val mockUserWithJournal: User = User(
+            userId = "KangDroid",
+            userPassword = "test",
+            userName = "KDR",
+            userDateOfBirth = "WHENEVER",
+            userPasswordAnswer = "WHAT",
+            userPasswordQuestion = "WHAT",
+            journalData = mutableListOf()
+        )
+        userRepository.addUser(mockUserWithJournal)
+
+        runCatching {
+            userRepository.removeUser(mockUserWithJournal.userId)
+        }.onFailure {
+            println(it.stackTraceToString())
+            fail("Should be succeed!")
+        }.onSuccess {
+            val userList: List<User> = mongoTemplate.findAll()
+            assertThat(userList.isEmpty()).isEqualTo(true)
+        }
+    }
+
+    @Test
+    fun is_removeUser_throws_unknownerror() {
+        runCatching {
+            userRepository.removeUser("mockUserWithJournal.userId")
+        }.onFailure {
+            assertThat(it is UnknownErrorException).isEqualTo(true)
+        }.onSuccess {
+            fail("User list is empty but it succeed?")
+        }
+    }
+
 }

src/test/kotlin/com/mptsix/todaydiary/service/UserServiceTest.kt

@@ -21,6 +21,7 @@ import org.junit.jupiter.api.extension.ExtendWith
 import org.springframework.beans.factory.annotation.Autowired
 import org.springframework.boot.test.context.SpringBootTest
 import org.springframework.data.mongodb.core.MongoTemplate
+import org.springframework.data.mongodb.core.findAll
 import org.springframework.data.mongodb.core.query.Query
 import org.springframework.data.mongodb.core.remove
 import org.springframework.mock.web.MockMultipartFile
@@ -309,4 +310,13 @@ internal class UserServiceTest {
         val user: User = userRepository.findByUserId(mockUser.userId)
         assertThat(user.userPassword).isEqualTo("whatever")
     }
+
+    @Test
+    fun is_removeUser_works_well() {
+        val loginToken: String = loginUser()
+        userService.removeUser(loginToken)
+
+        val userList: List<User> = mongoTemplate.findAll()
+        assertThat(userList.isEmpty()).isEqualTo(true)
+    }
 }