def test_setServiceStartupType_ALG(self):
self.assertTrue(winUtility.setServiceStartupType('ALG', 'LocalHost', 'auto'))
self.assertFalse(winUtility.setServiceStartupType('ALG', 'LocalHost', 'junk'))
self.assertEqual(winUtility.getServiceStartupType('ALG', 'LocalHost'), 'AUTO_START (DELAYED)')
self.assertTrue(winUtility.setServiceStartupType('ALG', 'LocalHost', 'manual'))
self.assertEqual(winUtility.getServiceStartupType('ALG', 'LocalHost'), 'DEMAND_START')
self.assertTrue(winUtility.setServiceStartupType('ALG', 'LocalHost', 'disabled'))
self.assertEqual(winUtility.getServiceStartupType('ALG', 'LocalHost'), 'DISABLED')
self.assertTrue(winUtility.setServiceStartupType('ALG', 'LocalHost', 'automatic'))
self.assertEqual(winUtility.getServiceStartupType('ALG', 'LocalHost'), 'AUTO_START (DELAYED)')
self.assertTrue(winUtility.setServiceStartupType('ALG', 'LocalHost', 'manual'))
help="The presence of this parameter indicates server names come from a file",
action="store_true",
default=False,
dest="fileInput",
)
parser.add_argument(
"-i",
"--input",
help="When -f is present, please provide the name of the file that contains a list of servers, each on its own line. Otherwise, please type a list of servers, separated by comma.",
required=True,
dest="iInput",
)
parser.add_argument("-s", "--service", help="Please provide the name of the service", required=True, dest="sService")
parser.add_argument(
"-t", "--type", help="Please provide the type name [manual|auto|disabled]", required=True, dest="tType"
)
argList = parser.parse_args()
if argList.fileInput:
text_file = open("%s" % argList.iInput, "r")
serverList = text_file.readlines()
else:
serverList = argList.iInput.split(",")
startType = {}
for server in serverList:
startType[server.strip()] = setServiceStartupType(argList.sService, server.strip(), argList.tType)
print startType
