"""Test UCC counters for Java and Assembly."""

from pathlib import Path
from pyucc.core.ucc_java_counter import UCCJavaCounter
from pyucc.core.ucc_assembly_counter import UCCAssemblyCounter
import tempfile


def test_java_counter():
    """Test Java counter on a sample file."""
    java_code = """
package com.example.test;

import java.util.ArrayList;
import java.util.HashMap;

/**
 * This is a multi-line
 * comment block
 */
public class TestClass {
    private int count;  // embedded comment
    private String name;
    
    public TestClass(int count, String name) {
        this.count = count;
        this.name = name;
    }
    
    public void execute() {
        for (int i = 0; i < count; i++) {
            System.out.println(name);
        }
    }
}
"""

    with tempfile.NamedTemporaryFile(mode="w", suffix=".java", delete=False) as f:
        f.write(java_code)
        temp_path = Path(f.name)

    try:
        counter = UCCJavaCounter()
        results = counter.analyze_file(temp_path)

        print("=== Java Counter Test ===")
        print(f"Blank: {results['blank_lines']}")
        print(
            f"Comments (W/E): {results['comment_whole']}/{results['comment_embedded']}"
        )
        print(f"Directives: {results['compiler_directives']}")
        print(f"Data: {results['data_declarations']}")
        print(f"Exec: {results['exec_instructions']}")
        print(f"Logical SLOC: {results['logical_sloc']}")
        print(f"Physical SLOC: {results['physical_sloc']}")

        # Assertions
        assert results["blank_lines"] > 0, "Should have blank lines"
        assert results["comment_whole"] > 0, "Should have whole comments"
        assert results["comment_embedded"] > 0, "Should have embedded comments"
        assert results["compiler_directives"] >= 2, "Should have import directives"
        assert results["data_declarations"] > 0, "Should have data declarations"
        assert results["exec_instructions"] > 0, "Should have exec instructions"
        assert results["logical_sloc"] > 0, "Should have logical SLOC"
        assert results["physical_sloc"] > 0, "Should have physical SLOC"

        print("✓ Java counter test PASSED\n")

    finally:
        temp_path.unlink()


def test_assembly_counter():
    """Test Assembly counter on a sample file."""
    asm_code = """
; This is a comment
    .data
myvar:  .word 0     ; data declaration
buffer: .space 100

    .text
    .globl main
main:
    # Another comment style
    li $t0, 5       ; load immediate
    move $a0, $t0   # move register
    jal print       ; jump and link
    
print:
    /* multi-line
       comment block */
    syscall
    jr $ra
"""

    with tempfile.NamedTemporaryFile(mode="w", suffix=".asm", delete=False) as f:
        f.write(asm_code)
        temp_path = Path(f.name)

    try:
        counter = UCCAssemblyCounter()
        results = counter.analyze_file(temp_path)

        print("=== Assembly Counter Test ===")
        print(f"Blank: {results['blank_lines']}")
        print(
            f"Comments (W/E): {results['comment_whole']}/{results['comment_embedded']}"
        )
        print(f"Directives: {results['compiler_directives']}")
        print(f"Data: {results['data_declarations']}")
        print(f"Exec: {results['exec_instructions']}")
        print(f"Logical SLOC: {results['logical_sloc']}")
        print(f"Physical SLOC: {results['physical_sloc']}")

        # Assertions
        assert results["blank_lines"] > 0, "Should have blank lines"
        assert results["comment_whole"] > 0, "Should have whole comments"
        assert results["comment_embedded"] > 0, "Should have embedded comments"
        assert (
            results["data_declarations"] > 0
        ), "Should have data declarations (in .data section)"
        assert (
            results["exec_instructions"] > 0
        ), "Should have exec instructions (in .text section)"
        assert results["logical_sloc"] > 0, "Should have logical SLOC"
        assert results["physical_sloc"] > 0, "Should have physical SLOC"

        print("✓ Assembly counter test PASSED\n")

    finally:
        temp_path.unlink()


if __name__ == "__main__":
    test_java_counter()
    test_assembly_counter()
    print("✅ All tests passed!")