From df45866b5fbb0027c80b8da3e7dfb38c6a3acc00 Mon Sep 17 00:00:00 2001
From: Giulio De Pasquale <giulio@depasquale.eu>
Date: Thu, 25 Feb 2021 20:39:04 +0000
Subject: [PATCH] Added todo

---
 src/main.rs  | 1 +
 src/tests.rs | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/src/main.rs b/src/main.rs
index b2ace4b..a6da5ca 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -21,6 +21,7 @@ mod managers;
 mod models;
 mod strategy;
 mod ticker;
+mod tests;
 
 pub type BoxError = Box<dyn std::error::Error + Send + Sync>;
 
diff --git a/src/tests.rs b/src/tests.rs
index ca5c46d..3c05adb 100644
--- a/src/tests.rs
+++ b/src/tests.rs
@@ -3,6 +3,8 @@ mod common {
     use crate::models::{Position, PositionState, TradingPlatform};
     use crate::models::PositionProfitState::Loss;
 
+    // TODO: generate other helper generator functions like the one below
+
     // Generates two short positions with different profit/loss ratios. Both are position in "Loss".
     pub fn get_short_loss_positions(pair: SymbolPair) -> (Position, Position) {
         let almost_critical = Position::new(pair.clone(),
@@ -69,6 +71,7 @@ mod positions {
         assert_eq!(one.id(), 0);
         assert_eq!(one.leverage(), 0.0);
         assert_eq!(one.price_liq(), 150.0);
+        assert_eq!(one.state(), PositionState::Open);
         assert!(one.price_liq() > one.base_price());
 
         let (two, three) = get_short_loss_positions(pair);