mirror of
https://github.com/genuinetools/reg.git
synced 2024-09-12 06:11:03 -04:00
247 lines
6 KiB
Go
247 lines
6 KiB
Go
// +build linux freebsd
|
|
|
|
package operatingsystem // import "github.com/docker/docker/pkg/parsers/operatingsystem"
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestGetOperatingSystem(t *testing.T) {
|
|
var backup = etcOsRelease
|
|
|
|
invalids := []struct {
|
|
content string
|
|
errorExpected string
|
|
}{
|
|
{
|
|
`PRETTY_NAME=Source Mage GNU/Linux
|
|
PRETTY_NAME=Ubuntu 14.04.LTS`,
|
|
"PRETTY_NAME needs to be enclosed by quotes if they have spaces: Source Mage GNU/Linux",
|
|
},
|
|
{
|
|
`PRETTY_NAME="Ubuntu Linux
|
|
PRETTY_NAME=Ubuntu 14.04.LTS`,
|
|
"PRETTY_NAME is invalid: invalid command line string",
|
|
},
|
|
{
|
|
`PRETTY_NAME=Ubuntu'
|
|
PRETTY_NAME=Ubuntu 14.04.LTS`,
|
|
"PRETTY_NAME is invalid: invalid command line string",
|
|
},
|
|
{
|
|
`PRETTY_NAME'
|
|
PRETTY_NAME=Ubuntu 14.04.LTS`,
|
|
"PRETTY_NAME needs to be enclosed by quotes if they have spaces: Ubuntu 14.04.LTS",
|
|
},
|
|
}
|
|
|
|
valids := []struct {
|
|
content string
|
|
expected string
|
|
}{
|
|
{
|
|
`NAME="Ubuntu"
|
|
PRETTY_NAME_AGAIN="Ubuntu 14.04.LTS"
|
|
VERSION="14.04, Trusty Tahr"
|
|
ID=ubuntu
|
|
ID_LIKE=debian
|
|
VERSION_ID="14.04"
|
|
HOME_URL="http://www.ubuntu.com/"
|
|
SUPPORT_URL="http://help.ubuntu.com/"
|
|
BUG_REPORT_URL="http://bugs.launchpad.net/ubuntu/"`,
|
|
"Linux",
|
|
},
|
|
{
|
|
`NAME="Ubuntu"
|
|
VERSION="14.04, Trusty Tahr"
|
|
ID=ubuntu
|
|
ID_LIKE=debian
|
|
VERSION_ID="14.04"
|
|
HOME_URL="http://www.ubuntu.com/"
|
|
SUPPORT_URL="http://help.ubuntu.com/"
|
|
BUG_REPORT_URL="http://bugs.launchpad.net/ubuntu/"`,
|
|
"Linux",
|
|
},
|
|
{
|
|
`NAME=Gentoo
|
|
ID=gentoo
|
|
PRETTY_NAME="Gentoo/Linux"
|
|
ANSI_COLOR="1;32"
|
|
HOME_URL="http://www.gentoo.org/"
|
|
SUPPORT_URL="http://www.gentoo.org/main/en/support.xml"
|
|
BUG_REPORT_URL="https://bugs.gentoo.org/"
|
|
`,
|
|
"Gentoo/Linux",
|
|
},
|
|
{
|
|
`NAME="Ubuntu"
|
|
VERSION="14.04, Trusty Tahr"
|
|
ID=ubuntu
|
|
ID_LIKE=debian
|
|
PRETTY_NAME="Ubuntu 14.04 LTS"
|
|
VERSION_ID="14.04"
|
|
HOME_URL="http://www.ubuntu.com/"
|
|
SUPPORT_URL="http://help.ubuntu.com/"
|
|
BUG_REPORT_URL="http://bugs.launchpad.net/ubuntu/"`,
|
|
"Ubuntu 14.04 LTS",
|
|
},
|
|
{
|
|
`NAME="Ubuntu"
|
|
VERSION="14.04, Trusty Tahr"
|
|
ID=ubuntu
|
|
ID_LIKE=debian
|
|
PRETTY_NAME='Ubuntu 14.04 LTS'`,
|
|
"Ubuntu 14.04 LTS",
|
|
},
|
|
{
|
|
`PRETTY_NAME=Source
|
|
NAME="Source Mage"`,
|
|
"Source",
|
|
},
|
|
{
|
|
`PRETTY_NAME=Source
|
|
PRETTY_NAME="Source Mage"`,
|
|
"Source Mage",
|
|
},
|
|
}
|
|
|
|
dir := os.TempDir()
|
|
etcOsRelease = filepath.Join(dir, "etcOsRelease")
|
|
|
|
defer func() {
|
|
os.Remove(etcOsRelease)
|
|
etcOsRelease = backup
|
|
}()
|
|
|
|
for _, elt := range invalids {
|
|
if err := ioutil.WriteFile(etcOsRelease, []byte(elt.content), 0600); err != nil {
|
|
t.Fatalf("failed to write to %s: %v", etcOsRelease, err)
|
|
}
|
|
s, err := GetOperatingSystem()
|
|
if err == nil || err.Error() != elt.errorExpected {
|
|
t.Fatalf("Expected an error %q, got %q (err: %v)", elt.errorExpected, s, err)
|
|
}
|
|
}
|
|
|
|
for _, elt := range valids {
|
|
if err := ioutil.WriteFile(etcOsRelease, []byte(elt.content), 0600); err != nil {
|
|
t.Fatalf("failed to write to %s: %v", etcOsRelease, err)
|
|
}
|
|
s, err := GetOperatingSystem()
|
|
if err != nil || s != elt.expected {
|
|
t.Fatalf("Expected %q, got %q (err: %v)", elt.expected, s, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIsContainerized(t *testing.T) {
|
|
var (
|
|
backup = proc1Cgroup
|
|
nonContainerizedProc1Cgroupsystemd226 = []byte(`9:memory:/init.scope
|
|
8:net_cls,net_prio:/
|
|
7:cpuset:/
|
|
6:freezer:/
|
|
5:devices:/init.scope
|
|
4:blkio:/init.scope
|
|
3:cpu,cpuacct:/init.scope
|
|
2:perf_event:/
|
|
1:name=systemd:/init.scope
|
|
`)
|
|
nonContainerizedProc1Cgroup = []byte(`14:name=systemd:/
|
|
13:hugetlb:/
|
|
12:net_prio:/
|
|
11:perf_event:/
|
|
10:bfqio:/
|
|
9:blkio:/
|
|
8:net_cls:/
|
|
7:freezer:/
|
|
6:devices:/
|
|
5:memory:/
|
|
4:cpuacct:/
|
|
3:cpu:/
|
|
2:cpuset:/
|
|
`)
|
|
containerizedProc1Cgroup = []byte(`9:perf_event:/docker/3cef1b53c50b0fa357d994f8a1a8cd783c76bbf4f5dd08b226e38a8bd331338d
|
|
8:blkio:/docker/3cef1b53c50b0fa357d994f8a1a8cd783c76bbf4f5dd08b226e38a8bd331338d
|
|
7:net_cls:/
|
|
6:freezer:/docker/3cef1b53c50b0fa357d994f8a1a8cd783c76bbf4f5dd08b226e38a8bd331338d
|
|
5:devices:/docker/3cef1b53c50b0fa357d994f8a1a8cd783c76bbf4f5dd08b226e38a8bd331338d
|
|
4:memory:/docker/3cef1b53c50b0fa357d994f8a1a8cd783c76bbf4f5dd08b226e38a8bd331338d
|
|
3:cpuacct:/docker/3cef1b53c50b0fa357d994f8a1a8cd783c76bbf4f5dd08b226e38a8bd331338d
|
|
2:cpu:/docker/3cef1b53c50b0fa357d994f8a1a8cd783c76bbf4f5dd08b226e38a8bd331338d
|
|
1:cpuset:/`)
|
|
)
|
|
|
|
dir := os.TempDir()
|
|
proc1Cgroup = filepath.Join(dir, "proc1Cgroup")
|
|
|
|
defer func() {
|
|
os.Remove(proc1Cgroup)
|
|
proc1Cgroup = backup
|
|
}()
|
|
|
|
if err := ioutil.WriteFile(proc1Cgroup, nonContainerizedProc1Cgroup, 0600); err != nil {
|
|
t.Fatalf("failed to write to %s: %v", proc1Cgroup, err)
|
|
}
|
|
inContainer, err := IsContainerized()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if inContainer {
|
|
t.Fatal("Wrongly assuming containerized")
|
|
}
|
|
|
|
if err := ioutil.WriteFile(proc1Cgroup, nonContainerizedProc1Cgroupsystemd226, 0600); err != nil {
|
|
t.Fatalf("failed to write to %s: %v", proc1Cgroup, err)
|
|
}
|
|
inContainer, err = IsContainerized()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if inContainer {
|
|
t.Fatal("Wrongly assuming containerized for systemd /init.scope cgroup layout")
|
|
}
|
|
|
|
if err := ioutil.WriteFile(proc1Cgroup, containerizedProc1Cgroup, 0600); err != nil {
|
|
t.Fatalf("failed to write to %s: %v", proc1Cgroup, err)
|
|
}
|
|
inContainer, err = IsContainerized()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !inContainer {
|
|
t.Fatal("Wrongly assuming non-containerized")
|
|
}
|
|
}
|
|
|
|
func TestOsReleaseFallback(t *testing.T) {
|
|
var backup = etcOsRelease
|
|
var altBackup = altOsRelease
|
|
dir := os.TempDir()
|
|
etcOsRelease = filepath.Join(dir, "etcOsRelease")
|
|
altOsRelease = filepath.Join(dir, "altOsRelease")
|
|
|
|
defer func() {
|
|
os.Remove(dir)
|
|
etcOsRelease = backup
|
|
altOsRelease = altBackup
|
|
}()
|
|
content := `NAME=Gentoo
|
|
ID=gentoo
|
|
PRETTY_NAME="Gentoo/Linux"
|
|
ANSI_COLOR="1;32"
|
|
HOME_URL="http://www.gentoo.org/"
|
|
SUPPORT_URL="http://www.gentoo.org/main/en/support.xml"
|
|
BUG_REPORT_URL="https://bugs.gentoo.org/"
|
|
`
|
|
if err := ioutil.WriteFile(altOsRelease, []byte(content), 0600); err != nil {
|
|
t.Fatalf("failed to write to %s: %v", etcOsRelease, err)
|
|
}
|
|
s, err := GetOperatingSystem()
|
|
if err != nil || s != "Gentoo/Linux" {
|
|
t.Fatalf("Expected %q, got %q (err: %v)", "Gentoo/Linux", s, err)
|
|
}
|
|
}
|