Add unit tests for full stack