gupta/memory_test.go

package gupta

import (
	"bytes"
	"encoding/json"
	"strings"
	"testing"
)

func TestFreeMem(t *testing.T) {
	m := Memory{}
	m.Free = uint64(1200)
	got := m.Free
	assertInt64(t, "free", got, uint64(1200))
}

func TestTotalMem(t *testing.T) {
	m := Memory{Total: uint64(2400)}
	got := m.Total
	assertInt64(t, "total", got, uint64(2400))
}

func TestUsedMem(t *testing.T) {
	m := Memory{Total: uint64(2400), Free: uint64(1200)}
	got := m.Used()
	assertInt64(t, "used", got, uint64(1200))
}

func TestNewMemory(t *testing.T) {
	//r, _ := os.open("/proc/meminfo")
	r := strings.NewReader("MemTotal:       16080192 kB\nMemFree:         3617004 kB\nMemAvailable:   11258504 kB\nBuffers:         1814108 kB\n")
	mem := NewMemory(r)
	t.Run("Test that we can get total memory", func(t *testing.T) {
		got := mem.Total
		assertInt64(t, "memory total", got, uint64(16080192))
	})

	t.Run("Test that we can properly report Memory usage", func(t *testing.T) {
		var buff bytes.Buffer
		enc := json.NewEncoder(&buff)
		enc.Encode(&mem)
		report := buff.String()
		reportFields := strings.Split(report, ":")
		if reportFields[2] != "\"16080192KB\",\"used\"" {
			t.Errorf("Encoding report failed")
		}
	})
}