MessageTest.java

package se.chalmers.gdcn.tests;

import se.chalmers.gdcn.network.NetworkMessage;
import org.testng.annotations.Test;

/**
 * Created by Leif on 2014-03-29.
 */
public class MessageTest {

    @Test
    public void encryptDecryptMessage(){
        NetworkMessage message = new NetworkMessage("SomeData", NetworkMessage.Type.REQUEST);

        Object encrypted = message.encrypt();
        NetworkMessage decrypted = NetworkMessage.decrypt(encrypted);

        assert decrypted.getType() == message.getType();
        assert decrypted.getObject().equals(message.getObject());
    }

}