Merge branch 'master' of ch1p.io:homekit

1 files changed, 8 insertions, 15 deletions

diff --git a/test/test_record_upload.py b/test/test_record_upload.py
index cbd3ca2..f9c83d8 100755
--- a/test/test_record_upload.py
+++ b/test/test_record_upload.py
@@ -1,19 +1,12 @@
 #!/usr/bin/env python3
+import __py_include
 import logging
-import sys
-import os.path
-sys.path.extend([
-    os.path.realpath(
-        os.path.join(os.path.dirname(os.path.join(__file__)), '..')
-    )
-])
-
 import time
 
-from src.home.api import WebAPIClient, RequestParams
-from src.home.config import config
-from src.home.media import SoundRecordClient
-from src.home.util import parse_addr
+from homekit.api import WebApiClient, RequestParams
+from homekit.config import config
+from homekit.media import SoundRecordClient
+from homekit.util import Addr
 
 logger = logging.getLogger(__name__)
 
@@ -64,17 +57,17 @@ def api_success_handler(response, name, req: RequestParams):
 
 
 if __name__ == '__main__':
-    config.load('test_record_upload')
+    config.load_app('test_record_upload')
 
     nodes = {}
     for name, addr in config['nodes'].items():
-        nodes[name] = parse_addr(addr)
+        nodes[name] = Addr(addr)
     record = SoundRecordClient(nodes,
                                error_handler=record_error,
                                finished_handler=record_finished,
                                download_on_finish=True)
 
-    api = WebAPIClient()
+    api = WebApiClient()
     api.enable_async(error_handler=api_error_handler,
                      success_handler=api_success_handler)