Add unit testing infrastructure