def testDelAttrs(self): d1 = Driver('d1') d1.add_attr(key='foo', value='bar1', number=True, subkey='one') d1.add_attr(key='foo', value='bar2', number=True, subkey='two') d1.add_attr(key='foo', value='bar3', number=True, subkey='three') d1.add_attr(key='foo', value='bar4', number=True, subkey='four') d1.del_attrs(key='foo', value='bar4') self.assertEqual(list(d1.attr_items(value='bar4')), []) self.assertEqual(list(d1.attr_items(value='bar3')), [(('foo', 2, 'three'), 'bar3')]) d1.del_attrs(key='foo', subkey='three', number=2) self.assertEqual(list(d1.attr_items(value='bar3')), [])
def testAttrCount(self): d1 = Driver('d1') d1.add_attr(key='foo', value='bar1', number=True, subkey='one') d1.add_attr(key='foo', value='bar2', number=True, subkey='two') d1.add_attr(key='foo', value='bar3', number=True, subkey='three') d1.add_attr(key='foo', value='bar4', number=True, subkey='four') self.assertEqual(d1.attr_query(key='foo', number=2, count=True), 1) self.assertEqual(d1.attr_query(key='foo', number=0, count=True), 1) self.assertEqual(d1.attr_query(key='foo', number=False, count=True), 0) self.assertEqual(d1.attr_query(key='foo', count=True), 4) self.assertEqual(d1.attr_query(subkey='four', count=True), 1) d1.del_attrs(key='foo', value='bar1', number=True, subkey='one') d1.add_attr(key='foo', value='bar5', number=True, subkey='five') self.assertEqual(d1.attr_query(key='foo', number=0, count=True), 0) self.assertEqual(d1.attr_query(key='foo', number=4, count=True), 1)