diff --git a/contracts/stream_contract/src/test.rs b/contracts/stream_contract/src/test.rs index afb15ca1..e211c085 100644 --- a/contracts/stream_contract/src/test.rs +++ b/contracts/stream_contract/src/test.rs @@ -180,6 +180,20 @@ fn test_update_fee_config_rejects_invalid_fee_rate() { assert_eq!(result, Err(Ok(StreamError::InvalidFeeRate))); } +#[test] +fn test_update_fee_config_rejects_not_initialized() { + let env = Env::default(); + env.mock_all_auths(); + let client = create_contract(&env); + + let admin = Address::generate(&env); + let treasury = Address::generate(&env); + + // Call update_fee_config before initialize + let result = client.try_update_fee_config(&admin, &treasury, &100); + assert_eq!(result, Err(Ok(StreamError::NotInitialized))); +} + #[test] fn test_initialize_emits_event() { let env = Env::default();