def handleNewObject(self, args): if self.version < 3: name = args parentId = None else: name, parentId = args self.disp().log(_('New category (parent: %s)'), parentId) if parentId is None or not self.categoryMap.has_key(parentId): category = Category(name) else: category = self.categoryMap[parentId].newChild(name) self.disp().window.addIPhoneCategory(category) self.categoryMap[category.id()] = category self.pack('s', category.id())
def handleNewObject(self, args): if self.version < 3: name = args parentId = None else: name, parentId = args if parentId is None: category = Category(name) else: category = self.categoryMap[parentId].newChild(name) self.disp().window.addIPhoneCategory(category) self.pack('s', category.id()) self.categoryMap[category.id()] = category self.count += 1 self.ui.SetProgress(self.count, self.total)
class MonitorCategorizableTest(test.TestCase): def setUp(self): self.monitor = ChangeMonitor() self.monitor.monitorClass(CategorizableCompositeObject) self.obj = CategorizableCompositeObject(subject=u'Object') self.list = ObservableList() self.monitor.monitorCollection(self.list) self.list.append(self.obj) self.cat1 = Category(subject=u'Cat #1') self.cat2 = Category(subject=u'Cat #2') self.catList = ObservableList() self.catList.append(self.cat1) self.catList.append(self.cat2) def tearDown(self): self.monitor.unmonitorClass(CategorizableCompositeObject) self.monitor.unmonitorCollection(self.list) def testAddCategory(self): self.monitor.resetAllChanges() self.obj.addCategory(self.cat1) self.assertEqual(self.monitor.getChanges(self.obj), set(['__add_category:%s' % self.cat1.id()])) def testRemoveCategory(self): self.obj.addCategory(self.cat1) self.monitor.resetAllChanges() self.obj.removeCategory(self.cat1) self.assertEqual(self.monitor.getChanges(self.obj), set(['__del_category:%s' % self.cat1.id()])) def testRemoveBadCategory(self): self.obj.addCategory(self.cat1) self.monitor.resetAllChanges() self.obj.removeCategory(self.cat2) self.assertEqual(self.monitor.getChanges(self.obj), set())