rust/sat-rs
rust/sat-rs
5
1
Fork 0
Code Issues 9 Pull Requests Projects Releases 22 Wiki Activity

added complete testset

Browse Source
This commit is contained in:
Robin Müller 2024-01-31 23:25:33 +01:00
parent 982b1afcb9
commit 0f639900e2
Signed by: muellerr
GPG Key ID: A649FB78196E3849
1 changed files with 47 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

76
satrs-core/src/pus/test.rs
View File

@ -110,22 +110,35 @@ mod tests {
use crate::pool::{LocalPool, PoolCfg, SharedPool};
use crate::pus::test::PusService17TestHandler;
use crate::pus::verification::{
RequestId, VerificationReporterCfg, VerificationReporterWithSender,
RequestId, VerificationReporterCfg, VerificationReporterWithSender, VerificationToken,
};
use crate::pus::{
EcssTcAndToken, EcssTcInStoreConverter, MpscTcReceiver, MpscTmInStoreSender, MpscTmAsVecSender, EcssTcInVecConverter, TcInMemory,
EcssTcAndToken, EcssTcInStoreConverter, EcssTcInVecConverter, MpscTcReceiver,
MpscTmAsVecSender, MpscTmInStoreSender, TcInMemory,
};
use crate::tmtc::tm_helper::SharedTmStore;
use spacepackets::ecss::tc::{PusTcCreator, PusTcSecondaryHeader};
use spacepackets::ecss::tm::PusTmReader;
use spacepackets::ecss::{PusPacket, WritablePusPacket};
use spacepackets::{SequenceFlags, SpHeader};
use spacepackets::{CcsdsPacket, SequenceFlags, SpHeader};
use std::boxed::Box;
use std::sync::{mpsc, RwLock};
use std::vec;
const TEST_APID: u16 = 0x101;
fn verify_verification_tm<STATE>(
subservice: u8,
tm: &PusTmReader<'_>,
token: VerificationToken<STATE>,
) {
assert_eq!(tm.service(), 1);
assert_eq!(tm.subservice(), subservice);
assert_eq!(tm.apid(), TEST_APID);
let req_id = RequestId::from_bytes(tm.user_data()).expect("generating request ID failed");
assert_eq!(req_id, token.req_id());
}
#[test]
fn test_basic_ping_processing_using_store() {
let mut pus_buf: [u8; 64] = [0; 64];
@ -177,11 +190,7 @@ mod tests {
let mut tm_addr = next_msg.unwrap();
let tm_pool = tm_pool_shared.read().unwrap();
let tm_raw = tm_pool.read(&tm_addr).unwrap();
let (tm, _) = PusTmReader::new(tm_raw, 0).unwrap();
assert_eq!(tm.service(), 1);
assert_eq!(tm.subservice(), 1);
let req_id = RequestId::from_bytes(tm.user_data()).expect("generating request ID failed");
assert_eq!(req_id, token.req_id());
verify_verification_tm(1, &PusTmReader::new(tm_raw, 0).unwrap().0, token);
// Acceptance TM
next_msg = tm_rx.try_recv();
@ -189,11 +198,7 @@ mod tests {
tm_addr = next_msg.unwrap();
let tm_raw = tm_pool.read(&tm_addr).unwrap();
// Is generated with CDS short timestamp.
let (tm, _) = PusTmReader::new(tm_raw, 7).unwrap();
assert_eq!(tm.service(), 1);
assert_eq!(tm.subservice(), 3);
let req_id = RequestId::from_bytes(tm.user_data()).expect("generating request ID failed");
assert_eq!(req_id, token.req_id());
verify_verification_tm(3, &PusTmReader::new(tm_raw, 7).unwrap().0, token);
// Ping reply
next_msg = tm_rx.try_recv();
@ -211,12 +216,7 @@ mod tests {
assert!(next_msg.is_ok());
tm_addr = next_msg.unwrap();
let tm_raw = tm_pool.read(&tm_addr).unwrap();
// Is generated with CDS short timestamp.
let (tm, _) = PusTmReader::new(tm_raw, 7).unwrap();
assert_eq!(tm.service(), 1);
assert_eq!(tm.subservice(), 7);
let req_id = RequestId::from_bytes(tm.user_data()).expect("generating request ID failed");
assert_eq!(req_id, token.req_id());
verify_verification_tm(7, &PusTmReader::new(tm_raw, 7).unwrap().0, token);
}
#[test]
@ -224,8 +224,7 @@ mod tests {
let (test_srv_tc_tx, test_srv_tc_rx) = mpsc::channel();
let (tm_tx, tm_rx) = mpsc::channel();
let verif_sender =
MpscTmAsVecSender::new(0, "verif_sender", tm_tx.clone());
let verif_sender = MpscTmAsVecSender::new(0, "verif_sender", tm_tx.clone());
let verif_cfg = VerificationReporterCfg::new(TEST_APID, 1, 2, 8).unwrap();
let mut verification_handler =
VerificationReporterWithSender::new(&verif_cfg, Box::new(verif_sender));
@ -244,11 +243,15 @@ mod tests {
let mut sp_header = SpHeader::tc(TEST_APID, SequenceFlags::Unsegmented, 0, 0).unwrap();
let sec_header = PusTcSecondaryHeader::new_simple(17, 1);
let ping_tc = PusTcCreator::new_no_app_data(&mut sp_header, sec_header, true);
let token = verification_handler.add_tc(&ping_tc);
verification_handler.acceptance_success(token, None);
let init_token = verification_handler.add_tc(&ping_tc);
let token = verification_handler
.acceptance_success(init_token, None)
.expect("acceptance failed");
let ping_tc_as_vec = ping_tc.to_vec().unwrap();
let tc_in_memory = TcInMemory::from(ping_tc_as_vec);
test_srv_tc_tx.send(EcssTcAndToken::new(tc_in_memory, token)).unwrap();
test_srv_tc_tx
.send(EcssTcAndToken::new(tc_in_memory, token))
.unwrap();
let result = pus_17_handler.handle_one_tc();
if let Err(e) = result.as_ref() {
@ -259,10 +262,25 @@ mod tests {
let mut next_msg = tm_rx.try_recv();
assert!(next_msg.is_ok());
let mut tm_raw = next_msg.unwrap();
let (tm, _) = PusTmReader::new(&tm_raw, 0).unwrap();
assert_eq!(tm.service(), 1);
assert_eq!(tm.subservice(), 1);
let req_id = RequestId::from_bytes(tm.user_data()).expect("generating request ID failed");
assert_eq!(req_id, token.req_id());
verify_verification_tm(1, &PusTmReader::new(&tm_raw, 0).unwrap().0, token);
next_msg = tm_rx.try_recv();
assert!(next_msg.is_ok());
tm_raw = next_msg.unwrap();
verify_verification_tm(3, &PusTmReader::new(&tm_raw, 7).unwrap().0, token);
next_msg = tm_rx.try_recv();
assert!(next_msg.is_ok());
tm_raw = next_msg.unwrap();
// Is generated with CDS short timestamp.
let (tm, _) = PusTmReader::new(&tm_raw, 7).unwrap();
assert_eq!(tm.service(), 17);
assert_eq!(tm.subservice(), 2);
assert!(tm.user_data().is_empty());
next_msg = tm_rx.try_recv();
assert!(next_msg.is_ok());
tm_raw = next_msg.unwrap();
verify_verification_tm(7, &PusTmReader::new(&tm_raw, 7).unwrap().0, token);
}
}