GradeUserCode

Author: markhuang1212

backend/cmd/exec_user_code.go

@@ -78,12 +78,11 @@ func ExecUserCode(ctx context.Context, gr types.GradeRequest, tmpDir string) (*E
 	}
 
 	defer func() {
-		err := cli.ContainerRemove(ctxdl, resp.ID, dockertypes.ContainerRemoveOptions{
+		err := cli.ContainerRemove(ctx, resp.ID, dockertypes.ContainerRemoveOptions{
 			Force: true,
 		})
 		if err != nil {
 			log.Panicln("cannot kill and remove container")
-			panic(err)
 		}
 	}()
 

backend/cmd/grade_user_code.go

@@ -19,6 +19,7 @@ func GradeUserCode(ctx context.Context, gr types.GradeRequest) (*types.GradeResu
 		return nil, errors.Wrap(err, "cannot create tmpDir")
 	}
 	defer os.RemoveAll(tmpDir)
+	os.Chmod(tmpDir, 0777)
 
 	cr, err := CompileUserCode(ctx, gr, tmpDir)
 	if err != nil {

backend/cmd/grade_user_code_test.go

@@ -0,0 +1,32 @@
+package cmd_test
+
+import (
+	"context"
+	"testing"
+
+	"github.com/markhuang1212/code-grader/backend/cmd"
+	"github.com/markhuang1212/code-grader/backend/types"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestGradeUserCode1(t *testing.T) {
+	ctx := context.Background()
+	gr := types.GradeRequest{
+		TestCaseName: "example-1",
+		UserCode:     "int main() { cout << \"Hello\" << endl; }",
+	}
+	result, err := cmd.GradeUserCode(ctx, gr)
+	assert.Nil(t, err)
+	assert.Equal(t, result.Status, types.GradeResultSuccess)
+}
+
+func TestGradeUserCode2(t *testing.T) {
+	ctx := context.Background()
+	gr := types.GradeRequest{
+		TestCaseName: "example-1",
+		UserCode:     "int main() { while (1) { } }",
+	}
+	result, err := cmd.GradeUserCode(ctx, gr)
+	assert.Nil(t, err)
+	assert.Equal(t, result.Status, types.GradeResultTimeLimitExceed)
+}

testcases/example-1/answer.txt

@@ -3,7 +3,7 @@ int main() {
     cin >> w;
     if (w == "Hello")
     {
-        cout << "World" << endl;
+        cout << "Hello" << endl;
     }
     return 0;
 }

testcases/test.sh

@@ -0,0 +1,17 @@
+#!/bin/bash
+
+TEST_CASE_DIR=$1
+
+cd ${TEST_CASE_DIR}
+cat prepend.txt > /tmp/main.cpp
+cat answer.txt >> /tmp/main.cpp
+cat append.txt >> /tmp/main.cpp
+
+g++ -std=c++11 -o /tmp/a.out /tmp/main.cpp
+/tmp/a.out < input.txt > /tmp/output.txt
+diff output.txt /tmp/output.txt
+
+if [ $? -eq 0 ]
+then
+    echo "SUCCESS"
+fi