Exemplo n.º 1
0
 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.")
Exemplo n.º 2
0
 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())
Exemplo n.º 3
0
 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)
Exemplo n.º 4
0
 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())
Exemplo n.º 5
0
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)
Exemplo n.º 6
0
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)
Exemplo n.º 7
0
 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])
Exemplo n.º 8
0
    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())
Exemplo n.º 9
0
 def test_readd_error(self):
     pi = PictureIndex(self.index)
     pi.add(self.mock_pic)
     self.assertRaises(PictureAlreadyIndexedError,
                       pi.add, self.mock_pic)
Exemplo n.º 10
0
 def test_replace_key_error(self):
     pi = PictureIndex(self.index)
     self.assertRaises(KeyError, pi.replace, self.mock_pic)
Exemplo n.º 11
0
 def test_key_error(self):
     pi = PictureIndex(self.index)
     self.assertRaises(KeyError, pi.__getitem__, "unknown")
Exemplo n.º 12
0
 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)
Exemplo n.º 13
0
 def test_pics(self):
     pi = PictureIndex(self.index)
     self.assertSequenceEqual(pi.pics(), sorted(self.index.values()))
Exemplo n.º 14
0
 def test_iterpics(self):
     pi = PictureIndex(self.index)
     self.assertItemsEqual(list(pi.iterpics()), self.index.values())
Exemplo n.º 15
0
 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())
Exemplo n.º 16
0
 def test_add(self):
     pi = PictureIndex(self.index)
     pi.add(self.mock_pic)
     self.assertIn(self.mock_pic, self.index.values())
Exemplo n.º 17
0
 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'}))
Exemplo n.º 18
0
 def test_type(self):
     mock_index = mock.Mock()
     pi = PictureIndex(d=mock_index)
     self.assertIsInstance(pi, PictureIndex)