github.com/1aal/kubeblocks@v0.0.0-20231107070852-e1c03e598921/pkg/configuration/config_manager/proc_util_test.go (about) 1 /* 2 Copyright (C) 2022-2023 ApeCloud Co., Ltd 3 4 This file is part of KubeBlocks project 5 6 This program is free software: you can redistribute it and/or modify 7 it under the terms of the GNU Affero General Public License as published by 8 the Free Software Foundation, either version 3 of the License, or 9 (at your option) any later version. 10 11 This program is distributed in the hope that it will be useful 12 but WITHOUT ANY WARRANTY; without even the implied warranty of 13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 14 GNU Affero General Public License for more details. 15 16 You should have received a copy of the GNU Affero General Public License 17 along with this program. If not, see <http://www.gnu.org/licenses/>. 18 */ 19 20 package configmanager 21 22 import ( 23 "fmt" 24 "os" 25 "testing" 26 27 "github.com/shirou/gopsutil/v3/process" 28 "github.com/stretchr/testify/require" 29 "go.uber.org/zap" 30 ) 31 32 func init() { 33 var zapLog, _ = zap.NewDevelopment() 34 SetLogger(zapLog) 35 } 36 37 func TestFindParentPidByProcessName(t *testing.T) { 38 processName := findCurrProcName() 39 fmt.Printf("current test program name: %s\n", processName) 40 pid, err := findParentPIDByProcessName(processName) 41 require.Nil(t, err) 42 require.Equal(t, PID(os.Getpid()), pid) 43 } 44 45 func findCurrProcName() string { 46 pid := int32(os.Getpid()) 47 procs, _ := process.Processes() 48 for _, proc := range procs { 49 if pid == proc.Pid { 50 name, _ := proc.Name() 51 return name 52 } 53 } 54 return "" 55 }