more tests

2019-05-29 00:28:35 +01:00
parent 16bf1e3fac
commit 8ff8e98424
1 changed files with 14 additions and 0 deletions
--- a/core/src/test/groovy/com/muwire/core/download/PiecesTest.groovy
+++ b/core/src/test/groovy/com/muwire/core/download/PiecesTest.groovy
@@ -20,4 +20,18 @@ class PiecesTest {
        pieces.markDownloaded(0)
        assert pieces.isComplete()
    }
+    
+    @Test
+    public void testTwoPieces() {
+        pieces = new Pieces(2)
+        assert !pieces.isComplete()
+        int piece = pieces.getRandomPiece()
+        assert piece == 0 || piece == 1
+        pieces.markDownloaded(piece)
+        assert !pieces.isComplete()
+        int piece2 = pieces.getRandomPiece()
+        assert piece != piece2
+        pieces.markDownloaded(piece2)
+        assert pieces.isComplete()
+    }
 }