ChatDocumentBuilderTest.java

package org.limewire.ui.swing.friends.chat;

import java.util.ArrayList;

import junit.framework.TestCase;

import org.limewire.ui.swing.friends.chat.Message.Type;
import org.limewire.xmpp.api.client.MockFileMetadata;
import org.limewire.friend.api.ChatState;
import org.limewire.friend.api.FriendPresence.Mode;

public class ChatDocumentBuilderTest extends TestCase {
    public void testBuildChatTextWithLinks() {
        compareOutput("hey http://www.foo.com there",
                      "<div class=\"me\">me:</div>hey <a href=\"http://www.foo.com\">http://www.foo.com</a> there<br/>");
        
        compareOutput("hey foo.com there", 
                      "<div class=\"me\">me:</div>hey <a href=\"http://foo.com\">foo.com</a> there<br/>");
        
        compareOutput("hey www.foo.com there", 
                      "<div class=\"me\">me:</div>hey <a href=\"http://www.foo.com\">www.foo.com</a> there<br/>");

        compareOutput("hey https://www.foo.com there", 
                      "<div class=\"me\">me:</div>hey <a href=\"https://www.foo.com\">https://www.foo.com</a> there<br/>");

        compareOutput("hey www.foo there", 
                      "<div class=\"me\">me:</div>hey www.foo there<br/>");

        compareOutput("hey magnet://foo.bar now", 
        "<div class=\"me\">me:</div>hey <a href=\"magnet://foo.bar\">magnet://foo.bar</a> now<br/>");
    }
    
    public void testBuildChatTextWithMarkup() {
        compareOutput("hey <b>foo</b> there", 
                      "<div class=\"me\">me:</div>hey <b>foo</b> there<br/>");
    }
    
    public void testBuildChatTextForRealMessage() {
        compareOutput("<div class=\"them\">you:</div>hey there<br/>", ChatState.active, 
                new Type[] {Type.RECEIVED}, 
                new String[] {"hey there"});
    }

    public void testBuildChatTextForExternallyInitiatedConversationWhenTheyAreTyping() {
        StringBuilder conversation = new StringBuilder();
        conversation.append("<div class=\"them\">you:</div>heynow<br/>")
                    .append("<div class=\"typing\">you is typing...</div><br/>");
        compareOutput(conversation.toString(), ChatState.composing, 
                new Type[] {Type.RECEIVED},  
                new String[] {"heynow"});
    }

    public void testBuildChatTextForExternallyInitiatedConversationWhenTheyArePaused() {
        StringBuilder conversation = new StringBuilder();
        conversation.append("<div class=\"them\">you:</div>heynow<br/>")
        .append("<div class=\"typing\">you has entered text</div><br/>");
        compareOutput(conversation.toString(), ChatState.paused, 
                new Type[] {Type.RECEIVED},  
                new String[] {"heynow"});
    }

    public void testBuildChatTextForConversationWhenTheyHaveSignedOff() {
        StringBuilder conversation = new StringBuilder();
        conversation.append("<div class=\"them\">you:</div>heynow<br/>")
        .append("<div class=\"typing\">you has signed off</div><br/>");
        compareOutput(conversation.toString(), null, true, 
                new Type[] {Type.RECEIVED},  
                new String[] {"heynow"});
    }

    public void testBuildChatTextShowingTypingMessagesRemovedAfterReceivingFollowupMessage() {
        StringBuilder conversation = new StringBuilder();
        conversation.append("<div class=\"them\">you:</div>heynow<br/>")
                    .append("foobar<br/>");
        compareOutput(conversation.toString(), ChatState.active,
                new Type[] {Type.RECEIVED, Type.RECEIVED},  
                new String[] {"heynow", "foobar"});
    }
    
    public void testBuildChatTextForATwoPartyExchangeThatIsExternallyInitiated() {
        StringBuilder conversation = new StringBuilder();
        conversation.append("<div class=\"them\">you:</div>heynow<br/><br/>")
                    .append("<div class=\"me\">me:</div>yo<br/><br/>")
                    .append("<div class=\"them\">you:</div>fooey<br/>");
        
        compareOutput(conversation.toString(), ChatState.active, 
                new Type[] {Type.RECEIVED, Type.SENT, Type.RECEIVED}, 
                new String[] {"heynow", "yo", "fooey"});
    }
    
    public void testBuildChatTextForALargeMessage() {
        StringBuilder conversation = new StringBuilder();
        conversation.append("<div class=\"them\">you:</div><a href=\"http://gooooooooooooooooooooooooooooooooooooooooooooooooooooooooogle.com/\">http://gooooooooooooooooooooooooooo<wbr>oooooooooooooooooooooooooooooogle.c<wbr>om/<wbr></a><br/><br/>")
                    .append("<div class=\"me\">me:</div>wow cool link<br/><br/>")
                    .append("<div class=\"them\">you:</div>yeah I can't beleive it is an actual site.<br/>");
        
        compareOutput(conversation.toString(), ChatState.active, 
                new Type[] {Type.RECEIVED, Type.SENT, Type.RECEIVED}, 
                new String[] {"http://gooooooooooooooooooooooooooooooooooooooooooooooooooooooooogle.com/", "wow cool link", "yeah I can't beleive it is an actual site."});
    }

    public void testBuildChatTextForMessagesISendMoreThan60SecondsApart() {
        StringBuilder conversation = new StringBuilder();
        conversation.append("<div class=\"me\">me:</div>heynow<br/><br/>")
        .append("<div class=\"me\">me:</div>yo<br/>");
        
        MockChatFriend friend = new MockChatFriend(null, null, Mode.available);
        ArrayList<Message> messages = new ArrayList<Message>();
        messages.add(new MockMessage(friend, "heynow", 0, "me", Type.SENT));      
        messages.add(new MockMessage(friend, "yo", 600001, "me", Type.SENT));
        compareOutput(conversation.toString(), null, false, messages);
    }
    

    /**
     * Testing chat text after receiving file offer
     */
    public void testBuildChatTextAfterBeingOfferedAFile() {
        StringBuilder conversation = new StringBuilder();
        conversation.append("<div class=\"them\">you:</div>myName wants to share a file with you<br/>")
                    .append("<form action=\"\"><input type=\"hidden\" name=\"fileid\" value=\"heynow-fileid\"/><input type=\"submit\" value=\"Download Foo doc.doc\"/></form><br/>")
                    .append("Download it now, or get it from them <a href=\"#library\">later</a>.<br/>");

        ArrayList<Message> messages = new ArrayList<Message>();
        messages.add(new MessageFileOfferImpl("you", "myName", Type.RECEIVED, new MockFileMetadata("heynow-fileid", "Foo doc.doc"), null));

        compareOutput(conversation.toString(), ChatState.active, false, messages);
    }

    /**
     * Testing chat text after sending file offer
     */
    public void testBuildChatTextAfterOfferingAFile() {
        StringBuilder conversation = new StringBuilder();
        conversation.append("<div class=\"me\">you:</div>Sharing file with myName<br/>")
                    .append("<form action=\"\"><input type=\"hidden\" name=\"fileid\" ")
                    .append("value=\"heynow-fileid\"/>")
                    .append("<input type=\"submit\" value=\"Foo doc.doc:disabled\"/></form><br/><br/>");

        ArrayList<Message> messages = new ArrayList<Message>();
        messages.add(new MessageFileOfferImpl("you", "myName", Type.SENT, new MockFileMetadata("heynow-fileid", "Foo doc.doc"), null));
        compareOutput(conversation.toString(), ChatState.active, false, messages);
    }

    private void compareOutput(String input, String expected) {
        compareOutput(expected, ChatState.active, new Type[] {Type.SENT}, input);
    }

    private void compareOutput(String expected, ChatState state, Type[] type, String... input) {
        compareOutput(expected, state, false, getMessages(type, input));
    }

    private void compareOutput(String expected, ChatState state, boolean friendHasSignedOff, Type[] type, String... input) {
        compareOutput(expected, state, friendHasSignedOff, getMessages(type, input));
    }

    private void compareOutput(String expected, ChatState state, boolean friendHasSignedOff, ArrayList<Message> messages) {
        String chatText = ChatDocumentBuilder.buildChatText(messages, state, "you", friendHasSignedOff);
        
        assertTrue(chatText.startsWith(ChatDocumentBuilder.TOP));
        assertTrue(chatText.endsWith(ChatDocumentBuilder.BOTTOM));
        
        chatText = chatText.replace(ChatDocumentBuilder.TOP, "");
        chatText = chatText.replace(ChatDocumentBuilder.BOTTOM, "");
        
        assertEquals(expected, chatText);
    }
    
    private ArrayList<Message> getMessages(Type types[], String... messages) {
        ArrayList<Message> list = new ArrayList<Message>();
        for(int i = 0; i < messages.length; i++) {
            Type type = types[i];
            list.add(new MessageTextImpl(type == Type.SENT ? "me" : "you", null, type, messages[i]));
        }
        return list;
    }
}