Swaps budget updates for budget drips

4 files changed, 25 insertions, 20 deletions

diff --git a/rust/schist_fakes/src/budget_drip.rs b/rust/schist_fakes/src/budget_drip.rs
new file mode 100644
index 0000000..f653565
--- /dev/null
+++ b/rust/schist_fakes/src/budget_drip.rs
@@ -0,0 +1,16 @@
+use schist_models::{budget_drip::{BudgetDrip, BudgetDripBuilder}, date_utc::DateUtc};
+use schist_traits::nowlike::Nowlike;
+
+pub fn make_fake_budget_drip(id: i32) -> BudgetDrip {
+    make_fake_budget_drip_builder(id).build().unwrap()
+}
+
+pub fn make_fake_budget_drip_builder(id: i32) -> BudgetDripBuilder {
+    let mut builder = BudgetDripBuilder::default();
+    builder
+        .id(id)
+        .category_id(0)
+        .date(DateUtc::now())
+        .quantity(0);
+    builder
+}
diff --git a/rust/schist_fakes/src/budget_update.rs b/rust/schist_fakes/src/budget_update.rs
deleted file mode 100644
index 2c8ba34..0000000
--- a/rust/schist_fakes/src/budget_update.rs
+++ /dev/null
@@ -1,17 +0,0 @@
-use schist_models::{budget_update::{BudgetUpdate, BudgetUpdateBuilder}, date_utc::DateUtc};
-use schist_traits::nowlike::Nowlike;
-
-pub fn make_fake_budget_update(id: i32) -> BudgetUpdate {
-    make_fake_budget_update_builder(id).build().unwrap()
-}
-
-pub fn make_fake_budget_update_builder(id: i32) ->  BudgetUpdateBuilder {
-    let mut builder = BudgetUpdateBuilder::default();
-    builder
-        .id(id)
-        .category_id(0)
-        .date(DateUtc::now())
-        .new_budget(0)
-        .new_period(0);
-    builder
-}
diff --git a/rust/schist_fakes/src/category.rs b/rust/schist_fakes/src/category.rs
index 39fddc7..c64818d 100644
--- a/rust/schist_fakes/src/category.rs
+++ b/rust/schist_fakes/src/category.rs
@@ -1,4 +1,5 @@
-use schist_models::category::{Category, CategoryBuilder};
+use schist_models::{budget_period_unit::BudgetPeriodUnit, category::{Category, CategoryBuilder}, date_utc::DateUtc};
+use schist_traits::nowlike::Nowlike;
 
 pub fn make_fake_category(id: i32) -> Category {
     make_fake_category_builder(id).build().unwrap()
@@ -8,6 +9,11 @@ pub fn make_fake_category_builder(id: i32) ->  CategoryBuilder {
     let mut builder = CategoryBuilder::default();
     builder
         .id(id)
-        .name(String::from("Fake category"));
+        .name(String::from("Fake category"))
+        .balance(0)
+        .balance_date(DateUtc::now())
+        .budget_period(0)
+        .budget_period_unit(BudgetPeriodUnit::Day)
+        .budget_quantity(0);
     builder
 }
diff --git a/rust/schist_fakes/src/lib.rs b/rust/schist_fakes/src/lib.rs
index 68f5209..2d865d7 100644
--- a/rust/schist_fakes/src/lib.rs
+++ b/rust/schist_fakes/src/lib.rs
@@ -1,5 +1,5 @@
 pub mod account;
-pub mod budget_update;
+pub mod budget_drip;
 pub mod category;
 pub mod category_transfer;
 pub mod transaction;