class zzzTestRuleSecureMDNS(RuleTest):
def setUp(self):
RuleTest.setUp(self)
self.rule = SecureMDNS(self.config,
self.environ,
self.logdispatch,
self.statechglogger)
self.rulename = self.rule.rulename
self.rulenumber = self.rule.rulenumber
self.ch = CommandHelper(self.logdispatch)
self.plb = "/usr/libexec/PlistBuddy"
self.sh = ServiceHelper(self.environ, self.logdispatch)
self.service = ""
self.serviceTarget=""
def tearDown(self):
if os.path.exists(self.service + ".stonixtmp"):
os.rename(self.service + ".stonixtmp", self.service)
def runTest(self):
self.simpleRuleTest()
def setConditionsForRule(self):
'''
Configure system for the unit test
@param self: essential if you override this definition
@return: boolean - If successful True; If failure False
@author: ekkehard j. koch
'''
success = True
if self.environ.getosfamily() == "darwin":
success = False
osxversion = str(self.environ.getosver())
if osxversion.startswith("10.10.0") or \
osxversion.startswith("10.10.1") or \
osxversion.startswith("10.10.2") or \
osxversion.startswith("10.10.3"):
debug = "Using discoveryd LaunchDaemon"
self.logdispatch.log(LogPriority.DEBUG, debug)
service = \
"/System/Library/LaunchDaemons/com.apple.discoveryd.plist"
servicename = "com.apple.networking.discoveryd"
parameter = "--no-multicast"
plistText = readFile(service, self.logdispatch)
newPlistText = re.sub("<string>" + parameter + "</string>",
"", "".join(plistText))
success = True
else:
debug = "Using mDNSResponder LaunchDaemon"
self.logdispatch.log(LogPriority.DEBUG, debug)
service = "/System/Library/LaunchDaemons/" + \
"com.apple.mDNSResponder.plist"
if osxversion.startswith("10.10"):
servicename = "com.apple.mDNSResponder.reloaded"
parameter = "-NoMulticastAdvertisements"
else:
servicename = "com.apple.mDNSResponder"
parameter = "-NoMulticastAdvertisements"
plistText = readFile(service, self.logdispatch)
newPlistText = re.sub("<string>" + parameter + "</string>",
"", "".join(plistText))
success = True
self.service = service
if success and self.sh.auditService(service, serviceTarget=servicename):
success = writeFile(service + ".stonixtmp", "".join(plistText),
self.logdispatch)
success = writeFile(service, newPlistText, self.logdispatch)
if success and self.sh.auditService(service, serviceTarget=servicename):
success = self.sh.reloadService(service, serviceTarget=servicename)
else:
ph = Pkghelper(self.logdispatch, self.environ)
package = "avahi-daemon"
service = "avahi-daemon"
if (ph.determineMgr() == "yum" or ph.determineMgr() == "dnf"):
package = "avahi"
path = "/etc/sysconfig/network"
if os.path.exists(path):
tmppath = path + ".tmp"
data = {"NOZEROCONF": "yes"}
editor = KVEditorStonix(self.statechglogger,
self.logdispatch, "conf",
path, tmppath, data,
"notpresent", "closedeq")
if not editor.report():
if editor.fix():
if not editor.commit():
success = False
else:
success = False
elif ph.determineMgr() == "zypper":
package = "avahi"
if not ph.check(package) and ph.checkAvailable(package):
success = ph.install(package)
if success and not self.sh.auditService(service, serviceTarget=self.serviceTarget):
self.sh.enableService(service, serviceTarget=self.serviceTarget)
return success
def checkReportForRule(self, pCompliance, pRuleSuccess):
'''
check on whether report was correct
@param self: essential if you override this definition
@param pCompliance: the self.iscompliant value of rule
@param pRuleSuccess: did report run successfully
@return: boolean - If successful True; If failure False
@author: ekkehard j. koch
'''
self.logdispatch.log(LogPriority.DEBUG, "pCompliance = " +
str(pCompliance) + ".")
self.logdispatch.log(LogPriority.DEBUG, "pRuleSuccess = " +
str(pRuleSuccess) + ".")
success = True
return success
def checkFixForRule(self, pRuleSuccess):
'''
check on whether fix was correct
@param self: essential if you override this definition
@param pRuleSuccess: did report run successfully
@return: boolean - If successful True; If failure False
@author: ekkehard j. koch
'''
self.logdispatch.log(LogPriority.DEBUG, "pRuleSuccess = " +
str(pRuleSuccess) + ".")
success = True
return success
def checkUndoForRule(self, pRuleSuccess):
'''
check on whether undo was correct
@param self: essential if you override this definition
@param pRuleSuccess: did report run successfully
@return: boolean - If successful True; If failure False
@author: ekkehard j. koch
'''
self.logdispatch.log(LogPriority.DEBUG, "pRuleSuccess = " +
str(pRuleSuccess) + ".")
success = True
return success
class zzzTestFrameworkServiceHelper(unittest.TestCase):
'''Class docs'''
def setUp(self):
'''initialize and set class variables and objects'''
self.environ = Environment()
self.environ.setdebugmode(True)
self.logger = LogDispatcher(self.environ)
self.mysh = ServiceHelper(self.environ, self.logger)
# set service name
self.myservice = 'crond'
self.myservicename = ""
if self.environ.getosfamily() == 'darwin':
self.myservice = "/Library/LaunchDaemons/gov.lanl.stonix.report.plist"
self.myservicename = "gov.lanl.stonix.report"
elif self.environ.getosfamily() == 'solaris':
self.myservice = 'svc:/system/cron:default'
elif self.environ.getosfamily() == 'freebsd':
self.myservice = 'cron'
elif os.path.exists('/usr/lib/systemd/system/cron.service'):
self.myservice = 'cron.service'
elif os.path.exists('/usr/lib/systemd/system/crond.service'):
self.myservice = 'crond.service'
elif os.path.exists('/etc/init.d/vixie-cron'):
self.myservice = 'vixie-cron'
elif os.path.exists('/etc/init.d/cron'):
self.myservice = 'cron'
if self.environ.getosfamily() == "darwin":
self.service = self.myservice, self.myservicename
else:
self.service = [self.myservice]
# store system initial state
self.orig_enabled = self.mysh.auditService(*self.service)
def tearDown(self):
'''restore system initial state'''
if self.orig_enabled:
self.mysh.enableService(*self.service)
else:
self.mysh.disableService(*self.service)
def testListServices(self):
'''test listing of services'''
services = self.mysh.listServices()
self.assertGreater(len(services), 0)
self.assertIsInstance(services, list)
def testDisable(self):
'''test disabling a service from initial state:
enabled
'''
# make sure service is started, so stopping it will be a valid test of the function
if not self.mysh.auditService(*self.service):
self.mysh.enableService(*self.service)
disabled = self.mysh.disableService(*self.service)
self.assertTrue(disabled)
def testEnable(self):
'''test enabling a service from initial state:
disabled
'''
# make sure service is stopped, so starting it will be a valid test of the function
if self.mysh.auditService(*self.service):
self.mysh.disableService(*self.service)
enabled = self.mysh.enableService(*self.service)
self.assertTrue(enabled)
def testReloadService(self):
'''test reloading a service from both initial states:
enabled
disabled
'''
self.mysh.disableService(*self.service)
reloaded1 = self.mysh.reloadService(*self.service)
self.assertTrue(reloaded1)
self.mysh.enableService(*self.service)
reloaded2 = self.mysh.reloadService(*self.service)
self.assertTrue(reloaded2)
def testIsRunning(self):
'''test status checking to see if a service
is running
(start and stop not implemented in all helpers)
'''
if self.mysh.startService(*self.service):
self.assertTrue(self.mysh.isRunning(*self.service))
if self.mysh.stopService(*self.service):
self.assertFalse(self.mysh.isRunning(*self.service))
class zzzTestFrameworkServiceHelper(unittest.TestCase):
"""
Class docs
"""
def setUp(self):
"""
initialize and set class variables and objects
"""
self.environ = Environment()
self.environ.setdebugmode(True)
self.logger = LogDispatcher(self.environ)
self.mysh = ServiceHelper(self.environ, self.logger)
# set service name
self.myservice = 'crond'
self.myservicename = ""
if self.environ.getosfamily() == 'darwin':
self.myservice = "/Library/LaunchDaemons/gov.lanl.stonix.report.plist"
self.myservicename = "gov.lanl.stonix.report"
elif self.environ.getosfamily() == 'solaris':
self.myservice = 'svc:/system/cron:default'
elif self.environ.getosfamily() == 'freebsd':
self.myservice = 'cron'
elif os.path.exists('/usr/lib/systemd/system/cron.service'):
self.myservice = 'cron.service'
elif os.path.exists('/usr/lib/systemd/system/crond.service'):
self.myservice = 'crond.service'
elif os.path.exists('/etc/init.d/vixie-cron'):
self.myservice = 'vixie-cron'
elif os.path.exists('/etc/init.d/cron'):
self.myservice = 'cron'
if self.environ.getosfamily() == "darwin":
self.service = self.myservice, self.myservicename
else:
self.service = [self.myservice]
# store system initial state
self.orig_enabled = self.mysh.auditService(*self.service)
def tearDown(self):
"""
restore system initial state
"""
if self.orig_enabled:
self.mysh.enableService(*self.service)
else:
self.mysh.disableService(*self.service)
def testListServices(self):
"""
test listing of services
"""
services = self.mysh.listServices()
self.assertGreater(len(services), 0)
self.assertIsInstance(services, list)
def testDisable(self):
"""
test disabling a service from initial state:
enabled
"""
# make sure service is started, so stopping it will be a valid test of the function
if not self.mysh.auditService(*self.service):
self.mysh.enableService(*self.service)
disabled = self.mysh.disableService(*self.service)
self.assertTrue(disabled)
def testEnable(self):
"""
test enabling a service from initial state:
disabled
"""
# make sure service is stopped, so starting it will be a valid test of the function
if self.mysh.auditService(*self.service):
self.mysh.disableService(*self.service)
enabled = self.mysh.enableService(*self.service)
self.assertTrue(enabled)
def testReloadService(self):
"""
test reloading a service from both initial states:
enabled
disabled
"""
self.mysh.disableService(*self.service)
reloaded1 = self.mysh.reloadService(*self.service)
self.assertTrue(reloaded1)
self.mysh.enableService(*self.service)
reloaded2 = self.mysh.reloadService(*self.service)
self.assertTrue(reloaded2)
def testIsRunning(self):
"""
test status checking to see if a service
is running
(start and stop not implemented in all helpers)
"""
if self.mysh.startService(*self.service):
self.assertTrue(self.mysh.isRunning(*self.service))
if self.mysh.stopService(*self.service):
self.assertFalse(self.mysh.isRunning(*self.service))
class zzzTestRuleSecureMDNS(RuleTest):
def setUp(self):
RuleTest.setUp(self)
self.rule = SecureMDNS(self.config, self.environ, self.logdispatch,
self.statechglogger)
self.rulename = self.rule.rulename
self.rulenumber = self.rule.rulenumber
self.ch = CommandHelper(self.logdispatch)
self.plb = "/usr/libexec/PlistBuddy"
self.sh = ServiceHelper(self.environ, self.logdispatch)
self.service = ""
self.serviceTarget = ""
def tearDown(self):
if os.path.exists(self.service + ".stonixtmp"):
os.rename(self.service + ".stonixtmp", self.service)
def runTest(self):
self.simpleRuleTest()
def setConditionsForRule(self):
'''Configure system for the unit test
:param self: essential if you override this definition
:returns: boolean - If successful True; If failure False
@author: ekkehard j. koch
'''
success = True
if self.environ.getosfamily() == "darwin":
success = False
osxversion = str(self.environ.getosver())
if osxversion.startswith("10.10.0") or \
osxversion.startswith("10.10.1") or \
osxversion.startswith("10.10.2") or \
osxversion.startswith("10.10.3"):
debug = "Using discoveryd LaunchDaemon"
self.logdispatch.log(LogPriority.DEBUG, debug)
service = \
"/System/Library/LaunchDaemons/com.apple.discoveryd.plist"
servicename = "com.apple.networking.discoveryd"
parameter = "--no-multicast"
plistText = readFile(service, self.logdispatch)
newPlistText = re.sub("<string>" + parameter + "</string>", "",
"".join(plistText))
success = True
else:
debug = "Using mDNSResponder LaunchDaemon"
self.logdispatch.log(LogPriority.DEBUG, debug)
service = "/System/Library/LaunchDaemons/" + \
"com.apple.mDNSResponder.plist"
if osxversion.startswith("10.10"):
servicename = "com.apple.mDNSResponder.reloaded"
parameter = "-NoMulticastAdvertisements"
else:
servicename = "com.apple.mDNSResponder"
parameter = "-NoMulticastAdvertisements"
plistText = readFile(service, self.logdispatch)
newPlistText = re.sub("<string>" + parameter + "</string>", "",
"".join(plistText))
success = True
self.service = service
if success and self.sh.auditService(service,
serviceTarget=servicename):
success = writeFile(service + ".stonixtmp", "".join(plistText),
self.logdispatch)
success = writeFile(service, newPlistText, self.logdispatch)
if success and self.sh.auditService(service,
serviceTarget=servicename):
success = self.sh.reloadService(service,
serviceTarget=servicename)
else:
ph = Pkghelper(self.logdispatch, self.environ)
package = "avahi-daemon"
service = "avahi-daemon"
if (ph.determineMgr() == "yum" or ph.determineMgr() == "dnf"):
package = "avahi"
path = "/etc/sysconfig/network"
if os.path.exists(path):
tmppath = path + ".tmp"
data = {"NOZEROCONF": "yes"}
editor = KVEditorStonix(self.statechglogger,
self.logdispatch, "conf", path,
tmppath, data, "notpresent",
"closedeq")
if not editor.report():
if editor.fix():
if not editor.commit():
success = False
else:
success = False
elif ph.determineMgr() == "zypper":
package = "avahi"
if not ph.check(package) and ph.checkAvailable(package):
success = ph.install(package)
if success and not self.sh.auditService(
service, serviceTarget=self.serviceTarget):
self.sh.enableService(service,
serviceTarget=self.serviceTarget)
return success
def checkReportForRule(self, pCompliance, pRuleSuccess):
'''check on whether report was correct
:param self: essential if you override this definition
:param pCompliance: the self.iscompliant value of rule
:param pRuleSuccess: did report run successfully
:returns: boolean - If successful True; If failure False
@author: ekkehard j. koch
'''
self.logdispatch.log(LogPriority.DEBUG,
"pCompliance = " + str(pCompliance) + ".")
self.logdispatch.log(LogPriority.DEBUG,
"pRuleSuccess = " + str(pRuleSuccess) + ".")
success = True
return success
def checkFixForRule(self, pRuleSuccess):
'''check on whether fix was correct
:param self: essential if you override this definition
:param pRuleSuccess: did report run successfully
:returns: boolean - If successful True; If failure False
@author: ekkehard j. koch
'''
self.logdispatch.log(LogPriority.DEBUG,
"pRuleSuccess = " + str(pRuleSuccess) + ".")
success = True
return success
def checkUndoForRule(self, pRuleSuccess):
'''check on whether undo was correct
:param self: essential if you override this definition
:param pRuleSuccess: did report run successfully
:returns: boolean - If successful True; If failure False
@author: ekkehard j. koch
'''
self.logdispatch.log(LogPriority.DEBUG,
"pRuleSuccess = " + str(pRuleSuccess) + ".")
success = True
return success