SebastianCzoch/lxc-exporter

package lxc

import (
    "testing"

    "github.com/stretchr/testify/assert"
)

func TestGetProcStatNotExistedContainer(t *testing.T) {
    l := &LXC{
        kernelVersion:  3,
        containersPath: "../test/sys/fs/cgroup/lxc",
    }

    _, err := l.GetProcStat("not-existed")
    assert.Error(t, err)
}

func TestGetCPUStatPathFail(t *testing.T) {
    l := &LXC{
        kernelVersion:  2,
        containersPath: "../test/sys/fs/cgroup/lxc",
    }

    _, err := l.getCPUStatPath("test")
    assert.Error(t, err)
}

func TestGetCPUStatPath(t *testing.T) {
    l := &LXC{
        kernelVersion:  3,
        containersPath: "../test/sys/fs/cgroup/lxc",
    }

    res, err := l.getCPUStatPath("test")
    assert.NoError(t, err)
    assert.Equal(t, "/sys/fs/cgroup/lxc/test/cpuacct.stat", res)
}

func TestFetchProcStatFail(t *testing.T) {
    l := &LXC{
        kernelVersion:  3,
        containersPath: "../test/sys/fs/cgroup/lxc",
    }

    _, err := l.fetchProcStat("not-existed")
    assert.Error(t, err)
}

func TestFetchProcStat(t *testing.T) {
    l := &LXC{
        kernelVersion:  3,
        containersPath: "../test/sys/fs/cgroup/lxc",
    }

    cgroupPath = "../test/sys/fs/cgroup"
    res, err := l.fetchProcStat("test-container")
    assert.NoError(t, err)
    assert.Equal(t, "user 3092210213\nsystem 311472536\n", string(res))
}

func TestParseProcStat(t *testing.T) {
    res := parseProcStat([]byte("user 3092210213\nsystem 311472536\n"))
    assert.Equal(t, float64(3092210213), res.User)
    assert.Equal(t, float64(311472536), res.System)
}

func TestGetProcStat(t *testing.T) {
    l := &LXC{
        kernelVersion:  3,
        containersPath: "../test/sys/fs/cgroup/lxc",
    }

    cgroupPath = "../test/sys/fs/cgroup"
    res, err := l.GetProcStat("test-container")
    assert.NoError(t, err)
    assert.Equal(t, float64(3092210213), res.User)
    assert.Equal(t, float64(311472536), res.System)
}