dupeguru/core_me/tests/scanner_test.py

# -*- coding: utf-8 -*-
# Created By: Virgil Dupras
# Created On: 2009-10-23
# Copyright 2010 Hardcoded Software (http://www.hardcoded.net)
# 
# This software is licensed under the "HS" License as described in the "LICENSE" file, 
# which should be included with this package. The terms are also available at 
# http://www.hardcoded.net/licenses/hs_license

from hsutil import io
from hsutil.path import Path
from hsutil.testcase import TestCase

from core.engine import getwords
from ..scanner import *

class NamedObject(object):
    def __init__(self, name="foobar", size=1):
        self.name = name
        self.size = size
        self.path = Path('')
        self.words = getwords(name)
    

no = NamedObject

class ScannerTestFakeFiles(TestCase):
    def setUp(self):
        # This is a hack to avoid invalidating all previous tests since the scanner started to test
        # for file existence before doing the match grouping.
        self.mock(io, 'exists', lambda _: True)
    
    def test_priorize_me(self):
        # in ScannerME, bitrate goes first (right after is_ref) in priorization
        s = ScannerME()
        o1, o2 = no('foo'), no('foo')
        o1.bitrate = 1
        o2.bitrate = 2
        [group] = s.GetDupeGroups([o1, o2])
        assert group.ref is o2