Ejemplo n.º 1
0
	def test__execute_will_not_load_more_resources_than_the_requested_quantity_even_if_the_rate_by_dt_is_larger(self):  # noqa
		dummy_resource = MagicMock()
		dummy_resource.harvest_rate = 3
		dummy_unit = MagicMock()
		harvest = Harvest(dummy_resource, 2.0)
		harvest._execute(dummy_unit, 1.0)
		dummy_unit.container.load_cargo.assert_called_once_with(
			dummy_resource,
			2.0,
			)
		self.assertEqual(harvest.quantity, 0.0)
Ejemplo n.º 2
0
	def test__execute_loads_resource_into_units_cargo_container_based_on_the_harvest_rate_and_dt(self):  # noqa
		dummy_resource = MagicMock()
		dummy_resource.harvest_rate = 0.5
		dummy_unit = MagicMock()
		dummy_unit.container.remaining_capacity.return_value = 2.0
		harvest = Harvest(dummy_resource, 1.5)

		harvest._execute(dummy_unit, 2.5)
		self.assertEqual(harvest.quantity, 0.5)
		dummy_unit.container.load_cargo.assert_called_once_with(
			dummy_resource,
			1,
		)