From 93aa48ea5549da8b5e59d554e3b2b80521faffa4 Mon Sep 17 00:00:00 2001
From: deadc0de6 <info@deadc0de.ch>
Date: Wed, 15 Mar 2017 22:44:07 +0100
Subject: [PATCH] adding tests for install

---
 tests/test_install.py | 81 +++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 79 insertions(+), 2 deletions(-)

diff --git a/tests/test_install.py b/tests/test_install.py
index 4ad8818..bd77a0b 100644
--- a/tests/test_install.py
+++ b/tests/test_install.py
@@ -5,14 +5,91 @@ basic unittest for the install function
 """
 
 import unittest
+import filecmp
+
+from tests.helpers import *
+from dotdrop.dotfile import Dotfile
+from dotdrop.dotdrop import install
 
 
 class TestInstall(unittest.TestCase):
 
+    CONFIG_NAME = 'config.yaml'
+
+    TEMPLATE = '''
+# launch the wm
+{%@@ if profile == "home" @@%}
+exec awesome
+{%@@ else @@%}
+exec bspwm
+{%@@ endif @@%}
+'''
+    RESULT = '''
+# launch the wm
+exec bspwm
+'''
+
+    def fake_config(self, path, dotfiles, profile, dotpath):
+        '''Create a fake config file'''
+        with open(path, 'w') as f:
+            f.write('config:\n')
+            f.write('  backup: false\n')
+            f.write('  create: true\n')
+            f.write('  dotpath: %s\n' % (dotpath))
+            f.write('dotfiles:\n')
+            for d in dotfiles:
+                f.write('  %s:\n' % (d.key))
+                f.write('    dst: %s\n' % (d.dst))
+                f.write('    src: %s\n' % (d.src))
+            f.write('profiles:\n')
+            f.write('  %s:\n' % (profile))
+            for d in dotfiles:
+                f.write('  - %s\n' % (d.key))
+        return path
+
     def test_install(self):
         '''Test the install function'''
-        # TODO
-        self.assertTrue(True)
+        tmp = get_tempfolder()
+        self.assertTrue(os.path.exists(tmp))
+        self.addCleanup(clean, tmp)
+
+        dst = get_tempfolder()
+        self.assertTrue(os.path.exists(dst))
+        self.addCleanup(clean, dst)
+
+        # create the dotfile in dotdrop
+        f1, c1 = create_random_file(tmp)
+        dst1 = os.path.join(dst, get_string(6))
+        d1 = Dotfile(get_string(5), dst1, os.path.basename(f1))
+        f2, c2 = create_random_file(tmp)
+        dst2 = os.path.join(dst, get_string(6))
+        d2 = Dotfile(get_string(5), dst2, os.path.basename(f2))
+        with open(f2, 'w') as f:
+            f.write(self.TEMPLATE)
+        f3, _ = create_random_file(tmp, binary=True)
+        dst3 = os.path.join(dst, get_string(6))
+        d3 = Dotfile(get_string(5), dst3, os.path.basename(f3))
+
+        # generate the config and stuff
+        profile = get_string(5)
+        confpath = os.path.join(tmp, self.CONFIG_NAME)
+        self.fake_config(confpath, [d1, d2, d3], profile, tmp)
+        conf = Cfg(confpath, tmp)
+        self.assertTrue(conf is not None)
+
+        # install them
+        conf, opts = load_config(confpath, tmp, profile)
+        install(opts, conf)
+
+        # now compare the generated files
+        self.assertTrue(os.path.exists(dst1))
+        self.assertTrue(os.path.exists(dst2))
+        self.assertTrue(os.path.exists(dst3))
+
+        self.assertTrue(filecmp.cmp(f1, dst1, shallow=True))
+        f2content = open(dst2, 'r').read()
+        self.assertTrue(f2content == self.RESULT)
+        self.assertTrue(filecmp.cmp(f3, dst3, shallow=True))
 
 
 def main():