def test_removeStatusesSelectType(self): ds = DataStore() ds.addStatus(Type.normal, u'status1') ds.addStatus(Type.normal, u'status2') ds.addStatus(Type.normal, u'status3') ds.addStatus(Type.priority, u'priority1') ds.addStatus(Type.priority, u'priority2') ds.addStatus(Type.priority, u'priority3') ds.addStatus(Type.omikuji, u'omikuji1') ds.addStatus(Type.omikuji, u'omikuji2') ds.addStatus(Type.omikuji, u'omikuji3') ds.removeStatuses(Type.normal) normallst = ds.getStatuses(Type.normal) assert_equal(0, len(normallst)) prioritylst = ds.getStatuses(Type.priority) assert_true((Type.priority==prioritylst[0][0] and prioritylst[0][1]==u'priority1')) assert_true((Type.priority==prioritylst[1][0] and prioritylst[1][1]==u'priority2')) assert_true((Type.priority==prioritylst[2][0] and prioritylst[2][1]==u'priority3')) omikujilst = ds.getStatuses(Type.omikuji) assert_true((Type.omikuji==omikujilst[0][0] and omikujilst[0][1]==u'omikuji1')) assert_true((Type.omikuji==omikujilst[1][0] and omikujilst[1][1]==u'omikuji2')) assert_true((Type.omikuji==omikujilst[2][0] and omikujilst[2][1]==u'omikuji3'))
def test_removeStatuses(self): ds = DataStore() ds.addStatus(Type.normal, u'status1') ds.addStatus(Type.normal, u'status2') ds.addStatus(Type.normal, u'status3') ds.addStatus(Type.priority, u'priority1') ds.addStatus(Type.priority, u'priority2') ds.addStatus(Type.priority, u'priority3') ds.addStatus(Type.omikuji, u'omikuji1') ds.addStatus(Type.omikuji, u'omikuji2') ds.addStatus(Type.omikuji, u'omikuji3') ds.removeStatuses() normallst = ds.getStatuses(Type.normal) assert_equal(0, len(normallst)) prioritylst = ds.getStatuses(Type.priority) assert_equal(0, len(prioritylst)) omikujilst = ds.getStatuses(Type.omikuji) assert_equal(0, len(omikujilst))
def test_DeleteAllStatus(self): ds = DataStore() ds.addStatus(Type.normal, u'abc') ds.addStatus(Type.normal, u'def') ds.addStatus(Type.normal, u'ghi') ds.addStatus(Type.priority, u'abc') ds.addStatus(Type.priority, u'def') ds.addStatus(Type.priority, u'ghi') ds.removeStatuses() # リストが0件であること lst = ds.getStatuses() assert_equal(0, len(lst))