github.com/chipaca/snappy@v0.0.0-20210104084008-1f06296fe8ad/osutil/testhelper.go (about) 1 // -*- Mode: Go; indent-tabs-mode: t -*- 2 3 /* 4 * Copyright (C) 2020 Canonical Ltd 5 * 6 * This program is free software: you can redistribute it and/or modify 7 * it under the terms of the GNU General Public License version 3 as 8 * published by the Free Software Foundation. 9 * 10 * This program is distributed in the hope that it will be useful, 11 * but WITHOUT ANY WARRANTY; without even the implied warranty of 12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 13 * GNU General Public License for more details. 14 * 15 * You should have received a copy of the GNU General Public License 16 * along with this program. If not, see <http://www.gnu.org/licenses/>. 17 * 18 */ 19 20 package osutil 21 22 import ( 23 "os" 24 "regexp" 25 ) 26 27 var goTestExeRe = regexp.MustCompile(`^.*/.*go-build.*/.*\.test$`) 28 29 // IsTestBinary checks whether the current process is a go test binary. 30 func IsTestBinary() bool { 31 return len(os.Args) > 0 && goTestExeRe.MatchString(os.Args[0]) 32 } 33 34 // MustBeTestBinary checks whether the executing process is a go test binary, 35 // panics otherwise. 36 func MustBeTestBinary(panicMsg string) { 37 if !IsTestBinary() { 38 panic(panicMsg) 39 } 40 }