bootstrap-vz/tests/integration/tools/__init__.py

# Register deserialization handlers for objects
# that will pass between server and client
from bootstrapvz.remote import register_deserialization_handlers
register_deserialization_handlers()


# Snatched from here: http://stackoverflow.com/a/7205107
def merge_dicts(*args):
	def merge(a, b, path=None):
		if path is None:
			path = []
		for key in b:
			if key in a:
				if isinstance(a[key], dict) and isinstance(b[key], dict):
					merge(a[key], b[key], path + [str(key)])
				elif a[key] == b[key]:
					pass
				else:
					raise Exception('Conflict at %s' % '.'.join(path + [str(key)]))
			else:
						a[key] = b[key]
		return a
	return reduce(merge, args, {})


def waituntil(predicate, timeout=5, interval=0.05):
	import time
	threshhold = time.time() + timeout
	while time.time() < threshhold:
		if predicate():
			return True
		time.sleep(interval)
	return False


def read_from_socket(socket_path, termination_string, timeout):
		import socket
		import select
		import errno
		console = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
		console.connect(socket_path)
		console.setblocking(0)

		output = ''
		ptr = 0
		continue_select = True
		nooutput_for = 0
		select_timeout = .5
		while continue_select:
			read_ready, _, _ = select.select([console], [], [], select_timeout)
			if console in read_ready:
				nooutput_for = 0
				while True:
					try:
						output += console.recv(1024)
						if termination_string in output[ptr:]:
							continue_select = False
						else:
							ptr = len(output) - len(termination_string)
						break
					except socket.error, e:
						if e.errno != errno.EWOULDBLOCK:
							raise Exception(e)
						continue_select = False
			else:
				nooutput_for += select_timeout
			if nooutput_for > timeout:
				from exceptions import SocketReadTimeout
				msg = ('Reading from socket `{path}\' timed out after {seconds} seconds.'
				       .format(path=socket_path, timeout=nooutput_for))
				raise SocketReadTimeout(msg)
		console.close()
		return output


def bootstrap(manifest, build_server):
	from bootstrapvz.remote.build_servers import LocalBuildServer
	if isinstance(build_server, LocalBuildServer):
		from bootstrapvz.base.main import run
		bootstrap_info = run(manifest)
	else:
		from bootstrapvz.remote.main import run
		bootstrap_info = run(manifest, build_server)
	return bootstrap_info


def test(instance):
	pass