source_sdk_2013 0.1.0-alpha.8

//! Tests of entities' classes, names, parents' names, spawn flags and teams,
//! read from the members their datamaps declare.

use crate::test_support::entities::{MockEntity, set_datamap, state_maps};
use crate::test_support::sdk_core::change_tracking_engine;
use std::ptr::null;

#[test]
fn classes_are_those_of_the_entitys_datamaps() {
	let mut mock = MockEntity::new(5);

	set_datamap(state_maps(vec![
		(c"CTFPlayer", vec![]),
		(c"CBaseAnimating", vec![]),
	]));

	let entity = mock.entity();

	assert!(entity.is_a(c"CTFPlayer"));
	assert!(entity.is_a(c"CBaseAnimating"));
	assert!(entity.is_a(c"CBaseEntity"));
	assert!(!entity.is_a(c"CObjectSentrygun"));
}

#[test]
fn names_are_read_from_their_pooled_strings() {
	let mut mock = MockEntity::new(5);

	set_datamap(state_maps(vec![]));
	assert_eq!(mock.entity().name(), None);
	assert_eq!(mock.entity().parent_name(), Ok(None));

	mock.set_name(sys::string_t {
		pszValue: c"door".as_ptr(),
	});
	mock.state().parent_name = sys::string_t {
		pszValue: c"train".as_ptr(),
	};
	assert_eq!(mock.entity().name(), Some(c"door"));
	assert_eq!(mock.entity().parent_name(), Ok(Some(c"train")));

	// Empty names are no names.
	mock.set_name(sys::string_t {
		pszValue: c"".as_ptr(),
	});
	mock.state().parent_name = sys::string_t { pszValue: null() };
	assert_eq!(mock.entity().name(), None);
	assert_eq!(mock.entity().parent_name(), Ok(None));
}

#[test]
fn spawn_flags_and_teams_are_read_and_written() {
	let mut mock = MockEntity::new(5);
	let engine = change_tracking_engine();

	set_datamap(state_maps(vec![]));
	mock.state().spawn_flags = 1 << 4;
	mock.state().team = 3;

	assert_eq!(mock.entity().spawn_flags(), Ok(1 << 4));
	assert_eq!(mock.entity().team(), Ok(3));

	mock.entity().set_spawn_flags(engine, 1 << 9 | 1).unwrap();
	assert_eq!(mock.state().spawn_flags, 1 << 9 | 1);
	assert_eq!(mock.entity().spawn_flags(), Ok(1 << 9 | 1));
}