Function test_knight_javelin

battlecode-engine/src/world.rs:2693–2735

()

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

        // Unlock knight's javelin 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 knight 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 knight = world.create_unit(Team::Red, loc_a, UnitType::Knight).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();

        // Knight Javelin is ready
        assert!(!world.is_javelin_ready(knight));
        world.end_round();
        assert!(world.is_javelin_ready(knight));

        // Knight should not be able to javelin target outside of range
        assert!(!world.can_javelin(knight, robot_b));

        // Knight should be able to javelin target within range
        assert!(world.can_javelin(knight, robot_a));

        // Javelin target.
        let robot_max_health = 250;
        let robot_damaged_health = 205;
        assert_eq!(world.get_unit(robot_a).unwrap().health(), robot_max_health);
        assert!(world.javelin(knight, robot_a).is_ok());
        assert_eq!(world.get_unit(robot_a).unwrap().health(), robot_damaged_health);
        assert!(!world.is_javelin_ready(knight));
    }

    #[test]
    fn test_mage_blink() {