Started to try robolectric but it's not working yet