def test_setServiceStatus_ALG(self):
self.assertTrue(winUtility.setServiceStatus('ALG', 'LocalHost', 'stop'))
self.assertEqual(winUtility.getServiceState('ALG', 'LocalHost'), 'STOPPED')
self.assertTrue(winUtility.setServiceStatus('ALG', 'LocalHost', 'start'))
self.assertTrue(winUtility.setServiceStatus('ALG', 'LocalHost', 'start'))
self.assertEqual(winUtility.getServiceState('ALG', 'LocalHost'), 'RUNNING')
self.assertTrue(winUtility.setServiceStatus('ALG', 'LocalHost', 'restart'))
self.assertEqual(winUtility.getServiceState('ALG', 'LocalHost'), 'RUNNING')
self.assertTrue(winUtility.setServiceStatus('ALG', 'LocalHost', 'stop'))
self.assertEqual(winUtility.getServiceState('ALG', 'LocalHost'), 'STOPPED')
import argparse
from winUtility import getServiceState
parser = argparse.ArgumentParser(description='Get service status')
parser.add_argument('-f', '--fileInput', 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')
argList = parser.parse_args()
if argList.fileInput:
text_file = open("%s" % argList.iInput, "r")
serverList = text_file.readlines()
else:
serverList = argList.iInput.split(',')
serviceStatus = {}
for server in serverList:
serviceStatus[server.strip()] = getServiceState(argList.sService, server.strip())
print serviceStatus
def test_getServiceState_junk(self):
self.assertEqual(winUtility.getServiceState('junk', 'LocalHost'), None)
def test_getServiceState_namedInstance(self):
self.assertEqual(winUtility.getServiceState('MSSQL$SQLEXPRESS', 'LocalHost').upper(), 'RUNNING')
def test_getServiceState_dhcp(self):
self.assertEqual(winUtility.getServiceState('Dhcp', 'LocalHost'), 'RUNNING')
