Refactor gtest adapter to fit new pipeline

1 files changed, 186 insertions, 0 deletions

diff --git a/adapters/gtest/executable.go b/adapters/gtest/executable.go
new file mode 100644
index 0000000..78b0b56
--- /dev/null
+++ b/adapters/gtest/executable.go
@@ -0,0 +1,186 @@
+package gtest
+
+import (
+        "os"
+	"errors"
+        "time"
+	"io/ioutil"
+	"log"
+	"os/exec"
+	"path"
+	"reflect"
+	"sort"
+        "context"
+
+	"golang.flu0r1ne.net/planr"
+)
+
+type executable struct {
+  exeNm  string
+  testpath string
+  srcs     []string
+  tcs      []planr.TestCase
+}
+
+func createExecutables(tcs []planr.TestCase) []executable {
+  exes := make(map[string] executable, 0)
+
+  for _, tc := range tcs {
+    cfg := tc.AdapterConfig().(*Config)
+    file := *cfg.Testfile
+    exe, contained := exes[file]
+
+    // For set comparison
+    sort.Strings(cfg.Srcs)
+
+    if !contained {
+      exeTcs := make([]planr.TestCase, 1)
+      exeTcs[0] = tc
+
+
+      exe := executable {
+        planr.Cname("", file),
+        file,
+        cfg.Srcs,
+        exeTcs,
+      }
+
+      exes[file] = exe
+      
+      continue
+    }
+
+    // We could create two different executables for each source list
+    // But, that would be confusing so we're going to disallow it
+    if !reflect.DeepEqual(exe.srcs, cfg.Srcs) {
+      log.Fatalf(
+        "Two test case definitions %s and %s have different lists of sources", 
+        exe.testpath, *cfg.Testfile,
+      )
+    }
+
+    exe.tcs = append(exe.tcs, tc)
+    
+    exes[file] = exe
+  }
+
+  exesList := make([]executable, 0)
+
+  for _, exe := range exes {
+    exesList = append(exesList, exe)
+  }
+
+  return exesList
+}
+
+func (exe executable) compile(builddir string) (succeeded bool, buildFailures []planr.TestResult) {
+  cmd := exec.Command("make", "-C", builddir, exe.exeNm)
+  out, err := cmd.CombinedOutput()
+  buildFailures = make([]planr.TestResult, 0)
+
+  outputLog := string(out)
+
+  if err != nil{
+    var exiterr *exec.ExitError
+    if errors.As(err, &exiterr) && exiterr.ExitCode() == 0 {
+      log.Fatalf("Unrecoverable build failure: %v", err)
+    }
+
+    for i := range exe.tcs {
+      res := planr.TestResult {}
+      res.Tc = exe.tcs[i]
+      res.DebugOutput = outputLog
+      res.Status = planr.COMPILATION_FAILURE
+
+      buildFailures = append(buildFailures, res)
+    }
+
+    succeeded = false
+
+    return
+  }
+
+  succeeded = true
+  return
+}
+
+const TMPFILENAME = "gtest_adapter_*.json"
+
+func runGtest(exe string, tc planr.TestCase, builddir string) planr.TestResult {
+  result := planr.TestResult {}
+  result.Tc = tc
+
+  exePath := path.Join(builddir, exe)
+  cfg := tc.AdapterConfig().(*Config)
+
+  f, err := ioutil.TempFile(builddir, TMPFILENAME)
+      
+  if err != nil {
+    log.Fatal(err)
+  }
+
+  timeout := time.Duration(*cfg.Timeout) * time.Millisecond
+
+  ctx, cancel := context.WithTimeout(context.Background(), timeout)
+
+  jsonFlag := "--gtest_output=json:" + f.Name()
+  testFlag := "--gtest_filter=" + *cfg.Suite + "." + *cfg.Name
+
+  cmd := exec.CommandContext(ctx, exePath, jsonFlag, testFlag)
+
+  defer cancel()
+  defer os.Remove(f.Name())
+
+  out, err := cmd.CombinedOutput()
+  if err != nil {
+    var exiterr *exec.ExitError
+
+    if !errors.As(err, &exiterr) {
+      log.Printf("%v\n", err)
+      os.Exit(exiterr.ExitCode())
+    }
+  }
+
+  results, err := decodeResults(f)
+
+  if err != nil {
+    log.Fatalf("Could not collect results from %s: %v", exe, err)
+  }
+
+  if len(results) < 1 {
+    log.Fatalf(
+      "Could not find testcase %s with name=\"%s\" and suite=\"%s\". Does such a test exist in the test source?",
+      tc.Cname,
+      *cfg.Name,
+      *cfg.Suite,
+    )
+  }
+
+  // TODO: Cleanup -- ZERO TESTS?
+  if len(results) > 1 {
+    log.Fatalf("Unexpected number of results")
+  }
+
+  decodeResult := results[0]
+  
+  result.TestOutput = string(out)
+
+  if decodeResult.pass {
+    result.Status = planr.PASSING
+  } else {
+    result.Status = planr.RUNTIME_FAILURE
+  }
+
+  return result
+}
+
+func (exe executable) execute(builddir string) []planr.TestResult {
+  results := make([]planr.TestResult, len(exe.tcs))
+
+  for i := range exe.tcs {
+    results[i] = runGtest(exe.exeNm, exe.tcs[i], builddir)
+  }
+
+  return results
+}
+