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)
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)