tests and fix for cpu pc

Author: simonmittag

cpugrep.go

@@ -11,7 +11,7 @@ const cpugrepcmd = "ps -m -o %%cpu,pid | grep %d | head -1"
 func Cpugrep(pid int) string {
 	cmd := fmt.Sprintf(cpugrepcmd, pid)
 	stdout, _, _ := Shellout(cmd)
-	cpu := strings.Split(stdout, " ")[0]
+	cpu := strings.Split(strings.TrimSpace(stdout), " ")[0]
 	r, _ := strconv.ParseFloat(cpu, 8)
 	return fmt.Sprintf("%.2f%%", r)
 }

cpugrep_test.go

@@ -0,0 +1,15 @@
+package l0la
+
+import (
+	"os"
+	"strings"
+	"testing"
+)
+
+func TestCpuGrep(t *testing.T) {
+	cpu := Cpugrep(os.Getpid())
+	t.Logf("cpu percent was %v", cpu)
+	if !strings.HasSuffix(cpu, "%") {
+		t.Errorf("invalid cpu pc, got: %v", cpu)
+	}
+}