pidfile: add the unit testing

Signed-off-by: Tonghao Zhang <tonghao@bamaicloud.com>

Author: hao022

go.mod

@@ -27,6 +27,7 @@ require (
 	github.com/safchain/ethtool v0.6.2
 	github.com/shirou/gopsutil v2.21.11+incompatible
 	github.com/sirupsen/logrus v1.9.3
+	github.com/stretchr/testify v1.10.0
 	github.com/tklauser/numcpus v0.6.1
 	github.com/urfave/cli/v2 v2.27.4
 	github.com/vishvananda/netlink v1.3.1
@@ -58,6 +59,7 @@ require (
 	github.com/coreos/go-semver v0.3.0 // indirect
 	github.com/coreos/go-systemd/v22 v22.5.0 // indirect
 	github.com/cpuguy83/go-md2man/v2 v2.0.4 // indirect
+	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect
 	github.com/dennwc/varint v1.0.0 // indirect
 	github.com/distribution/reference v0.6.0 // indirect
 	github.com/docker/go-connections v0.5.0 // indirect
@@ -129,6 +131,7 @@ require (
 	github.com/pelletier/go-toml/v2 v2.2.3 // indirect
 	github.com/pierrec/lz4/v4 v4.1.21 // indirect
 	github.com/planetscale/vtprotobuf v0.6.1-0.20240319094008-0393e58bdf10 // indirect
+	github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect
 	github.com/prometheus/client_model v0.6.1 // indirect
 	github.com/prometheus/common v0.62.0 // indirect
 	github.com/prometheus/prometheus v0.302.1 // indirect

internal/pidfile/pidfile_test.go

@@ -0,0 +1,140 @@
+// Copyright 2026 The HuaTuo Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package pidfile
+
+import (
+	"os"
+	"path/filepath"
+	"strconv"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestPath(t *testing.T) {
+	tests := []struct {
+		name     string
+		expected string
+	}{
+		{"app", "/var/run/app.pid"},
+		{"nginx", "/var/run/nginx.pid"},
+		{"", "/var/run/.pid"},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			assert.Equal(t, tt.expected, path(tt.name))
+		})
+	}
+}
+
+func TestLock_Success(t *testing.T) {
+	tmpDir := t.TempDir()
+	oldDefault := defaultDirPath
+	defaultDirPath = tmpDir
+	t.Cleanup(func() { defaultDirPath = oldDefault })
+
+	name := "testapp"
+	pidPath := path(name)
+
+	err := Lock(name)
+	require.NoError(t, err)
+	t.Cleanup(func() { UnLock(name) })
+
+	_, err = os.Stat(pidPath)
+	require.NoError(t, err)
+
+	data, err := os.ReadFile(pidPath)
+	require.NoError(t, err)
+	pid, err := strconv.Atoi(string(data))
+	require.NoError(t, err)
+	assert.Equal(t, os.Getpid(), pid)
+}
+
+func TestLock_AlreadyLocked(t *testing.T) {
+	tmpDir := t.TempDir()
+	oldDefault := defaultDirPath
+	defaultDirPath = tmpDir
+	t.Cleanup(func() { defaultDirPath = oldDefault })
+
+	name := "test-locked"
+
+	err := Lock(name)
+	require.NoError(t, err)
+	t.Cleanup(func() { UnLock(name) })
+
+	err = Lock(name)
+	assert.Error(t, err)
+	assert.Contains(t, err.Error(), "running")
+}
+
+func TestUnLock(t *testing.T) {
+	tmpDir := t.TempDir()
+	oldDefault := defaultDirPath
+	defaultDirPath = tmpDir
+	t.Cleanup(func() { defaultDirPath = oldDefault })
+
+	name := "toremove"
+	pidPath := path(name)
+
+	err := os.WriteFile(pidPath, []byte("12345"), 0o600)
+	require.NoError(t, err)
+
+	UnLock(name)
+
+	_, err = os.Stat(pidPath)
+	assert.True(t, os.IsNotExist(err))
+}
+
+func TestRead(t *testing.T) {
+	tmp := t.TempDir()
+	pidPath := filepath.Join(tmp, "test.pid")
+
+	tests := []struct {
+		name        string
+		content     string
+		wantPID     int
+		wantErrKind bool // 是否期望有错误
+	}{
+		{"normal", "12345", 12345, false},
+		{"with space", "  67890  \n", 67890, false},
+		{"negative", "-1", -1, false},
+		{"empty", "", 0, true},
+		{"invalid", "abc", 0, true},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			err := os.WriteFile(pidPath, []byte(tt.content), 0o600)
+			require.NoError(t, err)
+
+			got, err := Read(pidPath)
+			if tt.wantErrKind {
+				assert.Error(t, err)
+				assert.Zero(t, got)
+			} else {
+				assert.NoError(t, err)
+				assert.Equal(t, tt.wantPID, got)
+			}
+		})
+	}
+}
+
+func TestRead_NotExist(t *testing.T) {
+	_, err := Read("/this/file/does/not/exist.pid")
+	assert.Error(t, err)
+	assert.True(t, os.IsNotExist(err))
+}