diff --git a/src/test/java/org/bitlet/wetorrent/MetafileTest.java b/src/test/java/org/bitlet/wetorrent/MetafileTest.java index 7e812e1..4d0945d 100644 --- a/src/test/java/org/bitlet/wetorrent/MetafileTest.java +++ b/src/test/java/org/bitlet/wetorrent/MetafileTest.java @@ -20,6 +20,8 @@ import org.junit.Test; +import java.io.ByteArrayInputStream; +import java.io.ByteArrayOutputStream; import java.io.InputStream; import java.util.Collections; import java.util.LinkedList; @@ -38,7 +40,6 @@ public void readMetafile() throws Exception { InputStream stream = MetafileTest.class.getResourceAsStream(TEST_TORRENT_PATH); Metafile metafile = new Metafile(stream); assertEquals("http://torrent.ubuntu.com:6969/announce", metafile.getAnnounce()); - List announceList = new LinkedList(); List firstList = new LinkedList(); firstList.add(toByteBuffer("http://torrent.ubuntu.com:6969/announce")); @@ -57,4 +58,16 @@ public void readMetafile() throws Exception { assertEquals(524288L, metafile.getPieceLength().longValue()); assertEquals(2196, metafile.getPieces().size()); } + + @Test + public void updateManifestComment() throws Exception { + InputStream stream = MetafileTest.class.getResourceAsStream(TEST_TORRENT_PATH); + Metafile metafile = new Metafile(stream); + metafile.setComment("foo"); + ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); + metafile.print(outputStream); + ByteArrayInputStream inputStream = new ByteArrayInputStream(outputStream.toByteArray()); + Metafile metafileWithUpdatedFields = new Metafile(inputStream); + assertEquals("foo", metafile.getComment()); + } }