Added notifications, tests

2 files changed, 22 insertions, 6 deletions

diff --git a/functions.py b/functions.py
index 9fa4b4e..6b76dc4 100644
--- a/functions.py
+++ b/functions.py
@@ -12,12 +12,16 @@ from system_profiler import SystemProfile
 SERVICEDIR = '/Library/Services'
 
 
+def display_notification(msg, title='', subtitle=''):
+    osascript('display notification "{0}" with title "{1}" subtitle "{2}"'.format(msg, title, subtitle))
+
+
 def ditto(src, dst):
     subprocess.call(['/usr/bin/ditto', src, dst])
 
 
 def rsync(src, dst, flags='auE'):
-    subprocess.call(['rsync', '-' + flags, src, dst])
+    subprocess.call(['/usr/bin/rsync', '-' + flags, src, dst])
 
 
 def dscl(domain='.', *args):
@@ -216,7 +220,3 @@ def install_service(src):
         os.mkdir(SERVICEDIR)
 
     ditto(src, SERVICEDIR)
-
-
-if __name__ == '__main__':
-    delete_user('demouser')
diff --git a/tests.py b/tests.py
index 553b5ed..493fc0f 100644..100755
--- a/tests.py
+++ b/tests.py
@@ -1,9 +1,11 @@
+#!/usr/bin/env python
 # -*- coding: utf-8 -*-
 
 import logging
 import subprocess
 from unittest import main, skip, TestCase
 
+import functions as mh
 import system_profiler
 
 
@@ -24,7 +26,7 @@ class SystemProfilerTestCase(TestCase):
 
     def testOsVersion(self):
         """
-        Check that the OS version we get from SP is contained 
+        Check that the OS version we get from SP is contained
         in the output of sw_vers
         """
         build = subprocess.check_output(['sw_vers', '-buildVersion']).strip()
@@ -53,6 +55,20 @@ class AppsTestCase(TestCase):
         self.assertTrue(len(results) > 10)
 
 
+class FunctionsTestCase(TestCase):
+    def setUp(self):
+        self.stickes = '/Applications/Stickies.app'
+
+    def test_notification(self):
+        mh.display_notification('blaaa')
+
+    def test_add_login_item(self):
+        mh.add_login_item(self.stickes)
+
+    def test_remove_login_item(self):
+        mh.remove_login_item(path=self.stickes)
+
+
 if __name__ == '__main__':
     logging.basicConfig(level=logging.DEBUG)
     main()