diff options
| -rw-r--r-- | machammer/functions.py | 16 | ||||
| -rwxr-xr-x | tests.py | 4 |
2 files changed, 17 insertions, 3 deletions
diff --git a/machammer/functions.py b/machammer/functions.py index a52dea0..900569d 100644 --- a/machammer/functions.py +++ b/machammer/functions.py @@ -117,7 +117,7 @@ def enable_ard(username): def sleep(): - """Put this Mac to sleep""" + """Put this Mac to sleep.""" tell_app('Finder', 'sleep') @@ -191,8 +191,8 @@ def install_su(restart=True): def disable_wifi(port='en1'): - subprocess.call(['/usr/sbin/networksetup', '-setairportpower', port, 'off']) - subprocess.call(['/usr/sbin/networksetup', '-setnetworkserviceenabled', 'Wi-Fi', 'off']) + call('/usr/sbin/networksetup', '-setairportpower', port, 'off') + call('/usr/sbin/networksetup', '-setnetworkserviceenabled', 'Wi-Fi', 'off') def log(msg): @@ -205,3 +205,13 @@ def install_service(src): os.mkdir(SERVICEDIR) ditto(src, SERVICEDIR) + + +def clear_xattr(path): + """Clear all extended attributes on path.""" + call('/usr/bin/xattr', '-c', path) + + +def create_os_media(src, dst): + fp = os.path.join(src, 'Contents/Resources/createinstallmedia') + call(fp, '--volume', dst, '--applicationpath', src, '--nointeraction') @@ -96,6 +96,10 @@ class FunctionsTestCase(TestCase): p = mh.mount_image('/Users/filipp/Downloads/AdobeFlashPlayer_22au_a_install.dmg') self.assertEquals(p, '/Volumes/Adobe Flash Player Installer') + def test_create_media(self): + mh.create_os_media('/Applications/Install macOS Sierra.app', + '/Volumes/Untitled') + @skip('This works, trust me.') def test_sleep(self): mh.sleep() |