task-073

Co-authored-by: Joe Carstairs <jcarstairs@scottlogic.com>
Reviewed-on: https://git.joeac.net/joeac/schist/pulls/1
Co-authored-by: Joe Carstairs <me@joeac.net>
Co-committed-by: Joe Carstairs <me@joeac.net>

4 files changed, 29 insertions, 24 deletions

diff --git a/schist_core/schist_fakes/src/bucket.rs b/schist_core/schist_fakes/src/bucket.rs
new file mode 100644
index 0000000..1b447b2
--- /dev/null
+++ b/schist_core/schist_fakes/src/bucket.rs
@@ -0,0 +1,24 @@
+use schist_models::{
+    bucket::{Bucket, BucketBuilder},
+    budget_period_unit::BudgetPeriodUnit,
+    date_utc::DateUtc,
+};
+use schist_traits::nowlike::Nowlike;
+
+pub fn make_fake_bucket(id: i32) -> Bucket {
+    make_fake_bucket_builder(id).build().unwrap()
+}
+
+pub fn make_fake_bucket_builder(id: i32) -> BucketBuilder {
+    let mut builder = BucketBuilder::default();
+    builder
+        .id(id)
+        .name(String::from("Fake bucket"))
+        .balance(0)
+        .balance_date(DateUtc::now())
+        .budget_period(0)
+        .budget_period_unit(BudgetPeriodUnit::Day)
+        .budget_quantity(0)
+        .group(String::from("Fake group"));
+    builder
+}
diff --git a/schist_core/schist_fakes/src/category.rs b/schist_core/schist_fakes/src/category.rs
deleted file mode 100644
index c64818d..0000000
--- a/schist_core/schist_fakes/src/category.rs
+++ /dev/null
@@ -1,19 +0,0 @@
-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()
-}
-
-pub fn make_fake_category_builder(id: i32) ->  CategoryBuilder {
-    let mut builder = CategoryBuilder::default();
-    builder
-        .id(id)
-        .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/schist_core/schist_fakes/src/category_transfer.rs b/schist_core/schist_fakes/src/category_transfer.rs
index c28dffb..a16d238 100644
--- a/schist_core/schist_fakes/src/category_transfer.rs
+++ b/schist_core/schist_fakes/src/category_transfer.rs
@@ -1,11 +1,11 @@
-use schist_models::category_transfer::{CategoryTransfer, CategoryTransferBuilder};
+use schist_models::bucket_transfer::{BucketTransfer, BucketTransferBuilder};
 
-pub fn make_fake_category_transfer(id: i32) -> CategoryTransfer {
+pub fn make_fake_category_transfer(id: i32) -> BucketTransfer {
     make_fake_category_transfer_builder(id).build().unwrap()
 }
 
-pub fn make_fake_category_transfer_builder(id: i32) ->  CategoryTransferBuilder {
-    let mut builder = CategoryTransferBuilder::default();
+pub fn make_fake_category_transfer_builder(id: i32) -> BucketTransferBuilder {
+    let mut builder = BucketTransferBuilder::default();
     builder
         .id(id)
         .description(String::from("Fake category transfer"))
diff --git a/schist_core/schist_fakes/src/lib.rs b/schist_core/schist_fakes/src/lib.rs
index 6e20c31..4a22169 100644
--- a/schist_core/schist_fakes/src/lib.rs
+++ b/schist_core/schist_fakes/src/lib.rs
@@ -1,7 +1,7 @@
 pub mod account;
 pub mod account_transfer;
+pub mod bucket;
 pub mod budget_drip;
-pub mod category;
 pub mod category_transfer;
 pub mod transaction;
 pub mod transaction_categorisation;