Add tests for build task