mirror of
https://github.com/deadc0de6/dotdrop.git
synced 2026-02-04 22:04:44 +00:00
130 lines
3.6 KiB
Python
130 lines
3.6 KiB
Python
"""
|
|
author: deadc0de6 (https://github.com/deadc0de6)
|
|
Copyright (c) 2017, deadc0de6
|
|
basic unittest for the update function
|
|
"""
|
|
|
|
|
|
import unittest
|
|
import os
|
|
|
|
from dotdrop.dotdrop import cmd_update
|
|
from dotdrop.dotdrop import cmd_importer
|
|
|
|
from tests.helpers import create_dir, get_string, get_tempdir, clean, \
|
|
create_random_file, create_fake_config, load_options, edit_content
|
|
|
|
|
|
class TestUpdate(unittest.TestCase):
|
|
|
|
CONFIG_BACKUP = False
|
|
CONFIG_CREATE = True
|
|
CONFIG_DOTPATH = 'dotfiles'
|
|
CONFIG_NAME = 'config.yaml'
|
|
|
|
def test_update(self):
|
|
"""Test the update function"""
|
|
# setup some directories
|
|
fold_config = os.path.join(os.path.expanduser('~'), '.config')
|
|
create_dir(fold_config)
|
|
fold_subcfg = os.path.join(os.path.expanduser('~'), '.config',
|
|
get_string(5))
|
|
create_dir(fold_subcfg)
|
|
self.addCleanup(clean, fold_subcfg)
|
|
fold_tmp = get_tempdir()
|
|
create_dir(fold_tmp)
|
|
self.addCleanup(clean, fold_tmp)
|
|
|
|
# create the directories
|
|
tmp = get_tempdir()
|
|
self.assertTrue(os.path.exists(tmp))
|
|
self.addCleanup(clean, tmp)
|
|
|
|
dotfilespath = get_tempdir()
|
|
self.assertTrue(os.path.exists(dotfilespath))
|
|
self.addCleanup(clean, dotfilespath)
|
|
|
|
# create the dotfiles to test
|
|
d1, c1 = create_random_file(fold_config)
|
|
self.assertTrue(os.path.exists(d1))
|
|
self.addCleanup(clean, d1)
|
|
|
|
d2, c2 = create_random_file(fold_config)
|
|
self.assertTrue(os.path.exists(d2))
|
|
self.addCleanup(clean, d2)
|
|
|
|
# create the directory to test
|
|
dpath = os.path.join(fold_config, get_string(5))
|
|
dir1 = create_dir(dpath)
|
|
dirf1, _ = create_random_file(dpath)
|
|
self.addCleanup(clean, dir1)
|
|
|
|
# create the config file
|
|
profile = get_string(5)
|
|
confpath = create_fake_config(dotfilespath,
|
|
configname=self.CONFIG_NAME,
|
|
dotpath=self.CONFIG_DOTPATH,
|
|
backup=self.CONFIG_BACKUP,
|
|
create=self.CONFIG_CREATE)
|
|
self.assertTrue(os.path.exists(confpath))
|
|
o = load_options(confpath, profile)
|
|
dfiles = [d1, dir1, d2]
|
|
|
|
# import the files
|
|
o.import_path = dfiles
|
|
cmd_importer(o)
|
|
o = load_options(confpath, profile)
|
|
|
|
# edit the files
|
|
edit_content(d1, 'newcontent')
|
|
edit_content(dirf1, 'newcontent')
|
|
|
|
# add more file
|
|
dirf2, _ = create_random_file(dpath)
|
|
|
|
# add more dirs
|
|
dpath = os.path.join(dpath, get_string(5))
|
|
create_dir(dpath)
|
|
create_random_file(dpath)
|
|
|
|
# update it
|
|
o.safe = False
|
|
o.debug = True
|
|
o.update_path = [d1, dir1]
|
|
cmd_update(o)
|
|
|
|
# test content
|
|
newcontent = open(d1, 'r').read()
|
|
self.assertTrue(newcontent == 'newcontent')
|
|
newcontent = open(dirf1, 'r').read()
|
|
self.assertTrue(newcontent == 'newcontent')
|
|
|
|
edit_content(d2, 'newcontentbykey')
|
|
|
|
# update it by key
|
|
dfiles = o.dotfiles
|
|
d2key = ''
|
|
for ds in dfiles:
|
|
t = os.path.expanduser(ds.dst)
|
|
if t == d2:
|
|
d2key = ds.key
|
|
break
|
|
self.assertTrue(d2key != '')
|
|
o.safe = False
|
|
o.debug = True
|
|
o.update_path = [d2key]
|
|
o.iskey = True
|
|
cmd_update(o)
|
|
|
|
# test content
|
|
newcontent = open(d2, 'r').read()
|
|
self.assertTrue(newcontent == 'newcontentbykey')
|
|
|
|
|
|
def main():
|
|
unittest.main()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|