From 6a0f64e7c9b65df294e2e2bcdb11515d0e070b1f Mon Sep 17 00:00:00 2001
From: Peter Hunt <pehunt@redhat.com>
Date: Thu, 18 Jun 2020 10:01:39 -0400
Subject: [PATCH] systemd: add unit tests for systemdVersion

Signed-off-by: Peter Hunt <pehunt@redhat.com>
---
 libcontainer/cgroups/systemd/systemd_test.go | 32 ++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 libcontainer/cgroups/systemd/systemd_test.go

diff --git a/libcontainer/cgroups/systemd/systemd_test.go b/libcontainer/cgroups/systemd/systemd_test.go
new file mode 100644
index 00000000..0c98a117
--- /dev/null
+++ b/libcontainer/cgroups/systemd/systemd_test.go
@@ -0,0 +1,32 @@
+package systemd
+
+import (
+	"testing"
+)
+
+func TestSystemdVersion(t *testing.T) {
+	var systemdVersionTests = []struct {
+		verStr      string
+		expectedVer int
+		expectErr   bool
+	}{
+		{`"219"`, 219, false},
+		{`"v245.4-1.fc32"`, 245, false},
+		{`"241-1"`, 241, false},
+		{`"v241-1"`, 241, false},
+		{"NaN", 0, true},
+		{"", 0, true},
+	}
+	for _, sdTest := range systemdVersionTests {
+		ver, err := systemdVersionAtoi(sdTest.verStr)
+		if !sdTest.expectErr && err != nil {
+			t.Errorf("systemdVersionAtoi(%s); want nil; got %v", sdTest.verStr, err)
+		}
+		if sdTest.expectErr && err == nil {
+			t.Errorf("systemdVersionAtoi(%s); wanted failure; got nil", sdTest.verStr)
+		}
+		if ver != sdTest.expectedVer {
+			t.Errorf("systemdVersionAtoi(%s); want %d; got %d", sdTest.verStr, sdTest.expectedVer, ver)
+		}
+	}
+}