package main

import (
	"bytes"
	"strings"
	"testing"
)

func TestListDomainSubscriber(t *testing.T) {
	app := buildNewCmd()

	app.SetArgs([]string{"config", "import", "--json", "../misc/exampleConfig.json"})
	app.Execute()

	buffer := new(bytes.Buffer)
	app.SetOutput(buffer)

	app.SetArgs([]string{"domain", "list"})
	app.Execute()

	output := buffer.String()
	valid := ` - Subscriber domain :
subscription.example.jp
Total : 1
`
	if output != valid {
		t.Fatalf("Invalid Responce.")
	}

	relayState.RedisClient.FlushAll().Result()
	relayState.Load()
}

func TestListDomainLimited(t *testing.T) {
	app := buildNewCmd()

	app.SetArgs([]string{"config", "import", "--json", "../misc/exampleConfig.json"})
	app.Execute()

	buffer := new(bytes.Buffer)
	app.SetOutput(buffer)

	app.SetArgs([]string{"domain", "list", "-t", "limited"})
	app.Execute()

	output := buffer.String()
	valid := ` - Limited domain :
limitedDomain.example.jp
Total : 1
`
	if output != valid {
		t.Fatalf("Invalid Responce.")
	}

	relayState.RedisClient.FlushAll().Result()
	relayState.Load()
}

func TestListDomainBlocked(t *testing.T) {
	app := buildNewCmd()

	app.SetArgs([]string{"config", "import", "--json", "../misc/exampleConfig.json"})
	app.Execute()

	buffer := new(bytes.Buffer)
	app.SetOutput(buffer)

	app.SetArgs([]string{"domain", "list", "-t", "blocked"})
	app.Execute()

	output := buffer.String()
	valid := ` - Blocked domain :
blockedDomain.example.jp
Total : 1
`
	if output != valid {
		t.Fatalf("Invalid Responce.")
	}

	relayState.RedisClient.FlushAll().Result()
	relayState.Load()
}

func TestSetDomainBlocked(t *testing.T) {
	app := buildNewCmd()

	app.SetArgs([]string{"domain", "set", "-t", "blocked", "testdomain.example.jp"})
	app.Execute()

	valid := false
	for _, domain := range relayState.BlockedDomains {
		if domain == "testdomain.example.jp" {
			valid = true
		}
	}

	if !valid {
		t.Fatalf("Not set blocked domain")
	}

	relayState.RedisClient.FlushAll().Result()
	relayState.Load()
}

func TestSetDomainLimited(t *testing.T) {
	app := buildNewCmd()

	app.SetArgs([]string{"domain", "set", "-t", "limited", "testdomain.example.jp"})
	app.Execute()

	valid := false
	for _, domain := range relayState.LimitedDomains {
		if domain == "testdomain.example.jp" {
			valid = true
		}
	}

	if !valid {
		t.Fatalf("Not set limited domain")
	}

	relayState.RedisClient.FlushAll().Result()
	relayState.Load()
}

func TestUnsetDomainBlocked(t *testing.T) {
	app := buildNewCmd()

	app.SetArgs([]string{"config", "import", "--json", "../misc/exampleConfig.json"})
	app.Execute()

	app.SetArgs([]string{"domain", "set", "-t", "blocked", "-u", "blockedDomain.example.jp"})
	app.Execute()

	valid := true
	for _, domain := range relayState.BlockedDomains {
		if domain == "blockedDomain.example.jp" {
			valid = false
		}
	}

	if !valid {
		t.Fatalf("Not unset blocked domain")
	}

	relayState.RedisClient.FlushAll().Result()
	relayState.Load()
}

func TestUnsetDomainLimited(t *testing.T) {
	app := buildNewCmd()

	app.SetArgs([]string{"config", "import", "--json", "../misc/exampleConfig.json"})
	app.Execute()

	app.SetArgs([]string{"domain", "set", "-t", "limited", "-u", "limitedDomain.example.jp"})
	app.Execute()

	valid := true
	for _, domain := range relayState.LimitedDomains {
		if domain == "limitedDomain.example.jp" {
			valid = false
		}
	}

	if !valid {
		t.Fatalf("Not unset blocked domain")
	}

	relayState.RedisClient.FlushAll().Result()
	relayState.Load()
}

func TestSetDomainInvalid(t *testing.T) {
	app := buildNewCmd()

	app.SetArgs([]string{"config", "import", "--json", "../misc/exampleConfig.json"})
	app.Execute()

	buffer := new(bytes.Buffer)
	app.SetOutput(buffer)

	app.SetArgs([]string{"domain", "set", "-t", "hoge", "hoge.example.jp"})
	app.Execute()

	output := buffer.String()
	if strings.Split(output, "\n")[0] != "Invalid type given" {
		t.Fatalf("Invalid Responce.")
	}

	relayState.RedisClient.FlushAll().Result()
	relayState.Load()
}

func TestUnfollowDomain(t *testing.T) {
	app := buildNewCmd()

	app.SetArgs([]string{"config", "import", "--json", "../misc/exampleConfig.json"})
	app.Execute()

	buffer := new(bytes.Buffer)
	app.SetOutput(buffer)

	app.SetArgs([]string{"domain", "unfollow", "subscription.example.jp"})
	app.Execute()

	valid := true
	for _, domain := range relayState.LimitedDomains {
		if domain == "subscription.example.jp" {
			valid = false
		}
	}

	if !valid {
		t.Fatalf("Do not unfollow domain")
	}

	relayState.RedisClient.FlushAll().Result()
	relayState.Load()
}