def setUp(self):
localmode_test_config = """
[pda]
database-path = /tmp/.pdateststore
"""
cfg = PdaConfig(io.StringIO(u(localmode_test_config)))
self.db = GithubIssues(cfg)
# populate local data store with some data
self.tl = [str(self.db.add_task('test summary 1', 'todo', 'week', 'must')),
str(self.db.add_task('test summary 2', 'tolearn', 'month', 'high')),
str(self.db.add_task('test summary 3', 'toread', 'day', 'urgmust'))]
def setUp(self):
with open('.pdaconfig', 'r') as f:
cfg = PdaConfig(f)
self.db = GithubIssues(cfg)
self.db.sync_local_dbstore()
class ListDBTests(unittest.TestCase):
def setUp(self):
localmode_test_config = """
[pda]
database-path = /tmp/.pdateststore
"""
cfg = PdaConfig(io.StringIO(u(localmode_test_config)))
self.db = GithubIssues(cfg)
# populate local data store with some data
self.tl = [str(self.db.add_task('test summary 1', 'todo', 'week', 'must')),
str(self.db.add_task('test summary 2', 'tolearn', 'month', 'high')),
str(self.db.add_task('test summary 3', 'toread', 'day', 'urgmust'))]
def tearDown(self):
if os.path.exists(self.db.local_dbpath):
os.remove(self.db.local_dbpath)
del self.db
def testAddTask(self):
# initially the local data store should have 3 tasks
self.assertTrue(len(self.db.shelf) == 3)
# check tasks existence
self.assertTrue(self.tl[0] in self.db.shelf)
self.assertTrue(self.tl[1] in self.db.shelf)
self.assertTrue(self.tl[2] in self.db.shelf)
# check data integrity
self.assertTrue(self.db.shelf[self.tl[0]]['summary'] == 'test summary 1')
self.assertTrue(self.db.shelf[self.tl[1]]['type'] == 'tolearn')
self.assertTrue(self.db.shelf[self.tl[1]]['milestone'] == 'month')
self.assertTrue(self.db.shelf[self.tl[2]]['milestone'] == 'day')
self.assertTrue(self.db.shelf[self.tl[2]]['priority'] == 'urgmust')
self.assertFalse('CMDS_HISTORY' in self.db.shelf)
def testRemoveTask(self):
# initially the local data store should have 3 tasks
self.assertTrue(len(self.db.shelf) == 3)
# remove task 1
self.db.remove_task(int(self.tl[0]))
self.assertFalse(self.tl[0] in self.db.shelf)
self.assertTrue(len(self.db.shelf) == 2)
# remove non-existing task (task id = -1) and suppress output of remove_task
with patch('sys.stdout', new=MockDevice()) as fake_out:
self.db.remove_task(-1)
self.assertTrue(len(self.db.shelf) == 2)
self.assertFalse('CMDS_HISTORY' in self.db.shelf)
def testFinishTasks(self):
# initially the local data store should have 3 tasks
self.assertTrue(len(self.db.shelf) == 3)
# remove first two tasks
self.db.finish_tasks([int(self.tl[0]), int(self.tl[1])])
self.assertFalse(self.tl[0] in self.db.shelf)
self.assertFalse(self.tl[1] in self.db.shelf)
self.assertTrue(self.tl[2] in self.db.shelf)
self.assertTrue(len(self.db.shelf) == 1)
self.assertFalse('CMDS_HISTORY' in self.db.shelf)
def testRemoveAllTasks(self):
# initially the local data store should have 3 tasks
self.assertTrue(len(self.db.shelf) == 3)
# remove all tasks
self.db.remove_all_tasks()
# verified they are indeed removed
self.assertTrue(len(self.db.shelf) == 0)
self.assertFalse('CMDS_HISTORY' in self.db.shelf)
self.assertFalse(self.tl[0] in self.db.shelf)
def testEditTask(self):
# initially the local data store should have 3 tasks
self.assertTrue(len(self.db.shelf) == 3)
# edit the first and the third task
self.db.edit_task(self.tl[0],
new_summary='test summary 4',
new_tasktype='totest',
new_milestone='season')
self.db.edit_task(self.tl[2],
new_priority='low')
# check data integrity
self.assertTrue(len(self.db.shelf) == 3)
self.assertTrue(self.db.shelf[self.tl[0]]['summary'] == 'test summary 4')
self.assertTrue(self.db.shelf[self.tl[0]]['type'] == 'totest')
self.assertTrue(self.db.shelf[self.tl[0]]['milestone'] == 'season')
self.assertTrue(self.db.shelf[self.tl[0]]['priority'] == 'must')
self.assertTrue(self.db.shelf[self.tl[2]]['type'] == 'toread')
self.assertTrue(self.db.shelf[self.tl[2]]['priority'] == 'low')
self.assertTrue(self.db.shelf[self.tl[1]]['summary'] == 'test summary 2')
self.assertFalse('CMDS_HISTORY' in self.db.shelf)
class ListDBSyncingTests(unittest.TestCase):
def setUp(self):
with open('.pdaconfig', 'r') as f:
cfg = PdaConfig(f)
self.db = GithubIssues(cfg)
self.db.sync_local_dbstore()
def tearDown(self):
if os.path.exists(self.db.local_dbpath):
os.remove(self.db.local_dbpath)
del self.db
def testSyncLocalDBStore(self):
# retrieve remote data for testing
r = requests.get(self.db.url_issues, params={'state': 'open'}, auth=self.db.auth)
for issue in r.json():
task_no = str(issue["number"])
task_summary = issue["title"]
task_milestone = issue["milestone"]["title"]
task_prio = ""
task_type = ""
for label in issue["labels"]:
if label["color"] == GithubIssues.YELLOW:
task_prio = label["name"]
if label["color"] == GithubIssues.GREEN:
task_type = label["name"]
# after data syncing, the issue should be present in local store
self.assertTrue(task_no in self.db.shelf)
# the data stored at local should be the same as remote
self.assertTrue(self.db.shelf[task_no]["summary"] == task_summary)
self.assertTrue(self.db.shelf[task_no]["type"] == task_type)
self.assertTrue(self.db.shelf[task_no]["priority"] == task_prio)
self.assertTrue(self.db.shelf[task_no]["milestone"] == task_milestone)
self.assertTrue('CMDS_HISTORY' in self.db.shelf)
def testSyncRemoteDBStore(self):
if self.db.shelf:
task_numbers = [number for number in list(self.db.shelf.keys()) \
if number != 'CMDS_HISTORY']
# (1) remove tasks
if task_numbers:
number = task_numbers.pop()
self.db.remove_task(int(number))
if task_numbers:
number = task_numbers.pop()
self.db.remove_task(int(number))
# (2) add a task locally
self.db.add_task('first added', 'tolearn', 'month', 'low')
# (3) add another task
added_2nd = self.db.add_task('secondly added', 'todo', 'year', 'high')
# (4) edit a locally added task
self.db.edit_task(int(added_2nd), new_milestone='season', new_tasktype='toread')
# (5) edit a remotely imported task
if task_numbers:
number = task_numbers.pop()
self.db.edit_task(int(number), new_priority='urgmust')
# a data structure in memory to hold all the records before data is synced to remote
records = [self.db.shelf[k] for k in self.db.shelf if k != 'CMDS_HISTORY']
# syncing data to remote
self.db.sync_remote_dbstore()
# retrieving remote data to local memory
r = requests.get(self.db.url_issues, params={'state': 'open'}, auth=self.db.auth)
if r.status_code == requests.codes.ok:
remote_records = r.json()
# check if data is equivalent by making use of sets!
local, remote = set([]), set([])
for rec in records:
local.add((rec['summary'], rec['type'], rec['milestone'], rec['priority']))
for rec in remote_records:
prio, ltype = self.db.get_task_prio_and_type(rec)
remote.add((rec['title'], ltype, rec['milestone']['title'], prio))
# finally test if local and remote contents are synced!
self.assertTrue(local==remote)
