add gop.GetFileClassType

cl/builtin_test.go

@@ -314,6 +314,15 @@ func TestClassNameAndExt(t *testing.T) {
 	}
 }
 
+func TestClassTypeForTest(t *testing.T) {
+	if name := ClassTypeForTest("foo"); name != "case_foo" {
+		t.Fatal("ClassTypeForTest error:", name)
+	}
+	if name := ClassTypeForTest("Foo"); name != "caseFoo" {
+		t.Fatal("ClassTypeForTest error:", name)
+	}
+}
+
 func TestErrMultiStarRecv(t *testing.T) {
 	defer func() {
 		if e := recover(); e == nil {

cl/classfile.go

@@ -96,6 +96,10 @@ func ClassNameAndExt(file string) (name, clsfile, ext string) {
 	return
 }
 
+func ClassTypeForTest(classType string) string {
+	return casePrefix + testNameSuffix(classType)
+}
+
 func isGoxTestFile(ext string) bool {
 	return strings.HasSuffix(ext, "test.gox")
 }

load.go

@@ -403,3 +403,32 @@ var (
 )
 
 // -----------------------------------------------------------------------------
+
+// GetFileClassType get gop module file classType.
+func GetFileClassType(mod *gopmod.Module, filename string, file *ast.File) (classType string, isTest bool, ok bool) {
+	if file.IsClass {
+		var ext string
+		classType, _, ext = cl.ClassNameAndExt(filename)
+		ok = true
+		if file.IsNormalGox {
+			isTest = strings.HasSuffix(ext, "_test.gox")
+			if !isTest && classType == "main" {
+				classType = "_main"
+			}
+		} else {
+			isTest = strings.HasSuffix(ext, "test.gox")
+		}
+		if file.IsProj {
+			if gt, ok := mod.LookupClass(ext); ok {
+				classType = gt.Class
+			}
+		} else if isTest {
+			classType = cl.ClassTypeForTest(classType)
+		}
+	} else if strings.HasSuffix(filename, "_test.gop") {
+		isTest = true
+	}
+	return
+}
+
+// -----------------------------------------------------------------------------