diff options
Diffstat (limited to 'scripts/update_payload/test_utils.py')
| -rw-r--r-- | scripts/update_payload/test_utils.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/scripts/update_payload/test_utils.py b/scripts/update_payload/test_utils.py index 4e7881d3..8746fcef 100644 --- a/scripts/update_payload/test_utils.py +++ b/scripts/update_payload/test_utils.py @@ -8,6 +8,7 @@ from __future__ import print_function import cStringIO import hashlib +import os import struct import subprocess @@ -21,8 +22,10 @@ class TestError(Exception): # Private/public RSA keys used for testing. -_PRIVKEY_FILE_NAME = 'payload-test-key.pem' -_PUBKEY_FILE_NAME = 'payload-test-key.pub' +_PRIVKEY_FILE_NAME = os.path.join(os.path.dirname(__file__), + 'payload-test-key.pem') +_PUBKEY_FILE_NAME = os.path.join(os.path.dirname(__file__), + 'payload-test-key.pub') def KiB(count): |