From c9905747e6669ca16e2f13a126cb0686d9ff9944 Mon Sep 17 00:00:00 2001
From: Chris Baker <1675087+cgbaker@users.noreply.github.com>
Date: Wed, 27 Jan 2021 14:44:07 +0000
Subject: [PATCH] e2e: java driver isolation tests

---
 e2e/isolation/input/alloc_exec.nomad      |  2 +-
 e2e/isolation/input/alloc_exec_java.nomad | 39 +++++++++++
 e2e/isolation/input/java.nomad            | 39 +++++++++++
 e2e/isolation/isolation.go                | 85 +++++++++++++++++++++++
 4 files changed, 164 insertions(+), 1 deletion(-)
 create mode 100644 e2e/isolation/input/alloc_exec_java.nomad
 create mode 100644 e2e/isolation/input/java.nomad

diff --git a/e2e/isolation/input/alloc_exec.nomad b/e2e/isolation/input/alloc_exec.nomad
index 4c36d4af6..2135d05eb 100644
--- a/e2e/isolation/input/alloc_exec.nomad
+++ b/e2e/isolation/input/alloc_exec.nomad
@@ -13,7 +13,7 @@ job "alloc_exec" {
 
       config {
         command = "/bin/sleep"
-        args = ["30s"]
+        args    = ["30s"]
       }
 
       resources {
diff --git a/e2e/isolation/input/alloc_exec_java.nomad b/e2e/isolation/input/alloc_exec_java.nomad
new file mode 100644
index 000000000..7f284079b
--- /dev/null
+++ b/e2e/isolation/input/alloc_exec_java.nomad
@@ -0,0 +1,39 @@
+job "java_sleep" {
+  datacenters = ["dc1"]
+  type = "batch"
+
+  group "java" {
+
+    task "build" {
+      lifecycle {
+        hook = "prestart"
+        sidecar = false
+      }
+
+      driver = "exec"
+      config {
+        command = "javac"
+        args = ["-d", "${NOMAD_ALLOC_DIR}", "local/Sleep.java"]
+      }
+
+      template {
+        destination = "local/Sleep.java"
+        data = <<EOH
+public class Sleep {
+    public static void main(String... s) throws Exception {
+        Thread.sleep(30000);
+    }
+}
+EOH
+      }
+    }
+
+    task "sleep" {
+      driver = "java"
+      config {
+        class_path = "${NOMAD_ALLOC_DIR}"
+        class = "Sleep"
+      }
+    }
+  }
+}
\ No newline at end of file
diff --git a/e2e/isolation/input/java.nomad b/e2e/isolation/input/java.nomad
new file mode 100644
index 000000000..68e10045e
--- /dev/null
+++ b/e2e/isolation/input/java.nomad
@@ -0,0 +1,39 @@
+job "java_pid" {
+  datacenters = ["dc1"]
+  type = "batch"
+
+  group "java" {
+
+    task "build" {
+      lifecycle {
+        hook = "prestart"
+        sidecar = false
+      }
+
+      driver = "exec"
+      config {
+        command = "javac"
+        args = ["-d", "${NOMAD_ALLOC_DIR}", "local/Pid.java"]
+      }
+
+      template {
+        destination = "local/Pid.java"
+        data = <<EOH
+public class Pid {
+    public static void main(String... s) throws Exception {
+        System.out.println("my pid is " + ProcessHandle.current().pid());
+    }
+}
+EOH
+      }
+    }
+
+    task "pid" {
+      driver = "java"
+      config {
+        class_path = "${NOMAD_ALLOC_DIR}"
+        class = "Pid"
+      }
+    }
+  }
+}
\ No newline at end of file
diff --git a/e2e/isolation/isolation.go b/e2e/isolation/isolation.go
index bf0f6e00f..7cfd32c8b 100644
--- a/e2e/isolation/isolation.go
+++ b/e2e/isolation/isolation.go
@@ -121,6 +121,91 @@ func (tc *IsolationTest) TestIsolation_ExecDriver_PIDNamespacing_AllocExec(f *fr
 	require.Len(t, lines, 3)
 }
 
+func (tc *IsolationTest) TestIsolation_JavaDriver_PIDNamespacing(f *framework.F) {
+	t := f.T()
+
+	clientNodes, err := e2eutil.ListLinuxClientNodes(tc.Nomad())
+	require.Nil(t, err)
+
+	if len(clientNodes) == 0 {
+		t.Skip("no Linux clients")
+	}
+
+	uuid := uuid.Generate()
+	jobID := "isolation-pid-namespace-" + uuid[0:8]
+	file := "isolation/input/java.nomad"
+	allocs := e2eutil.RegisterAndWaitForAllocs(t, tc.Nomad(), file, jobID, "")
+	require.Equal(t, len(allocs), 1, fmt.Sprintf("failed to register %s", jobID))
+
+	tc.jobIDs = append(tc.jobIDs, jobID)
+	defer func() {
+		tc.Nomad().Jobs().Deregister(jobID, true, nil)
+	}()
+
+	allocID := allocs[0].ID
+	e2eutil.WaitForAllocStopped(t, tc.Nomad(), allocID)
+
+	out, err := e2eutil.AllocTaskLogs(allocID, "pid", e2eutil.LogsStdOut)
+	require.NoError(t, err, fmt.Sprintf("could not get logs for alloc %s", allocID))
+
+	require.Contains(t, out, "my pid is 1\n")
+}
+
+func (tc *IsolationTest) TestIsolation_JavaDriver_PIDNamespacing_AllocExec(f *framework.F) {
+	t := f.T()
+
+	clientNodes, err := e2eutil.ListLinuxClientNodes(tc.Nomad())
+	require.Nil(t, err)
+
+	if len(clientNodes) == 0 {
+		t.Skip("no Linux clients")
+	}
+
+	uuid := uuid.Generate()
+	jobID := "isolation-pid-namespace-" + uuid[0:8]
+	file := "isolation/input/alloc_exec_java.nomad"
+	allocs := e2eutil.RegisterAndWaitForAllocs(t, tc.Nomad(), file, jobID, "")
+	require.Equal(t, len(allocs), 1, fmt.Sprintf("failed to register %s", jobID))
+
+	defer func() {
+		tc.Nomad().Jobs().Deregister(jobID, true, nil)
+	}()
+
+	allocID := allocs[0].ID
+	e2eutil.WaitForAllocTaskRunning(t, tc.Nomad(), allocID, "sleep")
+
+	alloc, _, err := tc.Nomad().Allocations().Info(allocID, nil)
+	require.NoError(t, err)
+	require.NotNil(t, alloc)
+
+	resizeCh := make(chan api.TerminalSize)
+	var tty bool
+
+	ctx, cancelFn := context.WithTimeout(context.Background(), 15*time.Second)
+	defer cancelFn()
+
+	var stdout, stderr bytes.Buffer
+
+	exitCode, err := tc.Nomad().Allocations().Exec(
+		ctx,
+		alloc,
+		"sleep",
+		tty,
+		[]string{"ps", "ax"},
+		bytes.NewReader([]byte("")),
+		&stdout,
+		&stderr,
+		resizeCh,
+		nil,
+	)
+	require.NoError(t, err)
+	require.Equal(t, 0, exitCode)
+
+	lines := strings.Split(strings.TrimSpace(stdout.String()), "\n")
+	// header, sleep process, ps ax process are the only output lines expected
+	require.Len(t, lines, 3)
+}
+
 func (tc *IsolationTest) TestIsolation_RawExecDriver_NoPIDNamespacing(f *framework.F) {
 	t := f.T()