package main

import (
	"testing"
)

// TestGetFileName test func getFileName
func TestGetFileName(t *testing.T) {
	// all flags are added
	mk := &mkcert{
		keyFileFlag:  "test-key-name.pem",
		certFileFlag: "test-cert-name.pem",
		p12FileFlag:  "test.p12",
	}
	// check keyname, the result should be customized
	keyname, err := mk.getFileName("key", []string{"example.com"})
	if err != nil {
		t.Error("failed to get customized key file name")
	}
	if keyname != "test-key-name.pem" {
		t.Error("keyname check failed")
	}

	// check certname, the result should be customized
	certname, err := mk.getFileName("cert", []string{"example.com"})
	if err != nil {
		t.Error("failed to get customized cert file name")
	}
	if certname != "test-cert-name.pem" {
		t.Error("certname check failed")
	}

	// check p12name, the result should be customized
	p12name, err := mk.getFileName("p12", []string{"example.com"})
	if err != nil {
		t.Error("failed to get customized p12 file name")
	}
	if p12name != "test.p12" {
		t.Error("p12 check failed")
	}

	// default name will be generated if no flags passed
	mk = &mkcert{
		keyFileFlag: "test-key-name.pem",
	}
	// check keyname again, the result should be custoomized due to keyFileFlag
	keyname, err = mk.getFileName("key", []string{"example.com", "localhost"})
	if err != nil {
		t.Error("failed to get customized key file name")
	}
	if keyname != "test-key-name.pem" {
		t.Error("keyname check failed")
	}

	// check default certname, the result should be default file name generated by original principle
	certname, err = mk.getFileName("cert", []string{"*.example.com", "localhost"})
	if err != nil {
		t.Error("failed to get default cert file name")
	}
	if certname != "_wildcard.example.com+1.pem" {
		t.Error("certname check failed")
	}

	// check default p12name,  the result should be default file name generated by original principle
	p12name, err = mk.getFileName("p12", []string{"x.co:y.com"})
	if err != nil {
		t.Error("failed to get default p12 file name")
	}
	if p12name != "x.co_y.com.p12" {
		t.Error("p12 check failed")
	}
}