From 8ee55a32f54c8b6d5fb2b083b8dedc9415da90a3 Mon Sep 17 00:00:00 2001
From: Alfredo Deza <alfredo@deza.pe>
Date: Fri, 12 Dec 2014 11:09:27 -0500
Subject: [PATCH] new tests for the new pseudo-random content files

Signed-off-by: Alfredo Deza <alfredo@deza.pe>
---
 s3tests/tests/test_realistic.py | 64 +++++++++++++++++++++++++--------
 1 file changed, 49 insertions(+), 15 deletions(-)

diff --git a/s3tests/tests/test_realistic.py b/s3tests/tests/test_realistic.py
index 52f2529..9b313c9 100644
--- a/s3tests/tests/test_realistic.py
+++ b/s3tests/tests/test_realistic.py
@@ -1,4 +1,4 @@
-import realistic
+from s3tests import realistic
 import shutil
 import tempfile
 
@@ -17,19 +17,20 @@ class TestFiles(object):
     # the size and seed is what we can get when generating a bunch of files
     # with pseudo random numbers based on sttdev, seed, and mean.
 
-    # this fails, demonstrating the problem
-    def test_random_file_invalid(self):
-        size = 2506764
-        seed = 3391518755
-        source = realistic.RandomContentFile(size=size, seed=seed)
-        t = tempfile.SpooledTemporaryFile()
-        shutil.copyfileobj(source, t)
-        precomputed = realistic.PrecomputedContentFile(t)
+    # this fails, demonstrating the (current) problem
+    #def test_random_file_invalid(self):
+    #    size = 2506764
+    #    seed = 3391518755
+    #    source = realistic.RandomContentFile(size=size, seed=seed)
+    #    t = tempfile.SpooledTemporaryFile()
+    #    shutil.copyfileobj(source, t)
+    #    precomputed = realistic.PrecomputedContentFile(t)
+    #    assert precomputed.valid()
 
-        verifier = realistic.FileVerifier()
-        shutil.copyfileobj(precomputed, verifier)
+    #    verifier = realistic.FileVerifier()
+    #    shutil.copyfileobj(precomputed, verifier)
 
-        assert verifier.valid()
+    #    assert verifier.valid()
 
     # this passes
     def test_random_file_valid(self):
@@ -39,9 +40,42 @@ class TestFiles(object):
         t = tempfile.SpooledTemporaryFile()
         shutil.copyfileobj(source, t)
         precomputed = realistic.PrecomputedContentFile(t)
+        assert precomputed.valid()
 
-        verifier = realistic.FileVerifier()
-        shutil.copyfileobj(precomputed, verifier)
+        #verifier = realistic.FileVerifier()
+        #shutil.copyfileobj(precomputed, verifier)
 
-        assert verifier.valid()
+        #assert verifier.valid()
+        #assert 0
 
+
+# new implementation
+class TestFileValidator(object):
+
+    def test_new_file_is_valid(self):
+        size = 2506001
+        contents = realistic.generate_file_contents(size)
+        t = tempfile.SpooledTemporaryFile()
+        t.write(contents)
+        t.seek(0)
+        fp = realistic.FileValidator(t)
+        assert fp.valid()
+
+    def test_new_file_is_valid_when_size_is_1(self):
+        size = 1
+        contents = realistic.generate_file_contents(size)
+        t = tempfile.SpooledTemporaryFile()
+        t.write(contents)
+        t.seek(0)
+        fp = realistic.FileValidator(t)
+        assert fp.valid()
+
+    def test_new_file_is_valid_on_several_calls(self):
+        size = 2506001
+        contents = realistic.generate_file_contents(size)
+        t = tempfile.SpooledTemporaryFile()
+        t.write(contents)
+        t.seek(0)
+        fp = realistic.FileValidator(t)
+        assert fp.valid()
+        assert fp.valid()