add memory limit

dsl/memory_test.go

@@ -0,0 +1,95 @@
+package dsl
+
+import "testing"
+
+func TestParseSizeLiteral(t *testing.T) {
+	cases := []struct {
+		in   string
+		want int64
+	}{
+		{"256", 256},
+		{"256B", 256},
+		{"1K", 1024},
+		{"2KB", 2 * 1024},
+		{"4KiB", 4 * 1024},
+		{"256M", 256 * 1024 * 1024},
+		{"256MB", 256 * 1024 * 1024},
+		{"512MiB", 512 * 1024 * 1024},
+		{"1G", 1024 * 1024 * 1024},
+		{"2GB", 2 * 1024 * 1024 * 1024},
+		{"3GiB", 3 * 1024 * 1024 * 1024},
+	}
+	for _, c := range cases {
+		got, err := parseSizeLiteral(c.in, 0, 0)
+		if err != nil {
+			t.Errorf("parseSizeLiteral(%q) error: %v", c.in, err)
+			continue
+		}
+		if got != c.want {
+			t.Errorf("parseSizeLiteral(%q) = %d, want %d", c.in, got, c.want)
+		}
+	}
+}
+
+func TestParseSizeLiteralInvalid(t *testing.T) {
+	bad := []string{"abc", "100TB", "10XB", ""}
+	for _, s := range bad {
+		if _, err := parseSizeLiteral(s, 0, 0); err == nil {
+			t.Errorf("parseSizeLiteral(%q) expected error", s)
+		}
+	}
+}
+
+func TestParseMemoryLimit(t *testing.T) {
+	src := `
+build "go build -o solution ."
+timeout 10s
+memory_limit = 256MB
+
+group("g1") {
+    weight = 0.5
+    memory_limit = 128MiB
+
+    test("t1") {
+        stdout = "ok\n"
+    }
+
+    test("t2") {
+        memory_limit = 64M
+        stdout = "ok\n"
+    }
+}
+
+group("g2") {
+    weight = 0.5
+
+    test("inherits") {
+        stdout = "ok\n"
+    }
+}
+`
+	f, _, err := Parse(src)
+	if err != nil {
+		t.Fatalf("parse error: %v", err)
+	}
+	if f.MemoryLimit != 256*1024*1024 {
+		t.Errorf("file memory: got %d", f.MemoryLimit)
+	}
+	g1 := f.Groups[0]
+	if g1.MemoryLimit != 128*1024*1024 {
+		t.Errorf("g1 memory: got %d", g1.MemoryLimit)
+	}
+	if g1.Tests[0].MemoryLimit != 128*1024*1024 {
+		t.Errorf("g1.t1 memory (inherited from group): got %d", g1.Tests[0].MemoryLimit)
+	}
+	if g1.Tests[1].MemoryLimit != 64*1024*1024 {
+		t.Errorf("g1.t2 memory (override): got %d", g1.Tests[1].MemoryLimit)
+	}
+	g2 := f.Groups[1]
+	if g2.MemoryLimit != 256*1024*1024 {
+		t.Errorf("g2 memory (inherited from file): got %d", g2.MemoryLimit)
+	}
+	if g2.Tests[0].MemoryLimit != 256*1024*1024 {
+		t.Errorf("g2.inherits memory: got %d", g2.Tests[0].MemoryLimit)
+	}
+}