Can you write some tests for the use-cases?