diff options
| -rw-r--r-- | functions.py | 10 | ||||
| -rwxr-xr-x[-rw-r--r--] | tests.py | 18 |
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') @@ -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() |