Option Explicit

'=============================================================================
' TestModule - Unit tests for GaugeClassifier
' Validates all classification logic against PRD test cases
'=============================================================================

' Main test runner - validates all test cases from PRD
Sub TestClassifierLogic()
    Dim classifier As New GaugeClassifier
    Dim testCases As Variant
    Dim results As String
    Dim passedCount As Long
    Dim failedCount As Long
    Dim i As Long
    Dim actual As String
    Dim expected As String
    Dim passed As Boolean

    ' Initialize test suite (Input, Expected Output)
    testCases = Array( _
        Array("PYTH-100.A0.531.M201.M02|BP-088.2312.M02.0A3|FBP-251040.RF1.049.13.M20F.B.DPA", "喷涂产品"), _
        Array("PYTH-063.A0.513.M05.U38.F6A.K15.1.1|BP-058.1509.M05.BB2.F6", "卫生型隔膜表"), _
        Array("PYTH-063.A0.513.M05.U38.F6BP.K15.1.1|BP-058.1509.M05.BB2.F6", "喷涂产品"), _
        Array("PYTH-063.A0.513.M201.M09|BP-058.1509.M09.0A2|FC-1520.RF1.033.1.M20F.B.DPA", "隔膜表"), _
        Array("PYXHN-100.A0.531.M201.M09.H4|BP-088.2518.M09.0A3.YXH|F5.M42M.M20F.030.3.033.1.B", "隔膜表"), _
        Array("YCSH-100.A0.531.M201.M07|BP-088.2312.M07.0A3.YCS", "差压表"), _
        Array("YE-070.A0.204.G144.K21|BP-058.1509.K21.0A2", "膜盒压力表"), _
        Array("YB-120.A0.407.Z144.M23|BP-110.5250.P60.0A5", "精密压力表"), _
        Array("YX-100.A0.200.M204.M16.H4|BP-098.0088.M16.0A3.YX", "电接点压力表"), _
        Array("Y-040.Z0.200.M104.M07|BP-037.1007.M07.0A2", "常规表"), _
        Array("Y-100.BT.200.M204.M12|BP-098.0088.B17.PA3", "常规表"), _
        Array("", "其他") _
    )

    passedCount = 0
    failedCount = 0
    results = "========================================" & vbCrLf
    results = results & "GaugeClassifier Test Results" & vbCrLf
    results = results & "========================================" & vbCrLf & vbCrLf

    ' Run all test cases
    For i = LBound(testCases) To UBound(testCases)
        expected = testCases(i)(1)
        actual = classifier.Classify(CStr(testCases(i)(0)))
        passed = (actual = expected)

        If passed Then
            passedCount = passedCount + 1
            results = results & "[✓ PASS] Test " & (i + 1) & ": " & expected & vbCrLf
        Else
            failedCount = failedCount + 1
            results = results & "[✗ FAIL] Test " & (i + 1) & vbCrLf
            results = results & "  Input:    " & Left(CStr(testCases(i)(0)), 50) & "..." & vbCrLf
            results = results & "  Expected: " & expected & vbCrLf
            results = results & "  Actual:   " & actual & vbCrLf & vbCrLf
        End If
    Next i

    ' Summary
    results = results & "========================================" & vbCrLf
    results = results & "Summary:" & vbCrLf
    results = results & "  Total:   " & (passedCount + failedCount) & vbCrLf
    results = results & "  Passed:  " & passedCount & vbCrLf
    results = results & "  Failed:  " & failedCount & vbCrLf
    results = results & "========================================"

    ' Output to Immediate Window
    Debug.Print results

    ' Display message box
    If failedCount = 0 Then
        MsgBox "所有测试通过! (" & passedCount & "/" & (passedCount + failedCount) & ")", vbInformation, "测试成功"
    Else
        MsgBox "测试失败: " & failedCount & " 个失败" & vbCrLf & _
               "详见立即窗口 (Ctrl+G)", vbExclamation, "测试失败"
    End If
End Sub

' Test individual parsing methods
Sub TestParsingMethods()
    Dim classifier As New GaugeClassifier
    Dim testString As String
    Dim head As String
    Dim flange As String
    Dim mainModel As String

    Debug.Print "========================================"
    Debug.Print "Parsing Method Tests"
    Debug.Print "========================================"

    ' Test 1: Full model with flange
    testString = "PYTH-100.A0.531.M201.M06|BP-088.2312.M06.0A3|FB-5020.RF3.049.1"
    head = classifier.GetHead(testString)
    flange = classifier.GetFlange(testString)
    mainModel = classifier.GetMainModel(head)
    Debug.Print "Test 1: Full model with flange"
    Debug.Print "  Input:     " & testString
    Debug.Print "  Head:      " & head
    Debug.Print "  Flange:    " & flange
    Debug.Print "  MainModel: " & mainModel
    Debug.Print ""

    ' Test 2: Model without flange
    testString = "Y-040.Z0.200.M104.M07|BP-037.1007.M07.0A2"
    head = classifier.GetHead(testString)
    flange = classifier.GetFlange(testString)
    mainModel = classifier.GetMainModel(head)
    Debug.Print "Test 2: Model without flange"
    Debug.Print "  Input:     " & testString
    Debug.Print "  Head:      " & head
    Debug.Print "  Flange:    " & flange
    Debug.Print "  MainModel: " & mainModel
    Debug.Print ""

    ' Test 3: Single character main model
    testString = "Y-100.BT.200.M204.M12|BP-098.0088.B17.PA3"
    head = classifier.GetHead(testString)
    flange = classifier.GetFlange(testString)
    mainModel = classifier.GetMainModel(head)
    Debug.Print "Test 3: Single character main model"
    Debug.Print "  Input:     " & testString
    Debug.Print "  Head:      " & head
    Debug.Print "  Flange:    " & flange
    Debug.Print "  MainModel: " & mainModel
    Debug.Print ""

    ' Test 4: Spray product with F6AP
    testString = "PYTH-063.A0.513.M05.U38.F6AP.K15.1.1|BP-058.1509.M05.BB2.F6"
    head = classifier.GetHead(testString)
    flange = classifier.GetFlange(testString)
    mainModel = classifier.GetMainModel(head)
    Debug.Print "Test 4: Spray product with F6AP"
    Debug.Print "  Input:     " & testString
    Debug.Print "  Head:      " & head
    Debug.Print "  Flange:    " & flange
    Debug.Print "  MainModel: " & mainModel

    Debug.Print "========================================"
    Debug.Print "Parsing tests complete. Check results above."
End Sub

' Performance test with large dataset
Sub TestPerformance()
    Dim classifier As New GaugeClassifier
    Dim testStrings() As String
    Dim startTime As Double
    Dim endTime As Double
    Dim i As Long
    Dim count As Long

    count = 1000 ' Test with 1000 records

    ' Generate test data
    ReDim testStrings(1 To count)
    For i = 1 To count
        testStrings(i) = "PYTH-100.A0.531.M201.M02|BP-088.2312.M02.0A3|FBP-251040.RF1.049.13.M20F.B.DPA"
    Next i

    ' Measure performance
    startTime = Timer
    For i = 1 To count
        classifier.Classify(testStrings(i))
    Next i
    endTime = Timer

    Debug.Print "========================================"
    Debug.Print "Performance Test"
    Debug.Print "========================================"
    Debug.Print "Records processed: " & count
    Debug.Print "Time elapsed: " & Format(endTime - startTime, "0.000") & " seconds"
    Debug.Print "Average per record: " & Format((endTime - startTime) / count, "0.0000") & " seconds"
    Debug.Print "Records per second: " & Format(count / (endTime - startTime), "0")
    Debug.Print "========================================"

    MsgBox "性能测试完成!" & vbCrLf & _
           "处理 " & count & " 条记录，耗时: " & Format(endTime - startTime, "0.000") & " 秒", _
           vbInformation, "性能测试"
End Sub

' Interactive test - single input
Sub TestSingleInput()
    Dim classifier As New GaugeClassifier
    Dim inputStr As String
    Dim result As String

    inputStr = InputBox("请输入型号代码:", "单个型号测试")

    If inputStr <> "" Then
        result = classifier.Classify(inputStr)
        Debug.Print "========================================"
        Debug.Print "Single Input Test"
        Debug.Print "========================================"
        Debug.Print "Input:  " & inputStr
        Debug.Print "Output: " & result
        Debug.Print "========================================"

        MsgBox "型号: " & inputStr & vbCrLf & _
               "分类: " & result, vbInformation, "测试结果"
    End If
End Sub