Function test_factory

battlecode-engine/src/unit.rs:1389–1434

()

    #[test]
    fn test_factory() {
        // A worker cannot produce a robot.
        let loc = MapLocation::new(Planet::Earth, 0, 0);
        let worker = Unit::new(1, Team::Red, Worker, 0, OnMap(loc)).unwrap();
        assert_err!(worker.ok_if_can_produce_robot(Mage), GameError::InappropriateUnitType);

        // A factory cannot produce a structure, but it can produce a mage.
        let mut factory = Unit::new(1, Team::Red, Factory, 0, OnMap(loc)).unwrap();
        factory.is_built = true;
        assert_err!(factory.factory_rounds_left(), GameError::NullValue);
        assert_err!(factory.ok_if_can_produce_robot(Factory), GameError::InappropriateUnitType);
        assert_err!(factory.ok_if_can_produce_robot(Rocket), GameError::InappropriateUnitType);
        assert!(factory.ok_if_can_produce_robot(Mage).is_ok());

        // The factory cannot produce anything when it's already busy.
        factory.produce_robot(Mage);
        assert!(factory.ok_if_can_produce_robot(Mage).is_err());

        // After a few rounds, the factory can produce again.
        for _ in 0..factory.factory_max_rounds_left().unwrap() - 1 {
            assert_eq!(factory.process_factory_round(), None);
            assert!(factory.ok_if_can_produce_robot(Mage).is_err());
        }
        assert_eq!(factory.process_factory_round(), Some(Mage));
        assert!(factory.ok_if_can_produce_robot(Mage).is_ok());

        // Fill the factory to its max capacity.
        for id in 0..factory.structure_max_capacity().expect("unit has a capacity") {
            factory.load(id as UnitID);
        }

        // The factory can produce one more robot, but it won't go in its garrison.
        assert!(factory.ok_if_can_produce_robot(Mage).is_ok());
        factory.produce_robot(Mage);
        for _ in 0..factory.factory_max_rounds_left().unwrap() * 2 {
            assert_eq!(factory.process_factory_round(), None);
            assert!(factory.ok_if_can_produce_robot(Mage).is_err());
        }

        // After unloading the units, the factory will work again.
        for id in 0..factory.structure_max_capacity().expect("unit has a capacity") {
            assert_eq!(factory.unload_unit(), id as UnitID);
        }
        assert_eq!(factory.process_factory_round(), Some(Mage));
        assert!(factory.ok_if_can_produce_robot(Mage).is_ok());
    }

    #[test]
    fn test_rockets() {