test: update testcase

Author: SunBK201

tests/samples/c/main.c

@@ -14,7 +14,7 @@ int main(int argc, char** argv)
     int c = count + argc;
     while (a < 10 && add(c, c)) {
         a += 1;
-        if (b > 5) 
+        if (b > 5)
             a -= 1;
         else if (b < 3) {
             a += 2;
@@ -52,7 +52,7 @@ int mul(int a, int b)
     return a * add(a, b);
 }
 
-int cp(int *x, int *y)
+int cp(int* x, int* y)
 {
     *x = *y;
     return 0;
@@ -61,13 +61,13 @@ int cp(int *x, int *y)
 int test_switch(int a)
 {
     switch (a) {
-        case 1:
-            return 1;
-        case 2:
-            return 2;
-        default:
-            a = 0;
-        }
+    case 1:
+        return 1;
+    case 2:
+        return 2;
+    default:
+        a = 0;
+    }
     return a;
 }
 

tests/test_file.py

@@ -14,12 +14,19 @@ def setUp(self):
         )
         self.file = self.project.files.get("main.c")
 
+    def test_file_create(self):
+        file = scubatrace.File.create(
+            str(self.project_path / "main.c"),
+            self.project,
+        )
+        self.assertIsNotNone(file)
+
     def test_file_imports(self):
         assert self.file is not None
         imports = self.file.imports
         self.assertGreater(len(imports), 0)
         for imp in imports:
-            self.assertIsNotNone(imp.name)
+            self.assertTrue(imp.name in ["stdio.h", "sub.h"])
 
     def test_file_functions(self):
         assert self.file is not None
@@ -28,13 +35,38 @@ def test_file_functions(self):
         for func in functions:
             self.assertIsNotNone(func.name)
 
+    def test_file_function_by_line(self):
+        assert self.file is not None
+        function = self.file.function_by_line(5)
+        self.assertIsNotNone(function)
+        assert function is not None
+        self.assertEqual(function.name, "add")
+
     def test_file_statements(self):
         assert self.file is not None
         statements = self.file.statements
         self.assertGreater(len(statements), 0)
         for stmt in statements:
             self.assertIsNotNone(stmt.text)
 
+    def test_file_statement_by_line(self):
+        assert self.file is not None
+        statements = self.file.statements_by_line(14)
+        self.assertGreater(len(statements), 0)
+        statement = statements[0]
+        self.assertIsNotNone(statement)
+        self.assertEqual(statement.text, "int c = count + argc;")
+
+        self.assertEqual(len(self.file.statements_by_line(-1)), 0)
+        self.assertGreater(len(self.file.statements_by_line(1)), 0)
+
+    def test_file_identifiers(self):
+        assert self.file is not None
+        identifiers = self.file.identifiers
+        self.assertGreater(len(identifiers), 0)
+        for identifier in identifiers:
+            self.assertIsNotNone(identifier.name)
+
     def test_file_variables(self):
         assert self.file is not None
         variables = self.file.variables
@@ -48,3 +80,22 @@ def test_file_cfg(self):
         self.assertIsNotNone(cfg)
         self.assertGreater(len(cfg.nodes), 0)
         self.assertGreater(len(cfg.edges), 0)
+
+    def test_file_query(self):
+        assert self.file is not None
+        query_str = """
+            (call_expression
+                function: (identifier)@func
+                (#eq? @func "sub")
+            )@call
+        """
+        query = self.file.query(query_str)
+        target_lines = [6, 36]
+        self.assertEqual(len(query), len(target_lines))
+        for stat in query:
+            self.assertIn(stat.start_line, target_lines)
+
+        one_shot_result = self.file.query_oneshot(query_str)
+        self.assertIsNotNone(one_shot_result)
+        assert one_shot_result is not None
+        self.assertIn(one_shot_result.start_line, target_lines)

tests/test_function.py

@@ -0,0 +1,22 @@
+import unittest
+from pathlib import Path
+
+import scubatrace
+
+
+class TestProject(unittest.TestCase):
+    def setUp(self):
+        self.test_dir = Path(__file__).parent
+        self.samples_dir = self.test_dir / "samples"
+        self.project_path = self.samples_dir / "c"
+        self.project = scubatrace.Project.create(
+            str(self.project_path), language=scubatrace.language.C
+        )
+        self.file = self.project.files.get("main.c")
+        assert self.file is not None
+        self.function = self.file.function_by_line(11)
+
+    def test_function_create(self):
+        assert self.function is not None
+        function = scubatrace.Function.create(self.function.node, self.function.parent)
+        self.assertIsNotNone(function)

tests/test_identifer.py

@@ -0,0 +1,24 @@
+import unittest
+from pathlib import Path
+
+import scubatrace
+
+
+class TestProject(unittest.TestCase):
+    def setUp(self):
+        self.test_dir = Path(__file__).parent
+        self.samples_dir = self.test_dir / "samples"
+        self.project_path = self.samples_dir / "c"
+        self.project = scubatrace.Project.create(
+            str(self.project_path), language=scubatrace.language.C
+        )
+        self.file = self.project.files.get("main.c")
+        assert self.file is not None
+        self.assertGreater(len(self.file.statements), 0)
+        self.statement = self.file.statements_by_line(14)[0]
+        self.assertGreater(len(self.statement.identifiers), 0)
+        self.identifier = self.statement.identifiers[0]
+
+    def test_identifier_create(self):
+        identifier = scubatrace.Identifier(self.identifier.node, self.statement)
+        self.assertIsNotNone(identifier)

tests/test_statement.py

@@ -0,0 +1,28 @@
+import unittest
+from pathlib import Path
+
+import scubatrace
+
+
+class TestProject(unittest.TestCase):
+    def setUp(self):
+        self.test_dir = Path(__file__).parent
+        self.samples_dir = self.test_dir / "samples"
+        self.project_path = self.samples_dir / "c"
+        self.project = scubatrace.Project.create(
+            str(self.project_path), language=scubatrace.language.C
+        )
+        self.file = self.project.files.get("main.c")
+        assert self.file is not None
+        self.function = self.file.function_by_line(11)
+        assert self.function is not None
+        statement = self.function.statements_by_line(14)
+        assert statement is not None and len(statement) == 1
+        self.statement = statement[0]
+
+    def test_statement_create(self):
+        assert self.statement is not None
+        statement = scubatrace.SimpleStatement.create(
+            self.statement.node, self.statement.parent
+        )
+        self.assertIsNotNone(statement)