def setUp(self): sys.argv = sys.argv[:1] args = [ '-H', '10.20.104.223', '-l', '9NagiosDC', '-p', 'NglP(23M,n', '-d', 'corp', ] sys.argv.extend(args) self.plugin = NagiosPluginWMI()
class TestWMIPlugin(unittest.TestCase): """Test WMI plugin class.""" def setUp(self): sys.argv = sys.argv[:1] args = [ '-H', '10.20.104.223', '-l', '9NagiosDC', '-p', 'NglP(23M,n', '-d', 'corp', ] sys.argv.extend(args) self.plugin = NagiosPluginWMI() def test_get_hostname(self): """Test retrieving host name using WMI.""" result = self.plugin.execute('SELECT * FROM Win32_OperatingSystem') self.assertEqual(result[0]['CSName'], 'WWGRPCTS6401')