package mailgun
import (
"context"
"fmt"
"github.com/hashicorp/terraform/helper/resource"
"github.com/hashicorp/terraform/terraform"
"github.com/mailgun/mailgun-go"
"os"
"strconv"
"testing"
"time"
)
type fullDomain struct {
domainResponse mailgun.DomainResponse
domainConnection mailgun.DomainConnection
domainTracking mailgun.DomainTracking
ipAddress []string
credentials []mailgun.Credential
}
func getFullDomain(mg *mailgun.MailgunImpl, domainName string) (*fullDomain, error) {
ctx, cancel := context.WithTimeout(context.Background(), time.Second*30)
defer cancel()
mg = mailgun.NewMailgun(domainName, mg.APIKey())
var domain fullDomain
var err error
domain.domainResponse, err = mg.GetDomain(ctx, domainName)
if err != nil {
return nil, fmt.Errorf("Error Getting mailgun domain Details for %s: Error: %s", domainName, err)
}
domain.domainConnection, err = mg.GetDomainConnection(ctx, domainName)
if err != nil {
return nil, fmt.Errorf("Error Getting mailgun domain connection Details for %s: Error: %s", domainName, err)
}
domain.domainTracking, err = mg.GetDomainTracking(ctx, domainName)
if err != nil {
return nil, fmt.Errorf("Error Getting mailgun domain tracking Details for %s: Error: %s", domainName, err)
}
ipAddress, err := mg.ListDomainIPS(ctx)
if err != nil {
return nil, fmt.Errorf("Error Getting mailgun domain ips2 for %s: Error: %s", domainName, err)
}
ips := make([]string, len(ipAddress))
for i, r := range ipAddress {
ips[i] = r.IP
}
domain.ipAddress = ips
domain.credentials, err = ListCredentials(domainName, mg.APIKey())
if err != nil {
return nil, fmt.Errorf("Error Getting mailgun credentials for %s: Error: %s", domainName, err)
}
return &domain, nil
}
func TestAccMailgunDomain_basic(t *testing.T) {
var domain fullDomain
resource.Test(t, resource.TestCase{
PreCheck: func() { testAccPreCheck(t) },
Providers: testAccProviders,
CheckDestroy: testAccDomainCheckDestroy(&domain),
Steps: []resource.TestStep{
{
Config: interpolateTerraformTemplateDomain(testAccDomainConfig_basic),
Check: resource.ComposeTestCheckFunc(
testAccDomainCheckExists("mailgun_domain.exemple", &domain),
testAccDomainCheckAttributes("mailgun_domain.exemple", &domain),
),
},
},
})
}
func TestAccMailgunDomain_withUpdate(t *testing.T) {
var domain fullDomain
resource.Test(t, resource.TestCase{
PreCheck: func() { testAccPreCheck(t) },
Providers: testAccProviders,
CheckDestroy: testAccDomainCheckDestroy(&domain),
Steps: []resource.TestStep{
{
Config: interpolateTerraformTemplateDomain(testAccDomainConfig_basic),
Check: resource.ComposeTestCheckFunc(
testAccDomainCheckExists("mailgun_domain.exemple", &domain),
testAccDomainCheckAttributes("mailgun_domain.exemple", &domain),
),
},
{
Config: interpolateTerraformTemplateDomain(testAccDomainConfig_update),
Check: resource.ComposeTestCheckFunc(
testAccDomainCheckExists("mailgun_domain.exemple", &domain),
testAccDomainCheckAttributes("mailgun_domain.exemple", &domain),
),
},
},
})
}
func testAccDomainCheckExists(rn string, domain *fullDomain) resource.TestCheckFunc {
return func(s *terraform.State) error {
rs, ok := s.RootModule().Resources[rn]
if !ok {
return fmt.Errorf("resource not found: %s", rn)
}
if rs.Primary.ID == "" {
return fmt.Errorf("domainID not set")
}
mg := testAccProvider.Meta().(*mailgun.MailgunImpl)
domainId := rs.Primary.ID
gotDomain, err := getFullDomain(mg, domainId)
if err != nil {
return fmt.Errorf("error getting domain: %s", err)
}
*domain = *gotDomain
return nil
}
}
func testAccDomainCheckAttributes(rn string, domain *fullDomain) resource.TestCheckFunc {
return func(s *terraform.State) error {
attrs := s.RootModule().Resources[rn].Primary.Attributes
check := func(key, stateValue, domainValue string) error {
if domainValue != stateValue {
return fmt.Errorf("different values for %s in state (%s) and in mailgun (%s)",
key, stateValue, domainValue)
}
return nil
}
for key, value := range attrs {
var err error
switch key {
case "name":
err = check(key, value, domain.domainResponse.Domain.Name)
case "smtp_password":
err = check(key, value, domain.domainResponse.Domain.SMTPPassword)
case "smtp_login":
err = check(key, value, domain.domainResponse.Domain.SMTPLogin)
case "wildcard":
err = check(key, value, strconv.FormatBool(domain.domainResponse.Domain.Wildcard))
case "state":
err = check(key, value, domain.domainResponse.Domain.State)
case "open_tracking_settings_active":
err = check(key, value, strconv.FormatBool(domain.domainTracking.Open.Active))
case "click_tracking_settings_active":
err = check(key, value, strconv.FormatBool(domain.domainTracking.Click.Active))
case "unsubscribe_tracking_settings_active":
err = check(key, value, strconv.FormatBool(domain.domainTracking.Unsubscribe.Active))
case "unsubscribe_tracking_settings_html_footer":
err = check(key, value, domain.domainTracking.Unsubscribe.HTMLFooter)
case "unsubscribe_tracking_settings_text_footer":
err = check(key, value, domain.domainTracking.Unsubscribe.TextFooter)
case "skip_verification":
err = check(key, value, strconv.FormatBool(domain.domainConnection.SkipVerification))
case "require_tls":
err = check(key, value, strconv.FormatBool(domain.domainConnection.RequireTLS))
}
if err != nil {
return err
}
}
return nil
}
}
func testAccDomainCheckDestroy(domain *fullDomain) resource.TestCheckFunc {
return func(s *terraform.State) error {
mg := testAccProvider.Meta().(*mailgun.MailgunImpl)
ctx, cancel := context.WithTimeout(context.Background(), time.Second*30)
defer cancel()
_, err := mg.GetDomain(ctx, domain.domainResponse.Domain.Name)
if err == nil {
return fmt.Errorf("domain still exists")
}
return nil
}
}
func interpolateTerraformTemplateDomain(template string) string {
domainName := ""
if v := os.Getenv("MAILGUN_DOMAIN"); v != "" {
domainName = v
}
return fmt.Sprintf(template, domainName)
}
const testAccDomainConfig_basic = `
resource "mailgun_domain" "exemple" {
name="%s"
}
`
const testAccDomainConfig_update = `
resource "mailgun_domain" "exemple" {
name="%s"
}
`
