package zeptodns.protocol.fluent;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;
import zeptodns.protocol.messages.Message;

@RunWith(JUnit4.class)
public class MessageBuilderTest {
    @Test
    public void thatMessageBuilderQueryIsValid() {
        Message message = MessageBuilder
                .begin()
                .asQuery()
                .end();

        // TODO
    }
}

@@ -159,16 +159,92 @@
 Message;
+%0Aimport zeptodns.protocol.wire.FlagUtils;%0A%0Aimport static org.junit.Assert.*;
 %0A%0A@RunWi
@@ -485,15 +485,375 @@
     
-// TODO
+assertEquals(FlagUtils.getQueryResponse(message.getHeaderSection().getFlags()), 0);%0A    %7D%0A%0A    @Test%0A    public void thatMessageBuilderResponseIsValid() %7B%0A        Message message = MessageBuilder%0A                .begin()%0A                .asResponse()%0A                .end();%0A%0A        assertEquals(FlagUtils.getQueryResponse(message.getHeaderSection().getFlags()), 1);
 %0A   

Test MessaeTypeStep