#

# CDDL HEADER START

#

# The contents of this file are subject to the terms of the

# Common Development and Distribution License (the "License").

# You may not use this file except in compliance with the License.

#

# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE

# or http://www.opensolaris.org/os/licensing.

# See the License for the specific language governing permissions

# and limitations under the License.

#

# When distributing Covered Code, include this CDDL HEADER in each

# file and include the License file at usr/src/OPENSOLARIS.LICENSE.

# If applicable, add the following below this CDDL HEADER, with the

# fields enclosed by brackets "[]" replaced with your own identifying

# information: Portions Copyright [yyyy] [name of copyright owner]

#

# CDDL HEADER END

#

#

import socket

import rad.client as rad

import rad.adaptor as adapt

import sys

# Make connection (AF_UNIX socket lets us bypass authentication)

s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM);

#s = socket.socket(socket.AF_INET, socket.SOCK_STREAM);

#s.connect(("localhost", 1234))

rc = rad.RadConnection(s)

# Get an authorization token from authenticated session

#auth = rc.get_object_s("org.opensolaris.os.rad", [("type", "authentication")])

#block = auth.login("C", "root");

#print block.type

#for m in block.messages:

#	print m.style

#	print m.message

#block = auth.submit(["l1admin"])

#print block.type

# Continue using newly authenticated session

# Obtain service object for application/pkg/server:default

obj = adapt.Adaptor(rc.get_object_s("com.oracle.solaris.zonesManagement",

    [("type", "zonesManager")]))

IpType = obj._object.types.IpType

ZoneConfig = obj._object.types.ZoneConfig

print ""

print "Test getSystemState"

try:

	systemState = obj.getSystemState(0, 0, 0, 0, None)

	print "\tPASS"

except:

	print "\tFAIL"

print ""

print "Test zone name lists"

try:

	print "\tzones     :", systemState.zoneNames

	print "\tconfigured:", systemState.zonesConfigured

	print "\tinstalled :", systemState.zonesInstalled

	print "\trunning   :", systemState.zonesRunning

	print "\tPASS"

except:

	print "\tFAIL"

print ""

print "Test print zone states"

try:

	for state in systemState.zones:

		cfg = obj.getZoneConfig(state.name)

		print ""

		print "\t", state.name

		print "\tstate: ", state.state

		print "\tconfigured:"

		print "\tzonepath: ", cfg.zonePath

		print "\tiptype  : ", cfg.iptype

		print "\tscheduler: ", cfg.scheduler

		for network in cfg.networks:

			print "\tnetwork : ", network.physical

			if network.address:

				print "\t          ", network.address

		print "    installed:"

		if state.name in systemState.zonesInstalled:

			print "\tzonepath: ", state.installedState.zonepath

			print "\tuuid    : ", state.installedState.uuid

		print "    running:"

		if state.name in systemState.zonesRunning:

			print "\tiptype  : ", state.runningState.iptype

			for network in state.runningState.networks:

				print "\tnetwork : ", network.physical

				if network.address:

					print "\t          ", network.address

except:

	print "\tFAIL"

print ""

print "Test configure and unconfigure zone"

try:

	cfg = ZoneConfig("test-unconfig", "/export/test-unconfig",

	    "FSS", "1", IpType.SHARED, None, None, None, None)

	obj.zoneConfigure(cfg);

	cfg = obj.getZoneConfig("test-unconfig");

	obj.zoneUnconfigure("test-unconfig");

	try:

		obj.getZoneConfig("test-unconfig");

		print "\tFAIL"

	except rad.ObjectError, ex:

		print "\tPASS"

except rad.ObjectError, ex:

	print "\tFAIL"

	print ex.get_payload()

print ""

print "Test configure, install, boot, reboot, halt, uninstall, unconfigure"

try:

	cfg = ZoneConfig("test-install", "/export/test-install",

	    "FSS", "1", IpTypes.SHARED, None, None, None, None)

	obj.zoneConfigure(cfg);

	obj.zoneInstall("test-install", []);

	obj.zoneBoot("test-install")

	obj.zoneReboot("test-install")

	obj.zoneHalt("test-install")

	obj.zoneUninstall("test-install");

	obj.zoneUnconfigure("test-install");

	print "\tPASS"

except rad.ObjectError, ex:

	print "\tFAIL"

	err = ex.get_payload()

	print err;

print ""

print "Test boot non-existent zone:"

try:

	res = obj.zoneBoot("nosuchzone")

	print "\tFAILED"

except rad.ObjectError, ex:

	error = ex.get_payload();

	print "\t", error.message

	print "\tPASS"

print ""

print "Test setZoneConfig"

try:

	# Configure zone

	zoneConfig = ZoneConfig("test", "/export/oldpath", "", "",

	    IpType.SHARED, None, None, None, None)

	obj.zoneConfigure(zoneConfig);

	# Change zonepath

	newConfig = ZoneConfig("test", "/export/newpath", "", "",

	    IpType.SHARED, None, None, None, None)

	obj.setZoneConfig("test", newConfig);

	# Change zonename

	newConfig = ZoneConfig("test-new", "/export/newpath", "", "",

	    IpType.SHARED, None, None, None, None)

	obj.setZoneConfig("test", newConfig);

	# Change scheduler and cpu shares

	newConfig = ZoneConfig("test-new", "/export/newpath", "FSS", "10",

	    IpType.SHARED, None, None, None, None)

	obj.setZoneConfig("test-new", newConfig);

	# Unconfigure zones

	obj.zoneUnconfigure("test-new");

	print "\tPASS"

except rad.ObjectError, ex:

	error = ex.get_payload();

	print error.message

	print "\tFAILED"

s.close()