def setUp(self):
""" some test data, a temp dir and the pixget object """
self.good_ct = 'images/jpg'
self.bad_ct = 'text/html'
self.good_url = 'http://somewhere.com/image.jpg'
self.bad_url = 'http:somewhere.com/image.jpg'
self.tmpdir = tempfile.mkdtemp()
self.pixget = PixGet('input.txt', self.tmpdir)
def test_read_infile(self):
## one good, one bad
infile = self._setUp_infile([self.good_url, self.bad_url])
self.pixget = PixGet(infile, self.tmpdir)
self.pixget._read_infile()
self.assertEquals(len(self.pixget.valid_lines), 1)
self.assertEquals(len(self.pixget.invalid_lines), 1)
## two good
infile = self._setUp_infile([self.good_url, self.good_url])
self.pixget = PixGet(infile, self.tmpdir)
self.pixget._read_infile()
self.assertEquals(len(self.pixget.valid_lines), 2)
self.assertEquals(len(self.pixget.invalid_lines), 0)
## two bad
infile = self._setUp_infile([self.bad_url, self.bad_url])
self.pixget = PixGet(infile, self.tmpdir)
self.pixget._read_infile()
self.assertEquals(len(self.pixget.valid_lines), 0)
self.assertEquals(len(self.pixget.invalid_lines), 2)
def test_read_infile_with_whitespaces(self):
infile = self._setUp_infile([' %s ' % (self.good_url)])
self.pixget = PixGet(infile, self.tmpdir)
self.pixget._read_infile()
self.assertEquals(len(self.pixget.valid_lines), 1)
self.assertEquals(len(self.pixget.invalid_lines), 0)
def test_read_infile_empty_lines(self):
infile = self._setUp_infile(['', ''])
self.pixget = PixGet(infile, self.tmpdir)
self.pixget._read_infile()
self.assertEquals(len(self.pixget.valid_lines), 0)
self.assertEquals(len(self.pixget.invalid_lines), 0)
