//go:build linux // +build linux /* Copyright 2020 The Kubernetes 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 systemd import ( "context" "fmt" "testing" "time" "github.com/godbus/dbus/v5" "github.com/stretchr/testify/assert" ) type fakeDBusObject struct { properties map[string]interface{} bodyValue interface{} } func (obj *fakeDBusObject) Call(method string, flags dbus.Flags, args ...interface{}) *dbus.Call { return &dbus.Call{Err: nil, Body: []interface{}{obj.bodyValue}} } func (obj *fakeDBusObject) CallWithContext(ctx context.Context, method string, flags dbus.Flags, args ...interface{}) *dbus.Call { return nil } func (obj *fakeDBusObject) Go(method string, flags dbus.Flags, ch chan *dbus.Call, args ...interface{}) *dbus.Call { return nil } func (obj *fakeDBusObject) GoWithContext(ctx context.Context, method string, flags dbus.Flags, ch chan *dbus.Call, args ...interface{}) *dbus.Call { return nil } func (obj *fakeDBusObject) AddMatchSignal(iface, member string, options ...dbus.MatchOption) *dbus.Call { return nil } func (obj *fakeDBusObject) RemoveMatchSignal(iface, member string, options ...dbus.MatchOption) *dbus.Call { return nil } func (obj *fakeDBusObject) GetProperty(p string) (dbus.Variant, error) { value, ok := obj.properties[p] if !ok { return dbus.Variant{}, fmt.Errorf("property %q does not exist in properties: %+v", p, obj.properties) } return dbus.MakeVariant(value), nil } func (obj *fakeDBusObject) SetProperty(p string, v interface{}) error { return nil } func (obj *fakeDBusObject) StoreProperty(p string, v interface{}) error { return nil } func (obj *fakeDBusObject) Destination() string { return "" } func (obj *fakeDBusObject) Path() dbus.ObjectPath { return "" } type fakeSystemDBus struct { fakeDBusObject *fakeDBusObject signalChannel chan<- *dbus.Signal } func (f *fakeSystemDBus) Object(dest string, path dbus.ObjectPath) dbus.BusObject { return f.fakeDBusObject } func (f *fakeSystemDBus) Signal(ch chan<- *dbus.Signal) { f.signalChannel = ch } func (f *fakeSystemDBus) AddMatchSignal(options ...dbus.MatchOption) error { return nil } func TestCurrentInhibitDelay(t *testing.T) { thirtySeconds := time.Duration(30) * time.Second bus := DBusCon{ SystemBus: &fakeSystemDBus{ fakeDBusObject: &fakeDBusObject{ properties: map[string]interface{}{ "org.freedesktop.login1.Manager.InhibitDelayMaxUSec": uint64(thirtySeconds / time.Microsecond), }, }, }, } delay, err := bus.CurrentInhibitDelay() assert.NoError(t, err) assert.Equal(t, thirtySeconds, delay) } func TestInhibitShutdown(t *testing.T) { var fakeFd uint32 = 42 bus := DBusCon{ SystemBus: &fakeSystemDBus{ fakeDBusObject: &fakeDBusObject{ bodyValue: fakeFd, }, }, } fdLock, err := bus.InhibitShutdown() assert.Equal(t, InhibitLock(fakeFd), fdLock) assert.NoError(t, err) } func TestReloadLogindConf(t *testing.T) { bus := DBusCon{ SystemBus: &fakeSystemDBus{ fakeDBusObject: &fakeDBusObject{}, }, } assert.NoError(t, bus.ReloadLogindConf()) } func TestMonitorShutdown(t *testing.T) { var tests = []struct { desc string shutdownActive bool }{ { desc: "shutdown is active", shutdownActive: true, }, { desc: "shutdown is not active", shutdownActive: false, }, } for _, tc := range tests { tc := tc t.Run(tc.desc, func(t *testing.T) { fakeSystemBus := &fakeSystemDBus{} bus := DBusCon{ SystemBus: fakeSystemBus, } outChan, err := bus.MonitorShutdown() assert.NoError(t, err) done := make(chan bool) go func() { select { case res := <-outChan: assert.Equal(t, tc.shutdownActive, res) done <- true case <-time.After(5 * time.Second): t.Errorf("Timed out waiting for shutdown message") done <- true } }() signal := &dbus.Signal{Body: []interface{}{tc.shutdownActive}} fakeSystemBus.signalChannel <- signal <-done }) } }