Finish adding unit tests for local integration