// Wait for a machine to show up in the data store static void FindMachine(IResourcePool rootPool, string machineName) { Console.WriteLine("Looking for machine '{0}'", machineName); IResource machine = null; while (true) { try { machine = rootPool.GetResourceByName(machineName); } catch (Exception e) { Console.WriteLine("Warning: " + e.Message); } // Make sure the machine is valid if (machine != null && machine.OperatingSystem != null && machine.OperatingSystem.Length > 0 && machine.ProcessorArchitecture != null && machine.ProcessorArchitecture.Length > 0 && machine.GetDevices().Length > 0) { break; } System.Threading.Thread.Sleep(1000); } Console.WriteLine("Client machine '{0}' found ({1}, {2})", machineName, machine.OperatingSystem, machine.ProcessorArchitecture); }
// Look for a device in a machine, and if not found, keep trying for 3 minutes static IDevice GetDevice(IResourcePool rootPool, string machineName, string regexStr) { Regex deviceRegex = new Regex(regexStr, RegexOptions.IgnoreCase); int numAttempts = 1; DateTime endTime = DateTime.Now.AddSeconds(180); while (DateTime.Now < endTime) { IResource machine = rootPool.GetResourceByName(machineName); Console.WriteLine("Looking for device '{0}' in machine '{1}' (machine has {2} devices)", regexStr, machineName, machine.GetDevices().Length); foreach (IDevice d in machine.GetDevices()) { if (deviceRegex.IsMatch(d.FriendlyName)) { Console.WriteLine("Found device '{0}'", d.FriendlyName); return(d); } } Console.WriteLine("Device not found"); if (numAttempts % 5 == 0) { ResetMachine(rootPool, machineName, true); } else { System.Threading.Thread.Sleep(5000); } numAttempts++; } Console.WriteLine("Error: device '{0}' not found", deviceRegex); return(null); }
// Wait for a machine to show up in the data store static void FindMachine(IResourcePool rootPool, string machineName) { Console.WriteLine("Looking for machine '{0}'", machineName); IResource machine = null; while (true) { try { machine = rootPool.GetResourceByName(machineName); } catch (Exception e) { Console.WriteLine("Warning: " + e.Message); } // Make sure the machine is valid if (machine != null && machine.OperatingSystem != null && machine.OperatingSystem.Length > 0 && machine.ProcessorArchitecture != null && machine.ProcessorArchitecture.Length > 0 && machine.GetDevices().Length > 0) break; System.Threading.Thread.Sleep(1000); } Console.WriteLine("Client machine '{0}' found ({1}, {2})", machineName, machine.OperatingSystem, machine.ProcessorArchitecture); }
// Set the machine's status to 'Reset' and optionally wait for it to become ready static void ResetMachine(IResourcePool rootPool, string machineName, bool wait) { Console.WriteLine("Resetting machine '{0}'", machineName); IResource machine; while (true) { try { machine = rootPool.GetResourceByName(machineName); machine.ChangeResourceStatus("Reset"); break; } catch (Exception e) { Console.WriteLine("Warning: " + e.Message); System.Threading.Thread.Sleep(5000); } } if (wait) { Console.WriteLine("Waiting for machine '{0}' to be ready", machineName); while (machine.Status != "Ready") { try { machine = rootPool.GetResourceByName(machineName); } catch (Exception e) { Console.WriteLine("Warning: " + e.Message); } System.Threading.Thread.Sleep(1000); } Console.WriteLine("Machine '{0}' is ready", machineName); } }
// Set the machine's status to 'Reset' and optionally wait for it to become ready static void ResetMachine(IResourcePool rootPool, string machineName, bool wait) { Console.WriteLine("Resetting machine '{0}'", machineName); IResource machine; while (true) { try { machine = rootPool.GetResourceByName(machineName); machine.ChangeResourceStatus("Reset"); break; } catch (Exception e) { Console.WriteLine("Warning: " + e.Message); System.Threading.Thread.Sleep(5000); } } if (wait) { Console.WriteLine("Waiting for machine '{0}' to be ready", machineName); while (machine.Status != "Ready") { try { machine = rootPool.GetResourceByName(machineName); } catch (Exception e) { Console.WriteLine("Warning: " + e.Message); } System.Threading.Thread.Sleep(1000); } Console.WriteLine("Machine '{0}' is ready", machineName); } }
static int Main(string[] args) { if (args.Length != 2) { Console.WriteLine("Error: incorrect number of command line arguments"); Console.WriteLine("Usage: {0} serverName clientName", System.Environment.GetCommandLineArgs()[0]); return(1); } string serverName = args[0]; string clientName = args[1]; try { // Initialize DeviceScript and connect to data store Console.WriteLine("Initializing DeviceScript object"); DeviceScript script = new DeviceScript(); Console.WriteLine("Connecting to data store"); script.ConnectToNamedDataStore(serverName); // Find the client machine IResourcePool rootPool = script.GetResourcePoolByName("$"); Console.WriteLine("Looking for client machine '{0}'", clientName); IResource machine = rootPool.GetResourceByName(clientName); if (machine == null) { Console.WriteLine("Client machine not found"); return(0); } Console.WriteLine("Client machine '{0}' found ({1}, {2})", clientName, machine.OperatingSystem, machine.ProcessorArchitecture); // Change the client machine's status to 'unsafe' Console.WriteLine("Changing the client machine's status to 'Unsafe'"); try { machine.ChangeResourceStatus("Unsafe"); } catch (Exception e) { Console.WriteLine("Warning: " + e.Message); } while (machine.Status != "Unsafe") { try { machine = rootPool.GetResourceByName(clientName); } catch (Exception e) { Console.WriteLine("Warning: " + e.Message); } System.Threading.Thread.Sleep(1000); } // Delete the client machine from datastore Console.WriteLine("Deleting client machine from data store"); script.DeleteResource(machine.Id); return(0); } catch (Exception e) { Console.WriteLine("Error: " + e.Message); return(1); } }
static int Main(string[] args) { if (args.Length < 5) { Console.WriteLine("Error: incorrect number of command line arguments"); Console.WriteLine("Usage: {0} serverName machinePoolName submissionName timeout machineName0 machineName1 ...", System.Environment.GetCommandLineArgs()[0]); return(1); } string serverName = args[0]; string machinePoolName = args[1]; string submissionName = args[2]; double timeout = Convert.ToDouble(args[3]); List <string> machines = new List <string>(); for (int i = 4; i < args.Length; i++) { machines.Add(args[i]); } try { // Initialize DeviceScript and connect to data store Console.WriteLine("Initializing DeviceScript object"); DeviceScript script = new DeviceScript(); Console.WriteLine("Connecting to data store"); script.ConnectToNamedDataStore(serverName); // Wait for client machines to become available IResourcePool rootPool = script.GetResourcePoolByName("$"); foreach (string machineName in machines) { FindMachine(rootPool, machineName); } // Delete the machine pool if it already exists DeleteResourcePool(script, machinePoolName); // Create the machine pool and add the client machines to it // (this must be done because jobs cannot be scheduled for machines in the // default pool) try { script.CreateResourcePool(machinePoolName, rootPool.ResourcePoolId); } catch (Exception e) { Console.WriteLine("Warning: " + e.Message); } IResourcePool newPool = script.GetResourcePoolByName(machinePoolName); foreach (string machineName in machines) { Console.WriteLine("Moving machine '{0}' to pool '{1}'", machineName, machinePoolName); rootPool.GetResourceByName(machineName).ChangeResourcePool(newPool); } // Reset client machine foreach (string machineName in machines) { ResetMachine(rootPool, machineName, true); } // Get requested device regex and look for a matching device in the first machine Console.WriteLine("Device to test:"); IDevice device = GetDevice(rootPool, machines[0], Console.ReadLine()); if (device == null) { return(1); } // Get requested jobs regex Console.WriteLine("Jobs to run:"); Regex jobRegex = new Regex(Console.ReadLine(), RegexOptions.IgnoreCase); // Create a submission Object[] existingSubmissions = script.GetSubmissionByName(submissionName); if (existingSubmissions.Length > 0) { Console.WriteLine("Submission '{0}' already exists -- removing it", submissionName); script.DeleteSubmission(((ISubmission)existingSubmissions[0]).Id); } string hardwareId = device.InstanceId.Remove(device.InstanceId.LastIndexOf("\\")); Console.WriteLine("Creating submission '{0}' (hardware ID: {1})", submissionName, hardwareId); ISubmission submission = script.CreateHardwareSubmission(submissionName, newPool.ResourcePoolId, hardwareId); // Set submission DeviceData List <Object> deviceDataList = new List <Object>(); while (true) { ISubmissionDeviceData dd = script.CreateNewSubmissionDeviceData(); Console.WriteLine("DeviceData name:"); dd.Name = Console.ReadLine(); if (dd.Name.Length == 0) { break; } Console.WriteLine("DeviceData data:"); dd.Data = Console.ReadLine(); deviceDataList.Add(dd); } submission.SetDeviceData(deviceDataList.ToArray()); // Set submission descriptors List <Object> descriptorList = new List <Object>(); while (true) { Console.WriteLine("Descriptor path:"); string descriptorPath = Console.ReadLine(); if (descriptorPath.Length == 0) { break; } descriptorList.Add(script.GetDescriptorByPath(descriptorPath)); } submission.SetLogoDescriptors(descriptorList.ToArray()); // Set machine dimensions foreach (string machineName in machines) { IResource machine = rootPool.GetResourceByName(machineName); while (true) { Console.WriteLine("Dimension name ({0}):", machineName); string dimName = Console.ReadLine(); if (dimName.Length == 0) { break; } Console.WriteLine("Dimension value ({0}):", machineName); machine.SetDimension(dimName, Console.ReadLine()); } // Set the WDKSubmissionId dimension for all machines machine.SetDimension("WDKSubmissionId", submission.Id.ToString() + "_" + submission.Name); } // Get job parameters List <string> paramNames = new List <string>(); List <string> paramValues = new List <string>(); foreach (string machineName in machines) { while (true) { Console.WriteLine("Parameter name ({0}):", machineName); string paramName = Console.ReadLine(); if (paramName.Length == 0) { break; } Console.WriteLine("Device regex ({0}):", machineName); IDevice d = GetDevice(rootPool, machineName, Console.ReadLine()); if (d == null) { return(1); } string deviceName = d.GetAttribute("name")[0].ToString(); Console.WriteLine("Setting parameter value to '{0}'", deviceName); paramNames.Add(paramName); paramValues.Add(deviceName); } } // Find jobs that match the requested pattern Console.WriteLine("Scheduling jobs:"); List <IJob> jobs = new List <IJob>(); foreach (IJob j in submission.GetJobs()) { if (jobRegex.IsMatch(j.Name)) { Console.WriteLine(" " + j.Name); // Set job parameters for (int i = 0; i < paramNames.Count; i++) { IParameter p = j.GetParameterByName(paramNames[i]); if (p != null) { p.ScheduleValue = paramValues[i]; } } jobs.Add(j); } } if (jobs.Count == 0) { Console.WriteLine("Error: no submission jobs match pattern '{0}'", jobRegex); return(1); } // Create a schedule, add jobs to it and run it ISchedule schedule = script.CreateNewSchedule(); foreach (IScheduleItem item in submission.ProcessJobs(jobs.ToArray())) { item.Device = device; schedule.AddScheduleItem(item); } schedule.AddSubmission(submission); schedule.SetResourcePool(newPool); script.RunSchedule(schedule); // Wait for jobs to complete Console.WriteLine("Waiting for all jobs to complete (timeout={0}s)", timeout); DateTime endTime = DateTime.Now.AddSeconds(timeout); int numCompleted, numFailed; do { System.Threading.Thread.Sleep(30000); // Report results in a Python readable format and count completed and failed schedule jobs numCompleted = numFailed = 0; Console.WriteLine(); Console.WriteLine("---- ["); foreach (IResult r in schedule.GetResults()) { if (r.ResultStatus != "InProgress") { numCompleted++; } if (r.ResultStatus == "Investigate") { numFailed++; } Console.WriteLine(" {"); Console.WriteLine(" 'id': {0}, 'job': r'''{1}''',", r.Job.Id, r.Job.Name); Console.WriteLine(" 'logs': r'''{0}''',", r.LogLocation); if (r.ResultStatus != "InProgress") { Console.WriteLine(" 'report': r'''{0}''',", submission.GetSubmissionResultReport(r)); } Console.WriteLine(" 'status': '{0}',", r.ResultStatus); Console.WriteLine(" 'pass': {0}, 'fail': {1}, 'notrun': {2}, 'notapplicable': {3}", r.Pass, r.Fail, r.NotRun, r.NotApplicable); Console.WriteLine(" },"); } Console.WriteLine("] ----"); } while (numCompleted < schedule.GetResults().Length&& DateTime.Now < endTime); Console.WriteLine(); // Cancel incomplete jobs foreach (IResult r in schedule.GetResults()) { if (r.ResultStatus == "InProgress") { r.Cancel(); } } // Reset the machines foreach (string machineName in machines) { ResetMachine(rootPool, machineName, false); } // Report failures if (numCompleted < schedule.GetResults().Length) { Console.WriteLine("Some jobs did not complete on time."); } if (numFailed > 0) { Console.WriteLine("Some jobs failed."); } if (numFailed > 0 || numCompleted < schedule.GetResults().Length) { return(1); } Console.WriteLine("All jobs completed."); return(0); } catch (Exception e) { Console.WriteLine("Error: " + e.Message); return(1); } }
// Look for a device in a machine, and if not found, keep trying for 3 minutes static IDevice GetDevice(IResourcePool rootPool, string machineName, string regexStr) { Regex deviceRegex = new Regex(regexStr, RegexOptions.IgnoreCase); int numAttempts = 1; DateTime endTime = DateTime.Now.AddSeconds(180); while (DateTime.Now < endTime) { IResource machine = rootPool.GetResourceByName(machineName); Console.WriteLine("Looking for device '{0}' in machine '{1}' (machine has {2} devices)", regexStr, machineName, machine.GetDevices().Length); foreach (IDevice d in machine.GetDevices()) { if (deviceRegex.IsMatch(d.FriendlyName)) { Console.WriteLine("Found device '{0}'", d.FriendlyName); return d; } } Console.WriteLine("Device not found"); if (numAttempts % 5 == 0) ResetMachine(rootPool, machineName, true); else System.Threading.Thread.Sleep(5000); numAttempts++; } Console.WriteLine("Error: device '{0}' not found", deviceRegex); return null; }