[tor-commits] [bridgedb/master] Add utility function to test_parse_descriptors for writing test descriptor files.

[isis at torproject.org]

commit bd10fbc05bef64475b6ed3b9f340bd99e1cc2f29
Author: Isis Lovecruft <isis at torproject.org>
Date:   Tue Aug 26 01:50:05 2014 +0000

    Add utility function to test_parse_descriptors for writing test descriptor files.
    
     * ADDS new method:
       test_parse_descriptors.ParseDescriptorsTests.writeTestDescriptorsToFile().
---
 lib/bridgedb/test/test_parse_descriptors.py |   20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/lib/bridgedb/test/test_parse_descriptors.py b/lib/bridgedb/test/test_parse_descriptors.py
index 7c4fc93..df1df13 100644
--- a/lib/bridgedb/test/test_parse_descriptors.py
+++ b/lib/bridgedb/test/test_parse_descriptors.py
@@ -12,6 +12,7 @@
 from __future__ import print_function
 
 import io
+import os
 import textwrap
 
 from twisted.trial import unittest
@@ -28,7 +29,7 @@ else:
     HAS_STEM = True
 
 
-BRIDGE_NETWORKSTATUS = '''\
+BRIDGE_NETWORKSTATUS_0 = '''\
 r OutwitsPlod b6khbPOgbomgMSGswx9w+N/X3cw qBckYSWbwl/F/qzQWwMJWBxHZ+w 2014-03-12 16:07:08 152.78.9.20 17810 0
 a [bfbd:7a90:2347:cc4:e854:64b3:2c31:124f]:17810
 s Fast Guard Running Stable Valid
@@ -140,6 +141,23 @@ class ParseDescriptorsTests(unittest.TestCase):
         if self.skip:
             raise unittest.SkipTest("Couldn't import Stem.")
 
+    def writeTestDescriptorsToFile(self, filename, *descriptors):
+        """Write **descriptors** to **filename**.
+
+        :param str filename: A filename. It will be appended to the current
+            working directory automatically.
+        :param str descriptors: Some optional strings containing
+            descriptors. Each one will be written to **filename** as-is.
+        :rtype: str
+        :returns: The full path to the file which was written to.
+        """
+        descFilename = os.path.join(os.getcwd(), filename)
+        with open(descFilename, 'w') as fh:
+            for desc in descriptors:
+                fh.write(desc)
+                fh.flush()
+        return descFilename
+
     def test_parse_descriptors_parseBridgeDescriptorsFile(self):
         """Test for ``b.p.descriptors.parseBridgeDescriptorsFile``."""
         descFile = io.BytesIO(BRIDGE_SERVER_DESCRIPTOR)