src/test/kotlin/net/pterodactylus/sone/web/pages/DeleteAlbumPageTest.kt

   1 package net.pterodactylus.sone.web.pages
   2
   3 import net.pterodactylus.sone.data.Album
   4 import net.pterodactylus.sone.data.Sone
   5 import net.pterodactylus.sone.test.mock
   6 import net.pterodactylus.sone.test.whenever
   7 import net.pterodactylus.util.web.Method.POST
   8 import org.hamcrest.MatcherAssert.assertThat
   9 import org.hamcrest.Matchers.equalTo
  10 import org.junit.Before
  11 import org.junit.Test
  12 import org.mockito.ArgumentMatchers.anyString
  13 import org.mockito.Mockito.verify
  14
  15 /**
  16  * Unit test for [DeleteAlbumPage].
  17  */
  18 class DeleteAlbumPageTest: WebPageTest() {
  19
  20         private val page = DeleteAlbumPage(template, webInterface)
  21
  22         private val sone = mock<Sone>()
  23         private val album = mock<Album>()
  24         private val parentAlbum = mock<Album>()
  25
  26         override fun getPage() = page
  27
  28         @Before
  29         fun setupAlbums() {
  30                 whenever(sone.id).thenReturn("sone-id")
  31                 whenever(sone.isLocal).thenReturn(true)
  32                 whenever(parentAlbum.id).thenReturn("parent-id")
  33                 whenever(parentAlbum.isRoot).thenReturn(true)
  34                 whenever(album.id).thenReturn("album-id")
  35                 whenever(album.sone).thenReturn(sone)
  36                 whenever(album.parent).thenReturn(parentAlbum)
  37                 whenever(sone.rootAlbum).thenReturn(parentAlbum)
  38         }
  39
  40         @Test
  41         fun `page returns correct path`() {
  42                 assertThat(page.path, equalTo("deleteAlbum.html"))
  43         }
  44
  45         @Test
  46         fun `page requires login`() {
  47                 assertThat(page.requiresLogin(), equalTo(true))
  48         }
  49
  50         @Test
  51         fun `get request with invalid album ID results in redirect to invalid page`() {
  52                 whenever(core.getAlbum(anyString())).thenReturn(null)
  53                 verifyRedirect("invalid.html")
  54         }
  55
  56         @Test
  57         fun `get request with valid album ID sets album in template context`() {
  58                 val album = mock<Album>()
  59                 addAlbum("album-id", album)
  60                 addHttpRequestParameter("album", "album-id")
  61                 page.processTemplate(freenetRequest, templateContext)
  62                 assertThat(templateContext["album"], equalTo<Any>(album))
  63         }
  64
  65         @Test
  66         fun `post request redirects to invalid page if album is invalid`() {
  67                 setMethod(POST)
  68                 verifyRedirect("invalid.html")
  69         }
  70
  71         @Test
  72         fun `post request redirects to no permissions page if album is not local`() {
  73                 setMethod(POST)
  74                 whenever(sone.isLocal).thenReturn(false)
  75                 addAlbum("album-id", album)
  76                 addHttpRequestPart("album", "album-id")
  77                 verifyRedirect("noPermission.html")
  78         }
  79
  80         @Test
  81         fun `post request with abort delete parameter set redirects to album browser`() {
  82                 setMethod(POST)
  83                 addAlbum("album-id", album)
  84                 addHttpRequestPart("album", "album-id")
  85                 addHttpRequestPart("abortDelete", "true")
  86                 verifyRedirect("imageBrowser.html?album=album-id")
  87         }
  88
  89         @Test
  90         fun `album is deleted and page redirects to sone if parent album is root album`() {
  91                 setMethod(POST)
  92                 addAlbum("album-id", album)
  93                 addHttpRequestPart("album", "album-id")
  94                 verifyRedirect("imageBrowser.html?sone=sone-id") {
  95                         verify(core).deleteAlbum(album)
  96                 }
  97         }
  98
  99         @Test
 100         fun `album is deleted and page redirects to album if parent album is not root album`() {
 101                 setMethod(POST)
 102                 whenever(parentAlbum.isRoot).thenReturn(false)
 103                 whenever(sone.rootAlbum).thenReturn(mock<Album>())
 104                 addAlbum("album-id", album)
 105                 addHttpRequestPart("album", "album-id")
 106                 verifyRedirect("imageBrowser.html?album=parent-id") {
 107                         verify(core).deleteAlbum(album)
 108                 }
 109         }
 110
 111 }