[tor-commits] [stem/master] Integ test for the set_processed_files() method

[atagar at torproject.org]

commit ee5fa60dfb96b4ae0bc8f5a18aa244b4b5f296c8
Author: Damian Johnson <atagar at torproject.org>
Date:   Sun Mar 11 13:58:00 2012 -0700

    Integ test for the set_processed_files() method
    
    Testing that the DescriptorReader's set_processed_files() method properly
    causes us to skip priorly read files.
---
 test/integ/descriptor/reader.py |   24 ++++++++++++++++++++++++
 1 files changed, 24 insertions(+), 0 deletions(-)

diff --git a/test/integ/descriptor/reader.py b/test/integ/descriptor/reader.py
index 2568262..e8a6c20 100644
--- a/test/integ/descriptor/reader.py
+++ b/test/integ/descriptor/reader.py
@@ -3,6 +3,7 @@ Integration tests for stem.descriptor.reader.
 """
 
 import os
+import sys
 import time
 import signal
 import unittest
@@ -187,4 +188,27 @@ class TestDescriptorReader(unittest.TestCase):
         pass
     
     self.assertEquals(expected_results, reader.get_processed_files())
+  
+  def test_set_processed_files(self):
+    """
+    Checks that calling set_processed_files() prior to reading makes us skip
+    those files.
+    """
+    
+    # path and file contents that we want the DescriptorReader to skip
+    skip_file = os.path.join(DESCRIPTOR_TEST_DATA, "example_descriptor")
+    
+    with open(skip_file) as descriptor_file:
+      skip_contents = descriptor_file.read()
+    
+    initial_processed_files = {skip_file: sys.maxint}
+    
+    reader = stem.descriptor.reader.DescriptorReader([DESCRIPTOR_TEST_DATA])
+    reader.set_processed_files(initial_processed_files)
+    self.assertEquals(initial_processed_files, reader.get_processed_files())
+    
+    with reader:
+      for descriptor in reader:
+        if str(descriptor) == skip_contents:
+          self.fail() # we read the file that we were trying to skip