Examples with Message codeu.model.data.Message used on opensource projects

Example 36 with Message

use of codeu.model.data.Message in project CodeU-Spring-2018 by maksymko.

the class ChatServlet method doGet.

/**
 * This function fires when a user navigates to the chat page. It gets the conversation title from
 * the URL, finds the corresponding Conversation, and fetches the messages in that Conversation.
 * It then forwards to chat.jsp for rendering.
 */
@Override
public void doGet(HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException {
    String requestUrl = request.getRequestURI();
    String conversationTitle = requestUrl.substring("/chat/".length());
    Conversation conversation = conversationStore.getConversationWithTitle(conversationTitle);
    if (conversation == null) {
        // couldn't find conversation, redirect to conversation list
        System.out.println("Conversation was null: " + conversationTitle);
        response.sendRedirect("/conversations");
        return;
    }
    UUID conversationId = conversation.getId();
    List<Message> messages = messageStore.getMessagesInConversation(conversationId);
    request.setAttribute("conversation", conversation);
    request.setAttribute("messages", messages);
    request.getRequestDispatcher("/WEB-INF/view/chat.jsp").forward(request, response);
}

Example 37 with Message

use of codeu.model.data.Message in project CodeU-Spring-2018 by maksymko.

the class ChatServlet method doPost.

/**
 * This function fires when a user submits the form on the chat page. It gets the logged-in
 * username from the session, the conversation title from the URL, and the chat message from the
 * submitted form data. It creates a new Message from that data, adds it to the model, and then
 * redirects back to the chat page.
 */
@Override
public void doPost(HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException {
    String username = (String) request.getSession().getAttribute("user");
    if (username == null) {
        // user is not logged in, don't let them add a message
        response.sendRedirect("/login");
        return;
    }
    User user = userStore.getUser(username);
    if (user == null) {
        // user was not found, don't let them add a message
        response.sendRedirect("/login");
        return;
    }
    String requestUrl = request.getRequestURI();
    String conversationTitle = requestUrl.substring("/chat/".length());
    Conversation conversation = conversationStore.getConversationWithTitle(conversationTitle);
    if (conversation == null) {
        // couldn't find conversation, redirect to conversation list
        response.sendRedirect("/conversations");
        return;
    }
    String messageContent = request.getParameter("message");
    // this removes any HTML from the message content
    String cleanedMessageContent = Jsoup.clean(messageContent, Whitelist.none());
    Message message = new Message(UUID.randomUUID(), conversation.getId(), user.getId(), cleanedMessageContent, Instant.now());
    messageStore.addMessage(message);
    // redirect to a GET request
    response.sendRedirect("/chat/" + conversationTitle);
}

Example 38 with Message

use of codeu.model.data.Message in project CodeU-Spring-2018 by jwang281.

the class ChatServletTest method testDoPost_CleansHtmlContent.

@Test
public void testDoPost_CleansHtmlContent() throws IOException, ServletException {
    Mockito.when(mockRequest.getRequestURI()).thenReturn("/chat/test_conversation");
    Mockito.when(mockSession.getAttribute("user")).thenReturn("test_username");
    User fakeUser = new User(UUID.randomUUID(), "test_username", "test_password", Instant.now());
    Mockito.when(mockUserStore.getUser("test_username")).thenReturn(fakeUser);
    Conversation fakeConversation = new Conversation(UUID.randomUUID(), UUID.randomUUID(), "test_conversation", Instant.now());
    Mockito.when(mockConversationStore.getConversationWithTitle("test_conversation")).thenReturn(fakeConversation);
    Mockito.when(mockRequest.getParameter("message")).thenReturn("Contains <b>html</b> and <script>JavaScript</script> content.");
    chatServlet.doPost(mockRequest, mockResponse);
    ArgumentCaptor<Message> messageArgumentCaptor = ArgumentCaptor.forClass(Message.class);
    Mockito.verify(mockMessageStore).addMessage(messageArgumentCaptor.capture());
    Assert.assertEquals("Contains html and  content.", messageArgumentCaptor.getValue().getContent());
    Mockito.verify(mockResponse).sendRedirect("/chat/test_conversation");
}

Example 39 with Message

use of codeu.model.data.Message in project CodeU-Spring-2018 by jwang281.

the class PersistentStorageAgentTest method testWriteThroughMessage.

@Test
public void testWriteThroughMessage() {
    Message message = new Message(UUID.randomUUID(), UUID.randomUUID(), UUID.randomUUID(), "test content", Instant.now());
    persistentStorageAgent.writeThrough(message);
    Mockito.verify(mockPersistentDataStore).writeThrough(message);
}