package main

import (
	"testing"
)

func TestCreateDomain(t *testing.T) {
	d := VirtualDomain{Name: "example.org"}
	if !d.CreateDomain() {
		t.Fatal("Creating domain failed.")
	}
}

func TestDomainExists(t *testing.T) {
	d := VirtualDomain{Name: "blablabla"}
	if d.DomainExists() {
		t.Fatal("'" + d.Name + "' does not exists.")
	}
	d.Name = "example.org"
	if !d.DomainExists() {
		t.Fatal("'" + d.Name + "' should exist.")
	}
}

func TestValidateDomain(t *testing.T) {
	d := VirtualDomain{Name: "blablabla"}
	if d.ValidateDomain() {
		t.Fatal(d.Name + " is not a valid domain.")
	}
	d.Name = "dfalsdf.adfjalf.example.org"
	if d.ValidateDomain() {
		t.Fatal("'" + d.Name + "' is not a valid domain.")
	}
	d.Name = "https://google.com"
	if d.ValidateDomain() {
		t.Fatal("'" + d.Name + "' is not a valid domain.")
	}
	d.Name = "google.com"
	if !d.ValidateDomain() {
		t.Fatal("'" + d.Name + "' is a valid domain.")
	}
}

func TestGetDomain(t *testing.T) {
	domain := GetDomain("foo@example.org")
	if domain != "example.org" {
		t.Fatal("Can't get the domain from the adress.")
	}
}

func TestGetPrimaryKey(t *testing.T) {
	d := VirtualDomain{Name: "example.org"}
	if d.GetPrimaryKey() != 1 {
		t.Fatal("Primary key should be 1.")
	}
}