Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Unit tests #132

Merged
merged 8 commits into from
Apr 12, 2023
Merged

Unit tests #132

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
b9cab0b
add tests for chart module
GyulyVGC Apr 3, 2023
c8f72e7
started tests for gui module
GyulyVGC Apr 3, 2023
38a4d61
[ImgBot] Optimize images
ImgBotApp Apr 7, 2023
85bf7c0
Merge pull request #127 from GyulyVGC/imgbot
GyulyVGC Apr 7, 2023
64a2a15
fix: installation methods (Homebrew not up to date)
GyulyVGC Apr 11, 2023
3d15e31
added some tests on gui messages
GyulyVGC Apr 11, 2023
2e6374a
resolve merge conflicts
GyulyVGC Apr 11, 2023
a81e4cf
added some more tests on gui messages
GyulyVGC Apr 12, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
added some tests on gui messages
  • Loading branch information
@GyulyVGC
GyulyVGC committed Apr 11, 2023
commit 3d15e31454f788efab8dc7a32e7b00de17174abe
2 changes: 1 addition & 1 deletion src/gui/styles/types/style_type.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
use serde::{Deserialize, Serialize};

/// Used to specify the kind of style of the application
#[derive(Clone, Copy, Serialize, Deserialize, Debug, Hash)]
#[derive(Clone, Copy, Serialize, Deserialize, Debug, Hash, PartialEq)]
pub enum StyleType {
Night,
Day,
Expand Down
224 changes: 221 additions & 3 deletions src/gui/types/sniffer.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -443,14 +443,233 @@ impl Sniffer {

#[cfg(test)]
mod tests {
#![allow(unused_must_use)]

use crate::gui::styles::style_constants::get_color_mix_chart;
use crate::gui::styles::types::palette::to_rgb_color;
use crate::gui::types::message::Message;
use crate::{InfoTraffic, Sniffer, Status};
use crate::{
get_colors, AppProtocol, ChartType, InfoTraffic, IpVersion, ReportType, Sniffer, Status,
StyleType, TransProtocol,
};
use std::collections::HashSet;
use std::sync::{Arc, Mutex};

#[test]
fn test_correctly_update_ip_version() {
let mut sniffer = Sniffer::new(
Arc::new(Mutex::new(0)),
Arc::new(Mutex::new(InfoTraffic::new())),
Arc::new((Mutex::new(Status::Init), Default::default())),
&Default::default(),
&Default::default(),
Arc::new(Mutex::new(Err(String::new()))),
);

assert_eq!(sniffer.filters.ip, IpVersion::Other);
sniffer.update(Message::IpVersionSelection(IpVersion::IPv6));
assert_eq!(sniffer.filters.ip, IpVersion::IPv6);
sniffer.update(Message::IpVersionSelection(IpVersion::IPv4));
assert_eq!(sniffer.filters.ip, IpVersion::IPv4);
sniffer.update(Message::IpVersionSelection(IpVersion::IPv4));
assert_eq!(sniffer.filters.ip, IpVersion::IPv4);
sniffer.update(Message::IpVersionSelection(IpVersion::Other));
assert_eq!(sniffer.filters.ip, IpVersion::Other);
}

#[test]
fn test_correctly_update_transport_protocol() {
let mut sniffer = Sniffer::new(
Arc::new(Mutex::new(0)),
Arc::new(Mutex::new(InfoTraffic::new())),
Arc::new((Mutex::new(Status::Init), Default::default())),
&Default::default(),
&Default::default(),
Arc::new(Mutex::new(Err(String::new()))),
);

assert_eq!(sniffer.filters.transport, TransProtocol::Other);
sniffer.update(Message::TransportProtocolSelection(TransProtocol::UDP));
assert_eq!(sniffer.filters.transport, TransProtocol::UDP);
sniffer.update(Message::TransportProtocolSelection(TransProtocol::UDP));
assert_eq!(sniffer.filters.transport, TransProtocol::UDP);
sniffer.update(Message::TransportProtocolSelection(TransProtocol::TCP));
assert_eq!(sniffer.filters.transport, TransProtocol::TCP);
sniffer.update(Message::TransportProtocolSelection(TransProtocol::Other));
assert_eq!(sniffer.filters.transport, TransProtocol::Other);
}

#[test]
fn test_correctly_update_application_protocol() {
let mut sniffer = Sniffer::new(
Arc::new(Mutex::new(0)),
Arc::new(Mutex::new(InfoTraffic::new())),
Arc::new((Mutex::new(Status::Init), Default::default())),
&Default::default(),
&Default::default(),
Arc::new(Mutex::new(Err(String::new()))),
);

assert_eq!(sniffer.filters.application, AppProtocol::Other);
sniffer.update(Message::AppProtocolSelection(AppProtocol::HTTPS));
assert_eq!(sniffer.filters.application, AppProtocol::HTTPS);
sniffer.update(Message::AppProtocolSelection(AppProtocol::HTTP));
assert_eq!(sniffer.filters.application, AppProtocol::HTTP);
sniffer.update(Message::AppProtocolSelection(AppProtocol::HTTP));
assert_eq!(sniffer.filters.application, AppProtocol::HTTP);
sniffer.update(Message::AppProtocolSelection(AppProtocol::XMPP));
assert_eq!(sniffer.filters.application, AppProtocol::XMPP);
}

#[test]
fn test_correctly_update_chart_kind() {
let mut sniffer = Sniffer::new(
Arc::new(Mutex::new(0)),
Arc::new(Mutex::new(InfoTraffic::new())),
Arc::new((Mutex::new(Status::Init), Default::default())),
&Default::default(),
&Default::default(),
Arc::new(Mutex::new(Err(String::new()))),
);

assert_eq!(sniffer.traffic_chart.chart_type, ChartType::Packets);
sniffer.update(Message::ChartSelection(ChartType::Bytes));
assert_eq!(sniffer.traffic_chart.chart_type, ChartType::Bytes);
sniffer.update(Message::ChartSelection(ChartType::Bytes));
assert_eq!(sniffer.traffic_chart.chart_type, ChartType::Bytes);
sniffer.update(Message::ChartSelection(ChartType::Packets));
assert_eq!(sniffer.traffic_chart.chart_type, ChartType::Packets);
}

#[test]
fn test_correctly_update_report_kind() {
let mut sniffer = Sniffer::new(
Arc::new(Mutex::new(0)),
Arc::new(Mutex::new(InfoTraffic::new())),
Arc::new((Mutex::new(Status::Init), Default::default())),
&Default::default(),
&Default::default(),
Arc::new(Mutex::new(Err(String::new()))),
);

assert_eq!(sniffer.report_type, ReportType::MostRecent);
sniffer.update(Message::ReportSelection(ReportType::MostBytes));
assert_eq!(sniffer.report_type, ReportType::MostBytes);
sniffer.update(Message::ReportSelection(ReportType::MostPackets));
assert_eq!(sniffer.report_type, ReportType::MostPackets);
sniffer.update(Message::ReportSelection(ReportType::MostPackets));
assert_eq!(sniffer.report_type, ReportType::MostPackets);
sniffer.update(Message::ReportSelection(ReportType::MostRecent));
assert_eq!(sniffer.report_type, ReportType::MostRecent);
sniffer.update(Message::ReportSelection(ReportType::Favorites));
assert_eq!(sniffer.report_type, ReportType::Favorites);
}

#[test]
fn test_correctly_update_style() {
let mut sniffer = Sniffer::new(
Arc::new(Mutex::new(0)),
Arc::new(Mutex::new(InfoTraffic::new())),
Arc::new((Mutex::new(Status::Init), Default::default())),
&Default::default(),
&Default::default(),
Arc::new(Mutex::new(Err(String::new()))),
);

sniffer.update(Message::Style(StyleType::MonAmour));
assert_eq!(sniffer.style, StyleType::MonAmour);
assert_eq!(
sniffer.traffic_chart.color_font,
to_rgb_color(get_colors(StyleType::MonAmour).text_body)
);
assert_eq!(
sniffer.traffic_chart.color_outgoing,
to_rgb_color(get_colors(StyleType::MonAmour).outgoing)
);
assert_eq!(
sniffer.traffic_chart.color_incoming,
to_rgb_color(get_colors(StyleType::MonAmour).secondary)
);
assert_eq!(
sniffer.traffic_chart.color_mix,
get_color_mix_chart(StyleType::MonAmour)
);
sniffer.update(Message::Style(StyleType::Day));
assert_eq!(sniffer.style, StyleType::Day);
assert_eq!(
sniffer.traffic_chart.color_font,
to_rgb_color(get_colors(StyleType::Day).text_body)
);
assert_eq!(
sniffer.traffic_chart.color_outgoing,
to_rgb_color(get_colors(StyleType::Day).outgoing)
);
assert_eq!(
sniffer.traffic_chart.color_incoming,
to_rgb_color(get_colors(StyleType::Day).secondary)
);
assert_eq!(
sniffer.traffic_chart.color_mix,
get_color_mix_chart(StyleType::Day)
);
sniffer.update(Message::Style(StyleType::Night));
assert_eq!(sniffer.style, StyleType::Night);
assert_eq!(
sniffer.traffic_chart.color_font,
to_rgb_color(get_colors(StyleType::Night).text_body)
);
assert_eq!(
sniffer.traffic_chart.color_outgoing,
to_rgb_color(get_colors(StyleType::Night).outgoing)
);
assert_eq!(
sniffer.traffic_chart.color_incoming,
to_rgb_color(get_colors(StyleType::Night).secondary)
);
assert_eq!(
sniffer.traffic_chart.color_mix,
get_color_mix_chart(StyleType::Night)
);
sniffer.update(Message::Style(StyleType::DeepSea));
assert_eq!(sniffer.style, StyleType::DeepSea);
assert_eq!(
sniffer.traffic_chart.color_font,
to_rgb_color(get_colors(StyleType::DeepSea).text_body)
);
assert_eq!(
sniffer.traffic_chart.color_outgoing,
to_rgb_color(get_colors(StyleType::DeepSea).outgoing)
);
assert_eq!(
sniffer.traffic_chart.color_incoming,
to_rgb_color(get_colors(StyleType::DeepSea).secondary)
);
assert_eq!(
sniffer.traffic_chart.color_mix,
get_color_mix_chart(StyleType::DeepSea)
);
sniffer.update(Message::Style(StyleType::DeepSea));
assert_eq!(sniffer.style, StyleType::DeepSea);
assert_eq!(
sniffer.traffic_chart.color_font,
to_rgb_color(get_colors(StyleType::DeepSea).text_body)
);
assert_eq!(
sniffer.traffic_chart.color_outgoing,
to_rgb_color(get_colors(StyleType::DeepSea).outgoing)
);
assert_eq!(
sniffer.traffic_chart.color_incoming,
to_rgb_color(get_colors(StyleType::DeepSea).secondary)
);
assert_eq!(
sniffer.traffic_chart.color_mix,
get_color_mix_chart(StyleType::DeepSea)
);
}

#[test]
fn test_waiting_dots_update() {
#![allow(unused_must_use)]
let mut sniffer = Sniffer::new(
Arc::new(Mutex::new(0)),
Arc::new(Mutex::new(InfoTraffic::new())),
Expand All @@ -473,7 +692,6 @@ mod tests {

#[test]
fn test_modify_favorite_connections() {
#![allow(unused_must_use)]
let mut sniffer = Sniffer::new(
Arc::new(Mutex::new(0)),
Arc::new(Mutex::new(InfoTraffic::new())),
Expand Down