mirror of
https://github.com/vmware/vsphere-automation-sdk-python.git
synced 2024-11-22 17:39:59 -05:00
cfa9acdb88
Add unittests for vsphere client Fixed some format issues Run the tests as part of travis ci
134 lines
4.4 KiB
Python
134 lines
4.4 KiB
Python
#!/usr/bin/env python
|
|
|
|
"""
|
|
* *******************************************************
|
|
* Copyright (c) VMware, Inc. 2017, 2018. All Rights Reserved.
|
|
* SPDX-License-Identifier: MIT
|
|
* *******************************************************
|
|
*
|
|
* DISCLAIMER. THIS PROGRAM IS PROVIDED TO YOU "AS IS" WITHOUT
|
|
* WARRANTIES OR CONDITIONS OF ANY KIND, WHETHER ORAL OR WRITTEN,
|
|
* EXPRESS OR IMPLIED. THE AUTHOR SPECIFICALLY DISCLAIMS ANY IMPLIED
|
|
* WARRANTIES OR CONDITIONS OF MERCHANTABILITY, SATISFACTORY QUALITY,
|
|
* NON-INFRINGEMENT AND FITNESS FOR A PARTICULAR PURPOSE.
|
|
"""
|
|
|
|
__author__ = 'VMware, Inc.'
|
|
__vcenter_version__ = '6.7+'
|
|
|
|
|
|
from samples.vsphere.common import sample_cli
|
|
from samples.vsphere.common import sample_util
|
|
from samples.vsphere.common import vapiconnect
|
|
from tabulate import tabulate
|
|
|
|
from com.vmware.appliance.logging_client import Forwarding
|
|
|
|
|
|
class LogForwarding(object):
|
|
"""
|
|
Demonstrates log forwarding API operations
|
|
|
|
Prerequisites:
|
|
- vCenter
|
|
- Log host listening to syslog packets over any of the supported
|
|
protocols UDP/TCP/TLS
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.loghost = None
|
|
self.protocol = None
|
|
self.port = None
|
|
self.stub_config = None
|
|
self.log_forwarding_client = None
|
|
|
|
def setup(self):
|
|
parser = sample_cli.build_arg_parser()
|
|
|
|
parser.add_argument('--loghost',
|
|
required=True,
|
|
action='store',
|
|
help='The log host')
|
|
parser.add_argument('--port',
|
|
required=True,
|
|
action='store',
|
|
help='The log host port number')
|
|
parser.add_argument('--protocol',
|
|
required=True,
|
|
action='store',
|
|
help='The log host protocol (TCP/UDP/TLS)')
|
|
|
|
args = sample_util.process_cli_args(parser.parse_args())
|
|
self.loghost = args.loghost
|
|
self.protocol = args.protocol
|
|
self.port = int(args.port)
|
|
|
|
# Connect to vAPI services
|
|
self.stub_config = vapiconnect.connect(
|
|
host=args.server,
|
|
user=args.username,
|
|
pwd=args.password,
|
|
skip_verification=args.skipverification)
|
|
|
|
self.log_forwarding_client = Forwarding(self.stub_config)
|
|
|
|
def run(self):
|
|
# Set log forwarding configuration
|
|
self.set_log_forwarding()
|
|
|
|
# Get log forwarding configuration
|
|
self.get_log_forwarding()
|
|
|
|
# Test log forwarding configuration
|
|
self.test_log_forwarding()
|
|
|
|
# Update log forwarding configuration
|
|
self.update_log_forwarding()
|
|
|
|
def set_log_forwarding(self):
|
|
log_forwarding_config = [Forwarding.Config(hostname=self.loghost,
|
|
port=self.port,
|
|
protocol=self.protocol)]
|
|
self.log_forwarding_client.set(log_forwarding_config)
|
|
|
|
def get_log_forwarding(self):
|
|
configs = self.log_forwarding_client.get()
|
|
|
|
print("\nLog forwarding configurations:")
|
|
table = [[cfg.hostname, cfg.port, cfg.protocol] for cfg in configs]
|
|
headers = ["Loghost", "Port", "Protocol"]
|
|
print(tabulate(table, headers))
|
|
|
|
def test_log_forwarding(self):
|
|
test_response = self.log_forwarding_client.test(True)
|
|
|
|
print("\nLog forwarding test response:")
|
|
table = [[resp.hostname,
|
|
resp.state,
|
|
resp.message.default_message if resp.message else None]
|
|
for resp in test_response]
|
|
headers = ["Loghost", "State", "Message"]
|
|
print(tabulate(table, headers))
|
|
|
|
def update_log_forwarding(self):
|
|
# Read log forwarding configuration
|
|
log_forwarding_config = self.log_forwarding_client.get()
|
|
|
|
# Delete the newly added configuration
|
|
log_forwarding_config = list(filter(
|
|
lambda cfg: cfg.hostname != self.loghost,
|
|
log_forwarding_config))
|
|
|
|
# Apply the modified log forwarding configuration
|
|
self.log_forwarding_client.set(log_forwarding_config)
|
|
|
|
|
|
def main():
|
|
log_forwarding = LogForwarding()
|
|
log_forwarding.setup()
|
|
log_forwarding.run()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|