示例#1
0
文件: file_tests.py 项目: quad/pif
class FileIndexNullTests(unittest.TestCase):
    """Uninitialized FileIndex API Tests."""

    def setUp(self):
        self.index_fn = tempfile.mktemp()  # OK to use as we're just testing...
        self.index = FileIndex(self.index_fn)

    def test_no_hashes(self):
        """Empty FileIndex is empty"""
        assert not self.index.keys()
示例#2
0
文件: file_tests.py 项目: quad/pif
    def setUp(self):
        self.tempdir = tempfile.mkdtemp()
        self.index = FileIndex(os.path.join(self.tempdir, 'index'))

        imagedir = os.path.join(DATA, 'images')
        precalc_shs = json.load(file(os.path.join(imagedir, 'shorthashes.json')))

        self.shorthashes = {}

        for fn, sh in precalc_shs.iteritems():
            src = os.path.join(imagedir, fn)
            dest = os.path.join(self.tempdir, fn)

            shutil.copy(src, dest)

            self.shorthashes[dest] = sh
            self.index[dest]
示例#3
0
    def setUp(self):
        self.tempdir = tempfile.mkdtemp()
        self.index = FileIndex(os.path.join(self.tempdir, 'index'))

        # Build some test images!
        self.shorthashes = {}

        for num, meta in enumerate(self.FILES.iteritems()):
            basename, format = meta
            fn = os.path.join(self.tempdir, basename)

            i = PIL.Image.new('RGB', (self.IMAGE_WIDTH, self.IMAGE_HEIGHT))

            # Draw something different so the hash is unique.
            d = PIL.ImageDraw.ImageDraw(i)
            d.text((0, 0), "Test %u" % num)

            i.save(fn)

            # Gather the metadata to create the shorthash.
            statinfo = os.stat(fn)

            with file(fn) as f:
                f.seek(-pif.TAILHASH_SIZE, 2)
                tailhash = f.read()

            self.shorthashes[fn] = pif.make_shorthash(
                tailhash,
                format,
                statinfo.st_size,
                self.IMAGE_WIDTH,
                self.IMAGE_HEIGHT,
            )

            # Load the image into the cache.
            self.index[fn]
示例#4
0
文件: file_tests.py 项目: quad/pif
class FileIndexSmallDirTests(unittest.TestCase):
    """FileIndex tests with a small test fixture."""

    IMAGE_WIDTH, IMAGE_HEIGHT = 100, 50

    def setUp(self):
        self.tempdir = tempfile.mkdtemp()
        self.index = FileIndex(os.path.join(self.tempdir, 'index'))

        imagedir = os.path.join(DATA, 'images')
        precalc_shs = json.load(file(os.path.join(imagedir, 'shorthashes.json')))

        self.shorthashes = {}

        for fn, sh in precalc_shs.iteritems():
            src = os.path.join(imagedir, fn)
            dest = os.path.join(self.tempdir, fn)

            shutil.copy(src, dest)

            self.shorthashes[dest] = sh
            self.index[dest]

    def tearDown(self):
        shutil.rmtree(self.tempdir)

    def test_add(self):
        """Images able to be added to FileIndex"""

        assert len(self.index) == len(self.shorthashes), "%s != %s" % (self.index, self.shorthashes)

    def test_rescan(self):
        """FileIndex detects changed images"""

        fn = self.index.keys().pop()
        old_shorthash = self.index[fn]

        # Draw on top of the original image to change the hash.
        i = PIL.Image.new('RGB', (999, 999), color=2)

        d = PIL.ImageDraw.ImageDraw(i)
        d.text((0, 0), 'Something new!')

        time.sleep(1)   # The timestamp must tick.
        i.save(fn)

        assert self.index[fn] != old_shorthash

    @raises(KeyError)
    def test_add_file_invalid(self):
        """Invalid images ignored by FileIndex"""

        b_fn = os.path.join(self.tempdir, 'badfile.png')

        with file(b_fn, 'w') as bf:
            bf.write('abc123')

        self.index[b_fn]

    def test_adds(self):
        """FileIndex calculates shorthashes correctly"""

        for fn in self.shorthashes:
            assert fn in self.index
            assert self.index[fn] == self.shorthashes[fn], \
                    "Index %s\nTest %s" % (
                        repr(self.index[fn]),
                        repr(self.shorthashes[fn]),
                    )
示例#5
0
文件: file_tests.py 项目: quad/pif
 def setUp(self):
     self.index_fn = tempfile.mktemp()  # OK to use as we're just testing...
     self.index = FileIndex(self.index_fn)
示例#6
0
class FileIndexSmallDirTests(unittest.TestCase):
    """FileIndex tests with a small test fixture."""

    IMAGE_WIDTH, IMAGE_HEIGHT = 100, 50

    FILES = {
        'abc123.jpeg': 'jpeg',
        'test.jpg': 'jpeg',
        'superjoe.png': 'png',
        'xyzzy.GIF': 'gif',
    }

    def setUp(self):
        self.tempdir = tempfile.mkdtemp()
        self.index = FileIndex(os.path.join(self.tempdir, 'index'))

        # Build some test images!
        self.shorthashes = {}

        for num, meta in enumerate(self.FILES.iteritems()):
            basename, format = meta
            fn = os.path.join(self.tempdir, basename)

            i = PIL.Image.new('RGB', (self.IMAGE_WIDTH, self.IMAGE_HEIGHT))

            # Draw something different so the hash is unique.
            d = PIL.ImageDraw.ImageDraw(i)
            d.text((0, 0), "Test %u" % num)

            i.save(fn)

            # Gather the metadata to create the shorthash.
            statinfo = os.stat(fn)

            with file(fn) as f:
                f.seek(-pif.TAILHASH_SIZE, 2)
                tailhash = f.read()

            self.shorthashes[fn] = pif.make_shorthash(
                tailhash,
                format,
                statinfo.st_size,
                self.IMAGE_WIDTH,
                self.IMAGE_HEIGHT,
            )

            # Load the image into the cache.
            self.index[fn]

    def tearDown(self):
        shutil.rmtree(self.tempdir)

    def test_add(self):
        """Images able to be added to FileIndex"""

        assert len(self.index) == len(self.FILES)

    def test_rescan(self):
        """FileIndex detects changed images"""

        fn = self.index.keys().pop()
        old_shorthash = self.index[fn]

        # Draw on top of the original image to change the hash.
        i = PIL.Image.new('RGB', (999, 999), color=2)

        d = PIL.ImageDraw.ImageDraw(i)
        d.text((0, 0), 'Something new!')

        time.sleep(1)   # The timestamp must tick.
        i.save(fn)

        assert self.index[fn] != old_shorthash

    @raises(KeyError)
    def test_add_file_invalid(self):
        """Invalid images ignored by FileIndex"""

        b_fn = os.path.join(self.tempdir, 'badfile.png')

        with file(b_fn, 'w') as bf:
            bf.write('abc123')

        self.index[b_fn]

    def test_adds(self):
        """FileIndex calculates shorthashes correctly"""

        images = (os.path.join(self.tempdir, fn) for fn in self.FILES)

        for fn in images:
            assert fn in self.index

            shorthash = self.index[fn]
            assert self.index[fn] == self.shorthashes[fn], \
                    "Index %s\nTest %s" % (
                        repr(self.index[fn]),
                        repr(self.shorthashes[fn]),
                    )