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>

1 files changed, 24 insertions, 0 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
+}