Caroline Tice | f6ef439 | 2017-04-06 17:16:05 -0700 | [diff] [blame] | 1 | #!/usr/bin/env python2 |
asharif | bb91850 | 2013-02-15 05:15:01 +0000 | [diff] [blame] | 2 | # |
| 3 | # Copyright 2010 Google Inc. All Rights Reserved. |
cmtice | d96e457 | 2015-05-19 16:19:25 -0700 | [diff] [blame] | 4 | """Script to wrap test_that script. |
asharif | bb91850 | 2013-02-15 05:15:01 +0000 | [diff] [blame] | 5 | |
| 6 | This script can login to the chromeos machine using the test private key. |
| 7 | """ |
| 8 | |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 9 | from __future__ import print_function |
| 10 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 11 | __author__ = 'asharif@google.com (Ahmad Sharif)' |
asharif | bb91850 | 2013-02-15 05:15:01 +0000 | [diff] [blame] | 12 | |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 13 | import argparse |
asharif | bb91850 | 2013-02-15 05:15:01 +0000 | [diff] [blame] | 14 | import os |
asharif | bb91850 | 2013-02-15 05:15:01 +0000 | [diff] [blame] | 15 | import sys |
kbaclawski | 20082a0 | 2013-02-16 02:12:57 +0000 | [diff] [blame] | 16 | |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 17 | from cros_utils import command_executer |
| 18 | from cros_utils import misc |
asharif | bb91850 | 2013-02-15 05:15:01 +0000 | [diff] [blame] | 19 | |
| 20 | |
| 21 | def Usage(parser, message): |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 22 | print('ERROR: %s' % message) |
asharif | bb91850 | 2013-02-15 05:15:01 +0000 | [diff] [blame] | 23 | parser.print_help() |
| 24 | sys.exit(0) |
| 25 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 26 | |
asharif | bb91850 | 2013-02-15 05:15:01 +0000 | [diff] [blame] | 27 | def Main(argv): |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 28 | parser = argparse.ArgumentParser() |
Caroline Tice | f6ef439 | 2017-04-06 17:16:05 -0700 | [diff] [blame] | 29 | parser.add_argument( |
| 30 | '-c', |
| 31 | '--chromeos_root', |
| 32 | dest='chromeos_root', |
| 33 | help='ChromeOS root checkout directory') |
| 34 | parser.add_argument( |
| 35 | '-r', '--remote', dest='remote', help='Remote chromeos device.') |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 36 | options = parser.parse_args(argv) |
asharif | bb91850 | 2013-02-15 05:15:01 +0000 | [diff] [blame] | 37 | if options.chromeos_root is None: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 38 | Usage(parser, 'chromeos_root must be given') |
asharif | bb91850 | 2013-02-15 05:15:01 +0000 | [diff] [blame] | 39 | |
| 40 | if options.remote is None: |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 41 | Usage(parser, 'remote must be given') |
asharif | bb91850 | 2013-02-15 05:15:01 +0000 | [diff] [blame] | 42 | |
| 43 | options.chromeos_root = os.path.expanduser(options.chromeos_root) |
| 44 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 45 | command = 'ls -lt /' |
asharif | bb91850 | 2013-02-15 05:15:01 +0000 | [diff] [blame] | 46 | ce = command_executer.GetCommandExecuter() |
Caroline Tice | f6ef439 | 2017-04-06 17:16:05 -0700 | [diff] [blame] | 47 | ce.CrosRunCommand( |
| 48 | command, chromeos_root=options.chromeos_root, machine=options.remote) |
asharif | bb91850 | 2013-02-15 05:15:01 +0000 | [diff] [blame] | 49 | |
kbaclawski | 20082a0 | 2013-02-16 02:12:57 +0000 | [diff] [blame] | 50 | version_dir_path, script_name = misc.GetRoot(sys.argv[0]) |
| 51 | version_dir = misc.GetRoot(version_dir_path)[1] |
asharif | 29775b2 | 2013-02-15 05:15:38 +0000 | [diff] [blame] | 52 | |
| 53 | # Tests to copy directories and files to the chromeos box. |
Caroline Tice | f6ef439 | 2017-04-06 17:16:05 -0700 | [diff] [blame] | 54 | ce.CopyFiles( |
| 55 | version_dir_path, |
| 56 | '/tmp/' + version_dir, |
| 57 | dest_machine=options.remote, |
| 58 | dest_cros=True, |
| 59 | chromeos_root=options.chromeos_root) |
| 60 | ce.CopyFiles( |
| 61 | version_dir_path, |
| 62 | '/tmp/' + version_dir + '1', |
| 63 | dest_machine=options.remote, |
| 64 | dest_cros=True, |
| 65 | chromeos_root=options.chromeos_root) |
| 66 | ce.CopyFiles( |
| 67 | sys.argv[0], |
| 68 | '/tmp/' + script_name, |
| 69 | recursive=False, |
| 70 | dest_machine=options.remote, |
| 71 | dest_cros=True, |
| 72 | chromeos_root=options.chromeos_root) |
| 73 | ce.CopyFiles( |
| 74 | sys.argv[0], |
| 75 | '/tmp/' + script_name + '1', |
| 76 | recursive=False, |
| 77 | dest_machine=options.remote, |
| 78 | dest_cros=True, |
| 79 | chromeos_root=options.chromeos_root) |
asharif | 29775b2 | 2013-02-15 05:15:38 +0000 | [diff] [blame] | 80 | |
| 81 | # Test to copy directories and files from the chromeos box. |
Caroline Tice | f6ef439 | 2017-04-06 17:16:05 -0700 | [diff] [blame] | 82 | ce.CopyFiles( |
| 83 | '/tmp/' + script_name, |
| 84 | '/tmp/hello', |
| 85 | recursive=False, |
| 86 | src_machine=options.remote, |
| 87 | src_cros=True, |
| 88 | chromeos_root=options.chromeos_root) |
| 89 | ce.CopyFiles( |
| 90 | '/tmp/' + script_name, |
| 91 | '/tmp/' + script_name, |
| 92 | recursive=False, |
| 93 | src_machine=options.remote, |
| 94 | src_cros=True, |
| 95 | chromeos_root=options.chromeos_root) |
asharif | 6f4065c | 2013-02-15 09:04:02 +0000 | [diff] [blame] | 96 | board = ce.CrosLearnBoard(options.chromeos_root, options.remote) |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 97 | print(board) |
asharif | 29775b2 | 2013-02-15 05:15:38 +0000 | [diff] [blame] | 98 | return 0 |
asharif | bb91850 | 2013-02-15 05:15:01 +0000 | [diff] [blame] | 99 | |
| 100 | |
Luis Lozano | f2a3ef4 | 2015-12-15 13:49:30 -0800 | [diff] [blame] | 101 | if __name__ == '__main__': |
Caroline Tice | 88272d4 | 2016-01-13 09:48:29 -0800 | [diff] [blame] | 102 | Main(sys.argv[1:]) |