(tests) Add some stuff to do unit testing