1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
"""
Unit tests for stem.descriptor.reader.
"""
import io
import unittest
import stem.descriptor.reader
try:
# added in python 3.3
from unittest.mock import patch
except ImportError:
from mock import patch
class TestDescriptorReader(unittest.TestCase):
@patch('stem.descriptor.reader.open', create = True)
def test_load_processed_files(self, open_mock):
"""
Successful load of content.
"""
test_lines = (
u'/dir/ 0',
u'/dir/file 12345',
u'/dir/file with spaces 7138743',
u' /dir/with extra space 12345 ',
u' \t ',
u'',
u'/dir/after empty line 12345',
)
expected_value = {
'/dir/': 0,
'/dir/file': 12345,
'/dir/file with spaces': 7138743,
'/dir/with extra space': 12345,
'/dir/after empty line': 12345,
}
open_mock.return_value = io.StringIO(u'\n'.join(test_lines))
self.assertEquals(expected_value, stem.descriptor.reader.load_processed_files(''))
@patch('stem.descriptor.reader.open', create = True)
def test_load_processed_files_empty(self, open_mock):
"""
Tests the load_processed_files() function with an empty file.
"""
open_mock.return_value = io.StringIO(u'')
self.assertEquals({}, stem.descriptor.reader.load_processed_files(''))
@patch('stem.descriptor.reader.open', create = True)
def test_load_processed_files_no_file(self, open_mock):
"""
Tests the load_processed_files() function content that is malformed because
it is missing the file path.
"""
open_mock.return_value = io.StringIO(u' 12345')
self.assertRaises(TypeError, stem.descriptor.reader.load_processed_files, '')
@patch('stem.descriptor.reader.open', create = True)
def test_load_processed_files_no_timestamp(self, open_mock):
"""
Tests the load_processed_files() function content that is malformed because
it is missing the timestamp.
"""
open_mock.return_value = io.StringIO(u'/dir/file ')
self.assertRaises(TypeError, stem.descriptor.reader.load_processed_files, '')
@patch('stem.descriptor.reader.open', create = True)
def test_load_processed_files_malformed_file(self, open_mock):
"""
Tests the load_processed_files() function content that is malformed because
it has an invalid file path.
"""
open_mock.return_value = io.StringIO(u'not_an_absolute_file 12345')
self.assertRaises(TypeError, stem.descriptor.reader.load_processed_files, '')
@patch('stem.descriptor.reader.open', create = True)
def test_load_processed_files_malformed_timestamp(self, open_mock):
"""
Tests the load_processed_files() function content that is malformed because
it has a non-numeric timestamp.
"""
open_mock.return_value = io.StringIO(u'/dir/file 123a')
self.assertRaises(TypeError, stem.descriptor.reader.load_processed_files, '')
|
