dupeguru/core/tests/ignore_test.py

# Created By: Virgil Dupras
# Created On: 2006/05/02
# Copyright 2012 Hardcoded Software (http://www.hardcoded.net)
# 
# This software is licensed under the "BSD" 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/bsd_license

import io
from xml.etree import ElementTree as ET

from pytest import raises
from hscommon.testutil import eq_

from ..ignore import *

def test_empty():
    il = IgnoreList()
    eq_(0,len(il))
    assert not il.AreIgnored('foo','bar')

def test_simple():
    il = IgnoreList()
    il.Ignore('foo','bar')
    assert il.AreIgnored('foo','bar')
    assert il.AreIgnored('bar','foo')
    assert not il.AreIgnored('foo','bleh')
    assert not il.AreIgnored('bleh','bar')
    eq_(1,len(il))

def test_multiple():
    il = IgnoreList()
    il.Ignore('foo','bar')
    il.Ignore('foo','bleh')
    il.Ignore('bleh','bar')
    il.Ignore('aybabtu','bleh')
    assert il.AreIgnored('foo','bar')
    assert il.AreIgnored('bar','foo')
    assert il.AreIgnored('foo','bleh')
    assert il.AreIgnored('bleh','bar')
    assert not il.AreIgnored('aybabtu','bar')
    eq_(4,len(il))

def test_clear():
    il = IgnoreList()
    il.Ignore('foo','bar')
    il.Clear()
    assert not il.AreIgnored('foo','bar')
    assert not il.AreIgnored('bar','foo')
    eq_(0,len(il))

def test_add_same_twice():
    il = IgnoreList()
    il.Ignore('foo','bar')
    il.Ignore('bar','foo')
    eq_(1,len(il))

def test_save_to_xml():
    il = IgnoreList()
    il.Ignore('foo','bar')
    il.Ignore('foo','bleh')
    il.Ignore('bleh','bar')
    f = io.BytesIO()
    il.save_to_xml(f)
    f.seek(0)
    doc = ET.parse(f)
    root = doc.getroot()
    eq_(root.tag, 'ignore_list')
    eq_(len(root), 2)
    eq_(len([c for c in root if c.tag == 'file']), 2)
    f1, f2 = root[:]
    subchildren = [c for c in f1 if c.tag == 'file'] + [c for c in f2 if c.tag == 'file']
    eq_(len(subchildren), 3)

def test_SaveThenLoad():
    il = IgnoreList()
    il.Ignore('foo', 'bar')
    il.Ignore('foo', 'bleh')
    il.Ignore('bleh', 'bar')
    il.Ignore('\u00e9', 'bar')
    f = io.BytesIO()
    il.save_to_xml(f)
    f.seek(0)
    il = IgnoreList()
    il.load_from_xml(f)
    eq_(4,len(il))
    assert il.AreIgnored('\u00e9','bar')
    
def test_LoadXML_with_empty_file_tags():
    f = io.BytesIO()
    f.write(b'<?xml version="1.0" encoding="utf-8"?><ignore_list><file><file/></file></ignore_list>')
    f.seek(0)
    il = IgnoreList()
    il.load_from_xml(f)
    eq_(0,len(il))
    
def test_AreIgnore_works_when_a_child_is_a_key_somewhere_else():
    il = IgnoreList()
    il.Ignore('foo','bar')
    il.Ignore('bar','baz')
    assert il.AreIgnored('bar','foo')


def test_no_dupes_when_a_child_is_a_key_somewhere_else():
    il = IgnoreList()
    il.Ignore('foo','bar')
    il.Ignore('bar','baz')
    il.Ignore('bar','foo')
    eq_(2,len(il))

def test_iterate():
    #It must be possible to iterate through ignore list
    il = IgnoreList()
    expected = [('foo','bar'),('bar','baz'),('foo','baz')]
    for i in expected:
        il.Ignore(i[0],i[1])
    for i in il:
        expected.remove(i) #No exception should be raised
    assert not expected #expected should be empty

def test_filter():
    il = IgnoreList()
    il.Ignore('foo','bar')
    il.Ignore('bar','baz')
    il.Ignore('foo','baz')
    il.Filter(lambda f,s: f == 'bar')
    eq_(1,len(il))
    assert not il.AreIgnored('foo','bar')
    assert il.AreIgnored('bar','baz')

def test_save_with_non_ascii_items():
    il = IgnoreList()
    il.Ignore('\xac', '\xbf')
    f = io.BytesIO()
    try:
        il.save_to_xml(f)
    except Exception as e:
        raise AssertionError(str(e))

def test_len():
    il = IgnoreList()
    eq_(0,len(il))
    il.Ignore('foo','bar')
    eq_(1,len(il))

def test_nonzero():
    il = IgnoreList()
    assert not il
    il.Ignore('foo','bar')
    assert il

def test_remove():
    il = IgnoreList()
    il.Ignore('foo', 'bar')
    il.Ignore('foo', 'baz')
    il.remove('bar', 'foo')
    eq_(len(il), 1)
    assert not il.AreIgnored('foo', 'bar')

def test_remove_non_existant():
    il = IgnoreList()
    il.Ignore('foo', 'bar')
    il.Ignore('foo', 'baz')
    with raises(ValueError):
        il.remove('foo', 'bleh')