def test_unload_cargo_raises_exception_if_tray_cannot_hold_the_resource(self):  # noqa
		tray = SlottedCargoContainer()
		with self.assertRaises(CargoContainerException) as error_context:
			tray.unload_cargo(Fish, 1)
		self.assertIn(
			"A cargo slot of that type has not been added",
			error_context.exception.message,
			)
	def test_unload_cargo_decreases_load_of_slot_if_tray_can_hold_the_resource_and_all_available_cargo_up_to_the_requested_quantity(self):  # noqa
		tray = SlottedCargoContainer()
		tray.add_resource_slot(Fish, 3)
		tray.get_slot(Fish)['load'] = 3
		load = tray.unload_cargo(Fish, 7)
		self.assertEqual(tray.current_load(Fish), 0)
		self.assertEqual(load, 3)
	def test_unload_cargo_returns_nothing_if_slot_is_empty(self):
		tray = SlottedCargoContainer()
		tray.add_resource_slot(Fish, 3)
		load = tray.unload_cargo(Fish, 7)
		self.assertEqual(tray.current_load(Fish), 0)
		self.assertEqual(load, 0)
Example #4
0
class Building:

	name = 'Unnamed'
	build_time = 1.0

	def __init__(self):
		self.resource_plants = []
		self.unit_factories = {}
		self.container = SlottedCargoContainer()
		self.building_id = uuid4()
		self.harvestable_resources = set()
		self.producable_resources = set()
		self.pt = None
		self.action_queue = []

	@property
	def status(self):
		return '\n'.join([
			'building type:    {}'.format(self.name),
			'current position: {}'.format(self.pt),
			'current action:   {}'.format(self.current_action),
			'container:\n{}'.format(indent(str(self.container), '  '))
			])

	@property
	def current_action(self):
		if len(self.action_queue) == 0:
			return None
		else:
			return self.action_queue[0]

	def act(self, dt):
		if len(self.action_queue) == 0:
			return
		action = self.action_queue.pop(0)
		if not action.is_possible(self, dt):
			return
		action.execute(self, dt)
		if action.is_complete(self, dt):
			action.finish(self, dt)
		else:
			self.add_immediate_action(action.next_action(self, dt))

	def add_action(self, action):
		self.action_queue.append(action)

	def add_immediate_action(self, action):
		self.action_queue.insert(0, action)

	def clear_actions(self):
		self.action_queue = []

	def add_producable_resource(self, resource):
		self.producable_resources.add(resource)

	def add_resource_plant(self, resource_plant):
		self.resource_plants.append(resource_plant)

	def add_unit_factory(self, unit_factory):
		if unit_factory.product.name in self.unit_factories:
			message = "A factory for that unit has already been added"
			raise BuildingException(message)
		self.unit_factories[unit_factory.product.name] = unit_factory

	def receive_cargo(self, resource_type, quantity):
		return self.container.load_cargo(resource_type, quantity)

	def deliver_cargo(self, resource_type, quantity):
		return self.container.unload_cargo(resource_type, quantity)

	def produce_resources(self):
		[d.digest(self.container) for d in self.resource_plants]

	def build_unit(self, unit):
		if unit.name not in self.unit_factories:
			raise BuildingException("This building cannot build that unit")
		return self.unit_factories[unit.name].digest(self.container)

	def __repr__(self):
		lines = [
			'building_id: {}'.format(self.building_id),
			'name:        {}'.format(self.name),
			'container: \n{}'.format(indent(str(self.container), '  '))
			]
		for (index, resource_plant) in enumerate(self.resource_plants):
			lines.append('resource_plant {}:')
			lines.append(indent(str(resource_plant), '  '))
		for (index, unit_factory) in enumerate(self.unit_factories):
			lines.append('unit_factory {}:')
			lines.append(indent(str(unit_factory), '  '))
		return '\n'.join(lines)