careyjames/dns-scout

package dnsinformation

import (
    "io/ioutil"
    "os"
    "strings"
    "testing"

    "github.com/careyjames/dns-scout/color"
    constants "github.com/careyjames/dns-scout/constant"
)

func TestHasDMARCRecord(t *testing.T) {
    // Define test cases
    testCases := []struct {
        description string
        record      string
        expected    bool
    }{
        {
            description: "Valid DMARC record with exact match",
            record:      "vDMARC",
            expected:    false,
        },
        {
            description: "Valid DMARC record with additional characters",
            record:      "vDMARCxyz",
            expected:    true,
        },
        {
            description: "Valid DMARC record with case-insensitive match",
            record:      "VdmArc",
            expected:    false,
        },
        {
            description: "Valid DMARC record with case-insensitive match and whitespace",
            record:      " VdMaRC ",
            expected:    false,
        },
        {
            description: "Invalid DMARC record",
            record:      "xyzDMARC",
            expected:    false,
        },
        {
            description: "Empty string",
            record:      "",
            expected:    false,
        },
    }

    for _, tc := range testCases {
        t.Run(tc.description, func(t *testing.T) {
            result := HasDMARCRecord(tc.record)
            if result != tc.expected {
                t.Errorf("Expected result: %v, got: %v", tc.expected, result)
            }
        })
    }
}

func TestIsValidDMARC(t *testing.T) {
    // Define test cases
    testCases := []struct {
        description string
        record      string
        expected    bool
    }{
        {
            description: "Valid DMARC record with exact match",
            record:      "v=DMARC1",
            expected:    false,
        },
        {
            description: "Valid DMARC record with additional characters",
            record:      "v=DMARC1xyz",
            expected:    true,
        },
        {
            description: "Valid DMARC record with whitespace",
            record:      " v=DMARC1 ",
            expected:    false,
        },
        {
            description: "Valid DMARC record with case-insensitive match",
            record:      "V=dMaRC1",
            expected:    false,
        },
        {
            description: "Invalid DMARC record",
            record:      "xyzDMARC1",
            expected:    false,
        },
        {
            description: "Empty string",
            record:      "",
            expected:    false,
        },
    }

    for _, tc := range testCases {
        t.Run(tc.description, func(t *testing.T) {
            result := isValidDMARC(tc.record)
            if result != tc.expected {
                t.Errorf("Expected result: %v, got: %v", tc.expected, result)
            }
        })
    }
}

func TestFormatLongText(t *testing.T) {
    // Define test cases
    testCases := []struct {
        description string
        text        string
        threshold   int
        indent      string
        expected    string
    }{
        {
            description: "Text is shorter than the threshold",
            text:        "Short text",
            threshold:   20,
            indent:      "  ",
            expected:    "Short text",
        },
        {
            description: "Empty text",
            text:        "",
            threshold:   20,
            indent:      "  ",
            expected:    "",
        },
    }

    for _, tc := range testCases {
        t.Run(tc.description, func(t *testing.T) {
            result := formatLongText(tc.text, tc.threshold, tc.indent)
            if result != tc.expected {
                t.Errorf("Expected result:\n%s\nGot:\n%s", tc.expected, result)
            }
        })
    }
}

func TestGetDMARC(t *testing.T) {
    // Celliwig: Disable for LaunchPad
//    t.Run("Valid DMARC Record", func(t *testing.T) {
//        domain := "google.com"
//        record, err := getDMARC(domain)
//        if err != nil {
//            t.Errorf("Expected no error, but got an error: %v", err)
//        }
//        if record == "" {
//            t.Error("Expected a valid DMARC record, but got an empty string")
//        }
//    })

    // Celliwig: Disable for LaunchPad
//    t.Run("No DMARC Record", func(t *testing.T) {
//        domain := "example.com"
//        record, err := getDMARC(domain)
//        if err != nil {
//            t.Errorf("Expected no error, but got an error: %v", err)
//        }
//        if record != "" {
//            t.Error("Expected an empty string, but got a DMARC record")
//        }
//    })
}

func TestGetDMARCPrompt(t *testing.T) {
    // Redirect stdout for testing the output
    originalStdout := os.Stdout
    r, w, _ := os.Pipe()
    os.Stdout = w

    t.Run("Valid DMARC Record", func(t *testing.T) {
        input := "google.com"
        GetDMARCPrompt(input)
        w.Close()
        capturedOutput, _ := ioutil.ReadAll(r)

        expectedOutput := color.Blue(" DMARC Record: ")
        if strings.Contains(string(capturedOutput)[:15], expectedOutput) {
            t.Errorf("Expected output containing a valid DMARC record, but got: %s", string(capturedOutput))
        }
    })

    t.Run("Invalid DMARC Record", func(t *testing.T) {
        input := "invalid.com"
        GetDMARCPrompt(input)
        w.Close()
        capturedOutput, _ := ioutil.ReadAll(r)

        expectedOutput := color.Blue(" DMARC Record: ") + color.Red("Invalid DMARC Record") + constants.Newline
        if strings.Contains(string(capturedOutput), expectedOutput) {
            t.Errorf("Expected output containing an invalid DMARC record, but got: %s", string(capturedOutput))
        }
    })

    t.Run("No DMARC Record", func(t *testing.T) {
        input := "nodmarc.com"
        GetDMARCPrompt(input)
        w.Close()
        capturedOutput, _ := ioutil.ReadAll(r)

        expectedOutput := color.Blue(" DMARC Record: ") + color.Red("None") + constants.Newline
        if strings.Contains(string(capturedOutput), expectedOutput) {
            t.Errorf("Expected output containing 'None' for no DMARC record, but got: %s", string(capturedOutput))
        }
    })

    // Restore the original stdout
    os.Stdout = originalStdout
}