Skip to content

WorkspaceBookingContract #636

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

Merged
yusuftomilola merged 2 commits into from
Mar 9, 2026
Merged

WorkspaceBookingContract #636

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
806fdcf
add unit tests for WorkspaceBookingContract functionality
nottherealalanturing Mar 7, 2026
24cb3ad
Merge branch 'main' into ft/test
yusuftomilola Mar 9, 2026
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
61 changes: 34 additions & 27 deletions contracts/workspace_booking/src/test.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
// contracts/workspace_booking/src/test.rs
#![cfg(test)]

use super::*;
Expand All @@ -7,11 +8,15 @@ use soroban_sdk::{
Address, Env, String,
};

// ── Helpers ───────────────────────────────────────────────────────────────────

/// Register the workspace_booking contract and return its address.
fn setup_contract(env: &Env) -> Address {
env.register(WorkspaceBookingContract, ())
}

/// Register a mock token (Stellar Asset Contract), mint `amount` to `recipient`,
/// and return the token address.
fn setup_token(env: &Env, admin: &Address, recipient: &Address, amount: i128) -> Address {
let token_address = env.register_stellar_asset_contract_v2(admin.clone()).address();
StellarAssetClient::new(env, &token_address)
Expand All @@ -20,10 +25,12 @@ fn setup_token(env: &Env, admin: &Address, recipient: &Address, amount: i128) ->
token_address
}

/// Advance the ledger timestamp by `seconds`.
fn advance_time(env: &Env, seconds: u64) {
env.ledger().with_mut(|l| l.timestamp += seconds);
}

// ── Tests ─────────────────────────────────────────────────────────────────────

#[test]
fn test_initialize_success() {
Expand Down Expand Up @@ -53,7 +60,7 @@ fn test_initialize_twice_fails() {

env.mock_all_auths();
client.initialize(&admin, &token);
client.initialize(&admin, &token);
client.initialize(&admin, &token); // AlreadyInitialized = 3
}

#[test]
Expand All @@ -74,7 +81,7 @@ fn test_register_workspace_success() {
&String::from_str(&env, "Hot Desk A"),
&WorkspaceType::HotDesk,
&1u32,
&500i128,
&500i128, // 500 units per hour
);

let ws = client.get_workspace(&String::from_str(&env, "ws-001"));
Expand Down Expand Up @@ -109,7 +116,7 @@ fn test_register_workspace_duplicate_id_fails() {
&1u32,
&500i128,
);

// WorkspaceAlreadyExists = 5
client.register_workspace(
&admin,
&String::from_str(&env, "ws-001"),
Expand All @@ -133,7 +140,7 @@ fn test_register_workspace_non_admin_fails() {

env.mock_all_auths();
client.initialize(&admin, &token);

// Unauthorized = 2
client.register_workspace(
&non_admin,
&String::from_str(&env, "ws-001"),
Expand Down Expand Up @@ -163,13 +170,13 @@ fn test_book_workspace_success() {
&String::from_str(&env, "Meeting Room Alpha"),
&WorkspaceType::MeetingRoom,
&10u32,
&1_000i128,
&1_000i128, // 1000 units/hr
);


// Book for 2 hours starting 60 seconds from now
let now = env.ledger().timestamp();
let start = now + 60;
let end = start + 7_200;
let end = start + 7_200; // 2 hours

client.book_workspace(
&member,
Expand All @@ -183,9 +190,9 @@ fn test_book_workspace_success() {
assert_eq!(booking.workspace_id, String::from_str(&env, "ws-001"));
assert_eq!(booking.member, member);
assert_eq!(booking.status, BookingStatus::Active);
assert_eq!(booking.amount_paid, 2_000i128);

assert_eq!(booking.amount_paid, 2_000i128); // 2 hrs × 1000

// Member balance should have decreased
let balance = TokenClient::new(&env, &token_address).balance(&member);
assert_eq!(balance, 10_000 - 2_000);
}
Expand Down Expand Up @@ -225,12 +232,12 @@ fn test_book_workspace_conflict_fails() {
&end,
);


// Second booking overlaps — BookingConflict = 7
client.book_workspace(
&member,
&String::from_str(&env, "booking-002"),
&String::from_str(&env, "ws-001"),
&(start + 1_800),
&(start + 1_800), // starts in the middle of first booking
&(end + 1_800),
);
}
Expand Down Expand Up @@ -259,7 +266,7 @@ fn test_cancel_booking_refunds_member() {

let now = env.ledger().timestamp();
let start = now + 60;
let end = start + 3_600;
let end = start + 3_600; // 1 hour → cost 2000

client.book_workspace(
&member,
Expand All @@ -275,7 +282,7 @@ fn test_cancel_booking_refunds_member() {
client.cancel_booking(&member, &String::from_str(&env, "booking-001"));

let balance_after_cancel = TokenClient::new(&env, &token_address).balance(&member);
assert_eq!(balance_after_cancel, 10_000i128);
assert_eq!(balance_after_cancel, 10_000i128); // full refund

let booking = client.get_booking(&String::from_str(&env, "booking-001"));
assert_eq!(booking.status, BookingStatus::Cancelled);
Expand Down Expand Up @@ -315,7 +322,7 @@ fn test_complete_booking_by_admin() {
&end,
);


// Advance time past end
advance_time(&env, 4_000);

client.complete_booking(&admin, &String::from_str(&env, "booking-001"));
Expand Down Expand Up @@ -360,7 +367,7 @@ fn test_cancel_already_cancelled_fails() {
);

client.cancel_booking(&member, &String::from_str(&env, "booking-001"));

// BookingNotActive = 10
client.cancel_booking(&member, &String::from_str(&env, "booking-001"));
}

Expand Down Expand Up @@ -390,10 +397,10 @@ fn test_check_availability_no_conflict() {
let start = now + 3_600;
let end = start + 3_600;


// No bookings yet — should be available
assert!(client.check_availability(&String::from_str(&env, "ws-001"), &start, &end));


// Book it
client.book_workspace(
&member,
&String::from_str(&env, "booking-001"),
Expand All @@ -402,10 +409,10 @@ fn test_check_availability_no_conflict() {
&end,
);


// Same slot should no longer be available
assert!(!client.check_availability(&String::from_str(&env, "ws-001"), &start, &end));


// Non-overlapping slot after the booking should still be available
assert!(client.check_availability(
&String::from_str(&env, "ws-001"),
&end,
Expand Down Expand Up @@ -435,7 +442,7 @@ fn test_set_workspace_availability_blocks_new_bookings() {
&400i128,
);


// Disable the workspace
client.set_workspace_availability(&admin, &String::from_str(&env, "ws-001"), &false);

assert!(!client.check_availability(
Expand Down Expand Up @@ -481,7 +488,7 @@ fn test_multiple_workspaces_independent_availability() {
let start = now + 60;
let end = start + 3_600;


// Book ws-001 for the slot
client.book_workspace(
&member,
&String::from_str(&env, "booking-001"),
Expand All @@ -490,10 +497,10 @@ fn test_multiple_workspaces_independent_availability() {
&end,
);


// ws-002 and ws-003 should still be available for the same slot
assert!(client.check_availability(&String::from_str(&env, "ws-002"), &start, &end));
assert!(client.check_availability(&String::from_str(&env, "ws-003"), &start, &end));

// ws-001 should NOT be available
assert!(!client.check_availability(&String::from_str(&env, "ws-001"), &start, &end));
}

Expand Down Expand Up @@ -521,7 +528,7 @@ fn test_member_and_workspace_booking_indexes() {

let now = env.ledger().timestamp();


// Make two non-overlapping bookings for the same member and workspace
let s1 = now + 100;
let e1 = s1 + 3_600;
let s2 = e1 + 600;
Expand Down Expand Up @@ -571,15 +578,15 @@ fn test_hourly_rate_update_applies_to_future_bookings() {
&1_000i128,
);


// Update rate to 2000
client.set_workspace_rate(&admin, &String::from_str(&env, "ws-001"), &2_000i128);

let ws = client.get_workspace(&String::from_str(&env, "ws-001"));
assert_eq!(ws.hourly_rate, 2_000i128);

let now = env.ledger().timestamp();
let start = now + 60;
let end = start + 3_600;
let end = start + 3_600; // 1 hour

client.book_workspace(
&member,
Expand All @@ -590,5 +597,5 @@ fn test_hourly_rate_update_applies_to_future_bookings() {
);

let booking = client.get_booking(&String::from_str(&env, "bk-001"));
assert_eq!(booking.amount_paid, 2_000i128);
assert_eq!(booking.amount_paid, 2_000i128); // new rate applied
}
Loading