Updating dependencies and using requirements.txt

benchtester/MarionetteTest.py

@@ -12,7 +12,7 @@
 import shutil
 
 import mozprofile
-from marionette.runtests import MarionetteTestRunner
+from marionette_harness.runtests import MarionetteTestRunner
 
 
 class MarionetteTest(BenchTester.BenchTest):

benchtester/test_memory_usage.py

@@ -6,7 +6,7 @@
 import os
 import sys
 
-from marionette import MarionetteTestCase
+from marionette_harness import MarionetteTestCase
 from marionette_driver import Actions
 from marionette_driver.errors import JavascriptException, ScriptTimeoutException
 import mozlog.structured

requirements.txt

@@ -1,6 +1,6 @@
 boto==2.39.0
-marionette-client==3.1.0
+marionette-harness==4.0
 mercurial==3.2.4
-mozdownload==1.20.2
+mozdownload==1.21
 MozillaPulse==1.0.1
-treeherder-client==3.0.0
+treeherder-client==3.1.0

setup.py

@@ -2,8 +2,16 @@
 # License, v. 2.0. If a copy of the MPL was not distributed with this
 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
+import os
 from setuptools import setup
 
+THIS_DIR = os.path.dirname(os.path.realpath(__name__))
+
+
+def read(*parts):
+    with open(os.path.join(THIS_DIR, *parts)) as f:
+        return f.read()
+
 setup(
     name="awsy",
     version="0.0.1",
@@ -17,12 +25,5 @@
       "License :: OSI Approved :: Mozilla Public License 2.0 (MPL 2.0)"
     ],
     packages=["benchtester"],
-    install_requires=[
-      "boto",
-      "marionette-client",
-      "mercurial",
-      "mozdownload",
-      "MozillaPulse",
-      "treeherder-client>=3.0.0"
-    ],
+    install_requires=read('requirements.txt').splitlines(),
 )