Function test_healer_overcharge

battlecode-engine/src/world.rs:2820–2876

()

    #[test]
    fn test_healer_overcharge() {
        // Create the game world.
        let mut world = GameWorld::test_world();

        // Unlock healer's overcharge ability through research.
        let unlock_level = 3;
        let rounds = 200;

        for _ in 0..unlock_level {
            let my_research = world.my_research_mut();
            assert!(my_research.add_to_queue(&Branch::Healer));
            for _ in 0..rounds {
                my_research.end_round();
            }
        }

        // Unlock robot's ability through research.
        let unlock_level = 3;
        let rounds = 200;

        for _ in 0..unlock_level {
            let my_research = world.my_research_mut();
            assert!(my_research.add_to_queue(&Branch::Knight));
            for _ in 0..rounds {
                my_research.end_round();
            }
        }

        // Create healer and target robots.
        let loc_a = MapLocation::new(Planet::Earth, 0, 0);
        let loc_b = MapLocation::new(Planet::Earth, 0, 1);
        let loc_c = MapLocation::new(Planet::Earth, 0, 20);
        let healer = world.create_unit(Team::Red, loc_a, UnitType::Healer).unwrap();
        let robot_a = world.create_unit(Team::Red, loc_b, UnitType::Knight).unwrap();
        let robot_b = world.create_unit(Team::Red, loc_c, UnitType::Knight).unwrap();

        // Healer overcharge is ready.
        assert!(!world.is_overcharge_ready(healer));
        world.end_round();
        assert!(world.is_overcharge_ready(healer));

        // Healer should not be able to overcharge target robot outside of range.
        assert!(!world.can_overcharge(healer, robot_b));

        // Healer should be able to overcharge target robot within range.
        assert!(world.can_overcharge(healer, robot_a));

        // Robot uses ability.
        let loc_d = MapLocation::new(Planet::Earth, 0, 2);
        world.move_to(robot_b, loc_d);
        assert!(world.javelin(robot_a, robot_b).is_ok());
        assert!(!world.get_unit(robot_a).unwrap().ok_if_ability_ready().is_ok());

        // Healer uses overcharge to reset robot's ablity cooldown
        assert!(world.overcharge(healer, robot_a).is_ok());
        assert!(world.get_unit(robot_a).unwrap().ok_if_ability_ready().is_ok());
    }