How are you writing tests for this?