summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authortom barrett <spalf0@gmail.com>2019-03-21 09:50:47 -0500
committertom barrett <spalf0@gmail.com>2019-03-21 09:50:47 -0500
commit95902608506ae8ccd1203cbbf93efbe876384a1f (patch)
tree5ce5d578e11b8a5632fad7b38473d49a4d20d7fb /tests
parent6fb0b0d918963a22aeb4050139e8d0000c74a9ff (diff)
added acquiring items of tractorbeam and tests, moved control system over to math and gave it constants
Diffstat (limited to 'tests')
-rw-r--r--tests/tests.rs150
1 files changed, 128 insertions, 22 deletions
diff --git a/tests/tests.rs b/tests/tests.rs
index 574e9f6..b80be21 100644
--- a/tests/tests.rs
+++ b/tests/tests.rs
@@ -28,8 +28,8 @@ mod tests {
(ship, masses)
}
- fn setup_ship_target(ship: &mut Mass, masses: &mut HashMap<String, Mass>) {
- ship.give_received_data(ModuleType::Navigation, String::from("astroid"));
+ fn setup_ship_target(ship: &mut Mass, target_name: String, masses: &mut HashMap<String, Mass>) {
+ ship.give_received_data(ModuleType::Navigation, target_name);
ship.process(masses);
sleep(Duration::from_secs(constants::SHIP_NAVIGATION_TIME + 1));
ship.process(masses);
@@ -56,7 +56,7 @@ mod tests {
#[test]
fn test_navigation_range_targeted() {
let (mut ship, mut masses) = setup();
- setup_ship_target(&mut ship, &mut masses);
+ setup_ship_target(&mut ship, String::from("astroid"), &mut masses);
let data = ship.get_client_data(ModuleType::Navigation, &masses);
let navigation_data: navigation::ClientData = serde_json::from_str(&data).unwrap();
@@ -73,7 +73,7 @@ mod tests {
#[test]
fn test_mining_range() {
let (mut ship, mut masses) = setup();
- setup_ship_target(&mut ship, &mut masses);
+ setup_ship_target(&mut ship, String::from("astroid"), &mut masses);
ship.give_received_data(ModuleType::Mining, String::from("F"));
ship.process(&mut masses);
@@ -92,7 +92,7 @@ mod tests {
#[test]
fn test_mining_navigation_range() {
let (mut ship, mut masses) = setup();
- setup_ship_target(&mut ship, &mut masses);
+ setup_ship_target(&mut ship, String::from("astroid"), &mut masses);
ship.give_received_data(ModuleType::Mining, String::from("F"));
ship.process(&mut masses);
@@ -111,7 +111,7 @@ mod tests {
#[test]
fn test_mining() {
let (mut ship, mut masses) = setup();
- setup_ship_target(&mut ship, &mut masses);
+ setup_ship_target(&mut ship, String::from("astroid"), &mut masses);
ship.give_received_data(ModuleType::Mining, String::from("F"));
ship.process(&mut masses);
@@ -127,7 +127,7 @@ mod tests {
#[test]
fn test_mining_storage() {
let (mut ship, mut masses) = setup();
- setup_ship_target(&mut ship, &mut masses);
+ setup_ship_target(&mut ship, String::from("astroid"), &mut masses);
for _ in 0..10 {
ship.give_item(Item::new(ItemType::CrudeMinerals));
@@ -144,7 +144,7 @@ mod tests {
#[test]
fn test_refinery() {
let (mut ship, mut masses) = setup();
- setup_ship_target(&mut ship, &mut masses);
+ setup_ship_target(&mut ship, String::from("astroid"), &mut masses);
ship.give_received_data(ModuleType::Refinery, String::from("R"));
ship.process(&mut masses);
@@ -197,7 +197,7 @@ mod tests {
#[test]
fn test_engines() {
let (mut ship, mut masses) = setup();
- setup_ship_target(&mut ship, &mut masses);
+ setup_ship_target(&mut ship, String::from("astroid"), &mut masses);
let mut astroid = masses.remove("astroid").unwrap();
astroid.velocity = Vector::new(constants::SHIP_ENGINES_ACCELERATION * 2.0, 0.0, 0.0);
@@ -239,13 +239,18 @@ mod tests {
#[test]
fn test_tractorbeam_push_range() {
let (mut ship, mut masses) = setup();
- setup_ship_target(&mut ship, &mut masses);
+ setup_ship_target(&mut ship, String::from("astroid"), &mut masses);
let mut astroid = masses.remove("astroid").unwrap();
astroid.position = Vector::new(1.0, 0.0, 0.0);
masses.insert(String::from("astroid"), astroid);
- ship.give_received_data(ModuleType::Tractorbeam, String::from("p"));
+ let recv = serde_json::to_string(&tractorbeam::ServerRecvData {
+ key: String::from("p"),
+ desired_distance: None,
+ })
+ .unwrap();
+ ship.give_received_data(ModuleType::Tractorbeam, recv);
ship.process(&mut masses);
let data = ship.get_client_data(ModuleType::Tractorbeam, &masses);
let tractorbeam_data: tractorbeam::ClientData = serde_json::from_str(&data).unwrap();
@@ -264,13 +269,18 @@ mod tests {
#[test]
fn test_tractorbeam_pull_range() {
let (mut ship, mut masses) = setup();
- setup_ship_target(&mut ship, &mut masses);
+ setup_ship_target(&mut ship, String::from("astroid"), &mut masses);
let mut astroid = masses.remove("astroid").unwrap();
astroid.position = Vector::new(1.0, 0.0, 0.0);
masses.insert(String::from("astroid"), astroid);
- ship.give_received_data(ModuleType::Tractorbeam, String::from("o"));
+ let recv = serde_json::to_string(&tractorbeam::ServerRecvData {
+ key: String::from("o"),
+ desired_distance: None,
+ })
+ .unwrap();
+ ship.give_received_data(ModuleType::Tractorbeam, recv);
ship.process(&mut masses);
let data = ship.get_client_data(ModuleType::Tractorbeam, &masses);
let tractorbeam_data: tractorbeam::ClientData = serde_json::from_str(&data).unwrap();
@@ -289,13 +299,18 @@ mod tests {
#[test]
fn test_tractorbeam_bring_range() {
let (mut ship, mut masses) = setup();
- setup_ship_target(&mut ship, &mut masses);
+ setup_ship_target(&mut ship, String::from("astroid"), &mut masses);
let mut astroid = masses.remove("astroid").unwrap();
astroid.position = Vector::new(1.0, 0.0, 0.0);
masses.insert(String::from("astroid"), astroid);
- ship.give_received_data(ModuleType::Tractorbeam, String::from("b"));
+ let recv = serde_json::to_string(&tractorbeam::ServerRecvData {
+ key: String::from("b"),
+ desired_distance: Some(5.0),
+ })
+ .unwrap();
+ ship.give_received_data(ModuleType::Tractorbeam, recv);
ship.process(&mut masses);
let data = ship.get_client_data(ModuleType::Tractorbeam, &masses);
let tractorbeam_data: tractorbeam::ClientData = serde_json::from_str(&data).unwrap();
@@ -314,7 +329,7 @@ mod tests {
#[test]
fn test_tractorbeam() {
let (mut ship, mut masses) = setup();
- setup_ship_target(&mut ship, &mut masses);
+ setup_ship_target(&mut ship, String::from("astroid"), &mut masses);
let mut astroid = masses.remove("astroid").unwrap();
let start = 2.0;
@@ -323,7 +338,12 @@ mod tests {
assert!(astroid.position == Vector::new(start, 0.0, 0.0));
masses.insert(String::from("astroid"), astroid);
- ship.give_received_data(ModuleType::Tractorbeam, String::from("o"));
+ let recv = serde_json::to_string(&tractorbeam::ServerRecvData {
+ key: String::from("o"),
+ desired_distance: None,
+ })
+ .unwrap();
+ ship.give_received_data(ModuleType::Tractorbeam, recv);
let mut iterated = 1.0;
loop {
ship.process(&mut masses);
@@ -343,7 +363,12 @@ mod tests {
}
}
- ship.give_received_data(ModuleType::Tractorbeam, String::from("p"));
+ let recv = serde_json::to_string(&tractorbeam::ServerRecvData {
+ key: String::from("p"),
+ desired_distance: None,
+ })
+ .unwrap();
+ ship.give_received_data(ModuleType::Tractorbeam, recv);
let mut iterated = 1.0;
loop {
ship.process(&mut masses);
@@ -367,15 +392,21 @@ mod tests {
#[test]
fn test_tractorbeam_bring() {
let (mut ship, mut masses) = setup();
- setup_ship_target(&mut ship, &mut masses);
+ setup_ship_target(&mut ship, String::from("astroid"), &mut masses);
let mut astroid = masses.remove("astroid").unwrap();
let start = 25.0;
+ let desired_distance = 5.0;
astroid.position = Vector::new(start, 0.0, 0.0);
astroid.process(&mut masses);
masses.insert(String::from("astroid"), astroid);
- ship.give_received_data(ModuleType::Tractorbeam, String::from("b"));
+ let recv = serde_json::to_string(&tractorbeam::ServerRecvData {
+ key: String::from("b"),
+ desired_distance: Some(desired_distance),
+ })
+ .unwrap();
+ ship.give_received_data(ModuleType::Tractorbeam, recv);
let mut iterated = 1.0;
loop {
ship.process(&mut masses);
@@ -383,8 +414,7 @@ mod tests {
let mut astroid = masses.remove("astroid").unwrap();
astroid.process(&mut masses);
- if ship.position.distance_from(astroid.position.clone())
- < constants::SHIP_TRACTORBEAM_BRING_TO_DISTANCE
+ if ship.position.distance_from(astroid.position.clone()) < desired_distance
&& astroid.velocity.magnitude() < 1.0
{
break;
@@ -411,4 +441,80 @@ mod tests {
std::process::Command::new("feh").arg("--conversion-timeout").arg("1").arg("line.svg").output().expect("problem");
*/
}
+
+ #[test]
+ fn test_tractorbeam_acquire() {
+ let (mut ship, mut masses) = setup();
+ masses.insert(
+ String::from("iron"),
+ Mass::new_item(
+ Item::new(ItemType::Iron),
+ Vector::default(),
+ Vector::default(),
+ ),
+ );
+ setup_ship_target(&mut ship, String::from("iron"), &mut masses);
+
+ let recv = serde_json::to_string(&tractorbeam::ServerRecvData {
+ key: String::from("a"),
+ desired_distance: None,
+ })
+ .unwrap();
+ ship.give_received_data(ModuleType::Tractorbeam, recv);
+
+ assert!(masses.len() == 2);
+ assert!(ship.item_count(ItemType::Iron) == 0);
+ ship.process(&mut masses);
+ assert!(ship.item_count(ItemType::Iron) == 1);
+ assert!(masses.len() == 1);
+ }
+
+ #[test]
+ fn test_tractorbeam_acquire_range() {
+ let (mut ship, mut masses) = setup();
+ masses.insert(
+ String::from("iron"),
+ Mass::new_item(
+ Item::new(ItemType::Iron),
+ Vector::new(50.0, 0.0, 0.0),
+ Vector::default(),
+ ),
+ );
+ setup_ship_target(&mut ship, String::from("iron"), &mut masses);
+
+ let recv = serde_json::to_string(&tractorbeam::ServerRecvData {
+ key: String::from("a"),
+ desired_distance: None,
+ })
+ .unwrap();
+ ship.give_received_data(ModuleType::Tractorbeam, recv);
+
+ assert!(masses.len() == 2);
+ assert!(ship.item_count(ItemType::Iron) == 0);
+ ship.process(&mut masses);
+ assert!(ship.item_count(ItemType::Iron) == 0);
+ assert!(masses.len() == 2);
+
+ let mut iterated = 1.0;
+ loop {
+ ship.process(&mut masses);
+
+ match masses.remove("iron") {
+ Some(mut item) => {
+ item.process(&mut masses);
+ masses.insert(String::from("iron"), item);
+ }
+ None => {
+ assert!(ship.item_count(ItemType::Iron) == 1);
+ break;
+ }
+ }
+
+ iterated += 1.0;
+ if iterated > 100.0 {
+ assert!(false);
+ break;
+ }
+ }
+ }
}