package rfc



import (
	"testing"

	"github.com/zmap/zlint/v3/lint"
	"github.com/zmap/zlint/v3/test"
)

func TestNewRsaAllowedKUEe(t *testing.T) {
	testCases := []struct {
		name            string
		filename        string
		expectedStatus  lint.LintStatus
		expectedDetails string
	}{
		{
			name:            "Certificate with EC key",
			filename:        "ecdsaP384.pem",
			expectedStatus:  lint.NA,
			expectedDetails: "",
		},
		{
			name:            "Subscriber certificate with RSA key and key usages digitalSignature and nonRepudiation",
			filename:        "eeWithRSAAllowedKeyUsage.pem",
			expectedStatus:  lint.Pass,
			expectedDetails: "",
		},
		{
			name:            "CA certificate with RSA key and key usages digitalSignature, certificateSign, and crlSign",
			filename:        "caBasicConstCrit.pem",
			expectedStatus:  lint.NA,
			expectedDetails: "",
		},
		{
			name:            "Subscriber certificate with RSA key and key usages digitalSignature and nonRepudiation older than",
			filename:        "eeWithRSAAllowedKeyUsageOld.pem",
			expectedStatus:  lint.NE,
			expectedDetails: "",
		},
		{
			name:            "Subscriber certificate with RSA key and key usage keyAgreement",
			filename:        "eeWithRSADisallowedKeyUsage.pem",
			expectedStatus:  lint.Error,
			expectedDetails: "Subscriber certificate with an RSA key contains invalid key usage(s): KeyUsageKeyAgreement",
		},
	}

	for _, tc := range testCases {
		result := test.TestLint("e_rsa_allowed_ku_ee", tc.filename)
		if result.Status != tc.expectedStatus {
			t.Errorf("expected result %v. actual result was %v",
				tc.expectedStatus, result.Status)
		}
		if result.Details != tc.expectedDetails {
			t.Errorf("expected details %q. actual result was %q",
				tc.expectedDetails, result.Details)
		}
	}
}
