diff --git a/pkg/build/build_test.go b/pkg/build/build_test.go
index d6c0aa5ff..80ff8662b 100644
--- a/pkg/build/build_test.go
+++ b/pkg/build/build_test.go
@@ -562,6 +562,11 @@ func TestWriteBuildScript(t *testing.T) {
 	f.WriteEnv("DRONE_COMMIT", "e7e046b35")
 	f.WriteEnv("DRONE_PR", "123")
 	f.WriteEnv("DRONE_BUILD_DIR", "/var/cache/drone/github.com/drone/drone")
+	f.WriteEnv("CI_NAME", "DRONE")
+	f.WriteEnv("CI_BUILD_NUMBER", "e7e046b35")
+	f.WriteEnv("CI_BUILD_URL", "")
+	f.WriteEnv("CI_BRANCH", "master")
+	f.WriteEnv("CI_PULL_REQUEST", "123")
 	f.WriteHost("127.0.0.1")
 	f.WriteCmd("git clone --depth=0 --recursive --branch=master git://github.com/drone/drone.git /var/cache/drone/github.com/drone/drone")
 	f.WriteCmd("git fetch origin +refs/pull/123/head:refs/remotes/origin/pr/123")