package com.juick.www; import com.gargoylesoftware.htmlunit.Page; import com.gargoylesoftware.htmlunit.WebClient; import com.gargoylesoftware.htmlunit.html.HtmlPage; import com.juick.Message; import com.juick.User; import com.juick.configuration.MockDataConfiguration; import com.juick.service.MessagesService; import com.juick.service.UserService; import com.juick.tests.util.MockUtils; import com.juick.www.configuration.WwwAppConfiguration; import com.juick.www.configuration.WwwServletConfiguration; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Import; import org.springframework.test.context.ContextConfiguration; import org.springframework.test.context.junit4.SpringJUnit4ClassRunner; import org.springframework.test.context.web.WebAppConfiguration; import org.springframework.test.web.servlet.htmlunit.MockMvcWebClientBuilder; import org.springframework.web.context.WebApplicationContext; import javax.inject.Inject; import java.io.IOException; import java.util.Collections; import java.util.Optional; import static org.hamcrest.MatcherAssert.assertThat; import static org.hamcrest.Matchers.equalTo; import static org.mockito.Mockito.when; /** * Created by vitalyster on 12.01.2017. */ @RunWith(SpringJUnit4ClassRunner.class) @WebAppConfiguration @ContextConfiguration public class WebAppTests { @Configuration @Import(value = {WwwServletConfiguration.class, WwwAppConfiguration.class, MockDataConfiguration.class}) static class Config {} @Inject private WebApplicationContext wac; @Inject private WebApp webApp; private WebClient webClient; @Inject UserService userService; @Inject MessagesService messagesService; @Before public void setup() { webClient = MockMvcWebClientBuilder.webAppContextSetup(this.wac).build(); } @Test public void postWithoutTagsShouldNotHaveAsteriskInTitle() throws Exception { String ugnichName = "ugnich"; String ugnichPassword = "MyPassw0rd!"; String msgText = "Привет, я - Угнич"; String hash = "12345678"; User user = MockUtils.mockUser(1, ugnichName, ugnichPassword); Message msg = MockUtils.mockMessage(1, user, msgText); when(userService.getUIDbyName(ugnichName)) .thenReturn(1); when(userService.getUserByName(ugnichName)) .thenReturn(user); when(userService.getUserByUID(1)) .thenReturn(Optional.of(user)); when(userService.getFullyUserByName(ugnichName)) .thenReturn(user); when(messagesService.getMyFeed(1, 0)) .thenReturn(Collections.singletonList(1)); when(messagesService.getMessages(Collections.singletonList(1))) .thenReturn(Collections.singletonList(msg)); when(userService.getUIDbyHash(hash)) .thenReturn(1); when(messagesService.getMessageAuthor(1)).thenReturn(user); when(messagesService.canViewThread(1, 0)).thenReturn(true); when(messagesService.getMessage(1)).thenReturn(msg); HtmlPage threadPage = webClient.getPage("http://localhost:8080/ugnich/1"); assertThat(threadPage.getTitleText(), equalTo("ugnich:")); } @Test public void bannedUserBlogandPostShouldReturn404() throws IOException { String userName = "isilmine"; String userPassword = "secret"; String msgText = "автор этого поста был забанен"; String hash = "12345678"; User user = MockUtils.mockUser(2, userName, userPassword); user.setBanned(true); Message msg = MockUtils.mockMessage(2, user, msgText); when(userService.getUIDbyName(userName)) .thenReturn(2); when(userService.getUserByName(userName)) .thenReturn(user); when(userService.getUserByUID(2)) .thenReturn(Optional.of(user)); when(userService.getFullyUserByName(userName)) .thenReturn(user); when(messagesService.getMyFeed(2, 0)) .thenReturn(Collections.singletonList(2)); when(messagesService.getMessages(Collections.singletonList(2))) .thenReturn(Collections.singletonList(msg)); when(userService.getUIDbyHash(hash)) .thenReturn(1); when(messagesService.getMessageAuthor(2)).thenReturn(user); when(messagesService.canViewThread(2, 0)).thenReturn(true); when(messagesService.getMessage(2)).thenReturn(msg); webClient.getOptions().setThrowExceptionOnFailingStatusCode(false); Page blogPage = webClient.getPage("http://localhost:8080/isilmine"); Page threadPage = webClient.getPage("http://localhost:8080/isilmine/2"); assertThat(blogPage.getWebResponse().getStatusCode(), equalTo(404)); assertThat(threadPage.getWebResponse().getStatusCode(), equalTo(404)); } }