dbus: add GetUnitFileState method

Author: vaspahomov

dbus/methods.go

@@ -830,3 +830,18 @@ func (c *Conn) listJobsInternal(ctx context.Context) ([]JobStatus, error) {
 
 	return status, nil
 }
+
+// GetUnitFileStateContext returns UnitFileState
+func (c *Conn) GetUnitFileStateContext(ctx context.Context, name string) (string, error) {
+	var prop dbus.Variant
+	obj := c.sysconn.Object("org.freedesktop.systemd1", "/org/freedesktop/systemd1")
+	err := obj.CallWithContext(ctx, "org.freedesktop.systemd1.Manager.GetUnitFileState", 0, name).Store(&prop)
+	if err != nil {
+		return "", err
+	}
+	state, ok := prop.Value().(string)
+	if !ok {
+		return "", fmt.Errorf("failed to cast UnitFileState prop to string")
+	}
+	return state, nil
+}

dbus/methods_test.go

@@ -15,12 +15,14 @@
 package dbus
 
 import (
+	"context"
 	"fmt"
 	"os"
 	"os/exec"
 	"path"
 	"path/filepath"
 	"reflect"
+	"strings"
 	"syscall"
 	"testing"
 	"time"
@@ -1600,3 +1602,28 @@ func TestUnitName(t *testing.T) {
 		}
 	}
 }
+
+// TestGetUnitFileState reads the `systemd-udevd.service` which should exist on all systemd
+// systems and ensures that UnitFileState property is valid.
+func TestGetUnitFileState(t *testing.T) {
+	conn := setupConn(t)
+	defer conn.Close()
+	service := "systemd-udevd.service"
+	got, err := conn.GetUnitFileStateContext(context.Background(), service)
+	if err != nil {
+		t.Fatal(err)
+	}
+	ok := false
+	// valid UnitFileState values
+	validUnitFileStates := []string{"enabled", "disabled", "static", "bad"}
+	for _, v := range validUnitFileStates {
+		if got == v {
+			ok = true
+			break
+		}
+	}
+	if !ok {
+		t.Errorf("invalid UnitFileState returned from GetUnitFileState(%s): got %s, want one of [%s]",
+			service, got, strings.Join(validUnitFileStates, ", "))
+	}
+}