133 lines
3.8 KiB
Python
133 lines
3.8 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
import sys
|
||
|
|
sys.path.append(u'../ftplugin')
|
||
|
|
|
||
|
|
import vim
|
||
|
|
from orgmode.liborgmode.checkboxes import Checkbox
|
||
|
|
from orgmode._vim import ORGMODE
|
||
|
|
|
||
|
|
from orgmode.py3compat.encode_compatibility import *
|
||
|
|
|
||
|
|
def set_vim_buffer(buf=None, cursor=(2, 0), bufnr=0):
|
||
|
|
if buf is None:
|
||
|
|
buf = []
|
||
|
|
vim.current.buffer[:] = buf
|
||
|
|
vim.current.window.cursor = cursor
|
||
|
|
vim.current.buffer.number = bufnr
|
||
|
|
|
||
|
|
|
||
|
|
class CheckboxTestCase(unittest.TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
counter = 0
|
||
|
|
vim.CMDHISTORY = []
|
||
|
|
vim.CMDRESULTS = {}
|
||
|
|
vim.EVALHISTORY = []
|
||
|
|
vim.EVALRESULTS = {
|
||
|
|
# no org_todo_keywords for b
|
||
|
|
u_encode(u'exists("b:org_todo_keywords")'): u_encode('0'),
|
||
|
|
# global values for org_todo_keywords
|
||
|
|
u_encode(u'exists("g:org_todo_keywords")'): u_encode('1'),
|
||
|
|
u_encode(u'g:org_todo_keywords'): [u_encode(u'TODO'), u_encode(u'|'), u_encode(u'DONE')],
|
||
|
|
u_encode(u'exists("g:org_improve_split_heading")'): u_encode(u'0'),
|
||
|
|
u_encode(u'exists("b:org_improve_split_heading")'): u_encode(u'0'),
|
||
|
|
u_encode(u'exists("g:org_debug")'): u_encode(u'0'),
|
||
|
|
u_encode(u'exists("b:org_debug")'): u_encode(u'0'),
|
||
|
|
u_encode(u'exists("*repeat#set()")'): u_encode(u'0'),
|
||
|
|
u_encode(u'b:changedtick'): u_encode(u'%d' % counter),
|
||
|
|
u_encode(u'&ts'): u_encode(u'8'),
|
||
|
|
u_encode(u'exists("g:org_tag_column")'): u_encode(u'0'),
|
||
|
|
u_encode(u'exists("b:org_tag_column")'): u_encode(u'0'),
|
||
|
|
u_encode(u"v:count"): u_encode(u'0')}
|
||
|
|
|
||
|
|
self.c1 = """
|
||
|
|
* heading1 [/]
|
||
|
|
- [-] checkbox1 [%]
|
||
|
|
- [X] checkbox2
|
||
|
|
- [ ] checkbox3
|
||
|
|
- [X] checkbox4
|
||
|
|
""".split("\n")
|
||
|
|
|
||
|
|
self.c2 = """
|
||
|
|
* heading1
|
||
|
|
- [ ] checkbox1
|
||
|
|
- [ ] checkbox2
|
||
|
|
- [ ] checkbox3
|
||
|
|
- [ ] checkbox4
|
||
|
|
- [ ] checkbox5
|
||
|
|
- [ ] checkbox6
|
||
|
|
""".split("\n")
|
||
|
|
|
||
|
|
def test_init(self):
|
||
|
|
# test initialize Checkbox
|
||
|
|
c = Checkbox(level=1, title="checkbox1")
|
||
|
|
self.assertEqual(str(c), " - [ ] checkbox1")
|
||
|
|
c = Checkbox(level=3, title="checkbox2", status="[X]")
|
||
|
|
self.assertEqual(str(c), " - [X] checkbox2")
|
||
|
|
|
||
|
|
def test_basic(self):
|
||
|
|
bufnr = 1
|
||
|
|
set_vim_buffer(buf=self.c1, bufnr=bufnr)
|
||
|
|
h = ORGMODE.get_document(bufnr=bufnr).current_heading()
|
||
|
|
h.init_checkboxes()
|
||
|
|
|
||
|
|
c = h.current_checkbox(position=2)
|
||
|
|
self.assertEqual(str(c), self.c1[2])
|
||
|
|
self.assertFalse(c.are_children_all(Checkbox.STATUS_ON))
|
||
|
|
self.assertTrue(c.is_child_one(Checkbox.STATUS_OFF))
|
||
|
|
self.assertFalse(c.are_siblings_all(Checkbox.STATUS_ON))
|
||
|
|
|
||
|
|
for child in c.all_children():
|
||
|
|
pass
|
||
|
|
for sibling in c.all_siblings():
|
||
|
|
pass
|
||
|
|
c = h.current_checkbox(position=3)
|
||
|
|
new_checkbox = c.copy()
|
||
|
|
self.assertEqual(str(c), self.c1[3])
|
||
|
|
c.get_parent_list()
|
||
|
|
c.get_index_in_parent_list()
|
||
|
|
|
||
|
|
def test_identify(self):
|
||
|
|
# test identify_checkbox
|
||
|
|
self.assertEqual(Checkbox.identify_checkbox(self.c1[2]), 2)
|
||
|
|
self.assertEqual(Checkbox.identify_checkbox(self.c1[3]), 8)
|
||
|
|
# check for corner case
|
||
|
|
self.assertEqual(Checkbox.identify_checkbox(" - [ ]"), 1)
|
||
|
|
|
||
|
|
def test_toggle(self):
|
||
|
|
bufnr = 2
|
||
|
|
# test init_checkboxes
|
||
|
|
set_vim_buffer(buf=self.c1, bufnr=bufnr)
|
||
|
|
h = ORGMODE.get_document(bufnr=bufnr).current_heading()
|
||
|
|
h.init_checkboxes()
|
||
|
|
|
||
|
|
# toggle checkbox
|
||
|
|
c = h.current_checkbox(position=4)
|
||
|
|
c.toggle()
|
||
|
|
self.assertEqual(str(c), " - [X] checkbox3")
|
||
|
|
c.toggle()
|
||
|
|
self.assertEqual(str(c), " - [ ] checkbox3")
|
||
|
|
|
||
|
|
(total, on) = c.all_siblings_status()
|
||
|
|
self.assertEqual((total, on), (2, 1))
|
||
|
|
|
||
|
|
def test_subtasks(self):
|
||
|
|
bufnr = 3
|
||
|
|
set_vim_buffer(buf=self.c1, bufnr=bufnr)
|
||
|
|
h = ORGMODE.get_document(bufnr=bufnr).current_heading()
|
||
|
|
h.init_checkboxes()
|
||
|
|
c = h.current_checkbox(position=3)
|
||
|
|
c.toggle()
|
||
|
|
c = h.current_checkbox(position=2)
|
||
|
|
(total, on) = c.all_siblings_status()
|
||
|
|
c.update_subtasks(total=total, on=on)
|
||
|
|
self.assertEqual(str(c), " - [-] checkbox1 [50%]")
|
||
|
|
|
||
|
|
|
||
|
|
def suite():
|
||
|
|
return unittest.TestLoader().loadTestsFromTestCase(CheckboxTestCase)
|
||
|
|
|
||
|
|
# vim: set noexpandtab:
|