def test_write_read_cycle(self):
index_a = PictureIndex({'pic1': "Picture 1", 'pic2': "Picture 2"})
index_a.write(self.fh)
self.fh.seek(0) # rewind buffer
index_b = PictureIndex()
index_b.read(self.fh)
self.assertEqual(index_a, index_b, "Dumped and read index don't match.")
self.assertIsNot(index_a, index_b, "Dumped and read index is same obj.")
def test_remove_many(self):
pi = PictureIndex(self.index)
mock_pics = [mock.Mock(), mock.Mock(), mock.Mock()]
pi.add(mock_pics)
pi.remove(mock_pics[:2])
self.assertNotIn(mock_pics[0], pi.pics())
self.assertNotIn(mock_pics[1], pi.pics())
self.assertIn(mock_pics[2], pi.pics())
def test_read_error(self):
# creating buffer content that can't be unpickled
corrupt_buf = "42\n"
self.fh.write(corrupt_buf)
self.fh.seek(0) # rewind buffer
pi = PictureIndex()
self.assertRaises(IndexParsingError, pi.read, self.fh)
def test_replace(self):
pi = PictureIndex(self.index)
pi.add(self.mock_pic)
new_mock_pic = mock.Mock()
new_mock_pic.filename = self.mock_pic.filename
pi.replace(new_mock_pic)
self.assertIn(new_mock_pic, self.index.values())
self.assertNotIn(self.mock_pic, self.index.values())
def new_mock_repo(path, num_pics=0):
"""
Return a repository containing MockPictures using MockConnector backend.
Arguments:
num_pics -- Add that many MockPictures to the repository index (default: 0).
Returns:
Repository instance.
"""
connector = MockConnector.from_string(path)
with connector.connected():
# test index
pi = PictureIndex()
pi.add(MockPicture.create_many(num_pics))
# test config
conf = repo.new_repo_config()
conf['index.file'] = ".pic/testindex"
conf['test.test'] = "foo"
return repo.Repo.create_on_disk(connector, conf, pi)
def new_mock_repo(path, num_pics=0):
"""
Return a repository containing MockPictures using MockConnector backend.
Arguments:
num_pics -- Add that many MockPictures to the repository index (default: 0).
Returns:
Repository instance.
"""
connector = MockConnector.from_string(path)
with connector.connected():
# test index
pi = PictureIndex()
pi.add(MockPicture.create_many(num_pics))
# test config
conf = repo.new_repo_config()
conf['index.file'] = ".pic/testindex"
conf['test.test'] = "foo"
return repo.Repo.create_on_disk(connector, conf, pi)
def test_find_by_filename(self):
pi = PictureIndex()
pic1 = mock.Mock()
pic1.get_filenames.return_value = ['pic1_file1', 'pic1_file2']
pic2 = mock.Mock()
pic2.get_filenames.return_value = ['pic2_file1', 'shared_file']
pic3 = mock.Mock()
pic3.get_filenames.return_value = ['pic3_file1', 'shared_file']
pi.add([pic1, pic2, pic3])
self.assertEqual(pi.find_by_filename('pic1_file1'), [pic1])
self.assertItemsEqual(pi.find_by_filename('shared_file'), [pic2, pic3])
def test_create_on_disk_empty(self):
r = Repo.create_on_disk(self.connector, self.conf)
# test repo
self.assertIsInstance(r, Repo)
self.assertIs(r.connector, self.connector)
self.assertIs(r.config, self.conf)
self.assertEqual(r.index, PictureIndex())
# test conf on disk
self.assertTrue(self.connector.opened(repo.CONFIG_FILE))
conf_on_disk = config.Config()
conf_on_disk.read(self.connector.get_file(repo.CONFIG_FILE))
self.assertEqual(conf_on_disk, self.conf)
# test picture index on disk
self.assertTrue(self.connector.opened('mock-index-path'))
index_on_disk = index.PictureIndex()
index_on_disk.read(self.connector.get_file('mock-index-path'))
self.assertEqual(index_on_disk, index.PictureIndex())
def test_readd_error(self):
pi = PictureIndex(self.index)
pi.add(self.mock_pic)
self.assertRaises(PictureAlreadyIndexedError,
pi.add, self.mock_pic)
def test_replace_key_error(self):
pi = PictureIndex(self.index)
self.assertRaises(KeyError, pi.replace, self.mock_pic)
def test_key_error(self):
pi = PictureIndex(self.index)
self.assertRaises(KeyError, pi.__getitem__, "unknown")
def test_get(self):
pi = PictureIndex(self.index)
pi.add(self.mock_pic)
key = self.mock_pic.filename
pic = pi[key]
self.assertEqual(pic, self.mock_pic)
def test_pics(self):
pi = PictureIndex(self.index)
self.assertSequenceEqual(pi.pics(), sorted(self.index.values()))
def test_iterpics(self):
pi = PictureIndex(self.index)
self.assertItemsEqual(list(pi.iterpics()), self.index.values())
def test_add_many(self):
pi = PictureIndex(self.index)
mock_pics = [mock.Mock(), mock.Mock(), mock.Mock()]
pi.add(mock_pics)
for mock_pic in mock_pics:
self.assertIn(mock_pic, pi.pics())
def test_add(self):
pi = PictureIndex(self.index)
pi.add(self.mock_pic)
self.assertIn(self.mock_pic, self.index.values())
def test_equality(self):
self.assertEqual(PictureIndex(self.index), PictureIndex(self.index))
self.assertNotEqual(PictureIndex(self.index), PictureIndex())
self.assertNotEqual(PictureIndex(self.index),
PictureIndex({'pic1': 'file1'}))
def test_type(self):
mock_index = mock.Mock()
pi = PictureIndex(d=mock_index)
self.assertIsInstance(pi, PictureIndex)
