g-node-gin/internal/db/db_gin_test.go

package db

import (
	"io/ioutil"
	"math/rand"
	"path/filepath"
	"strings"
	"testing"
	"time"

	"github.com/G-Node/gogs/internal/setting"
)

const ALNUM = "1234567890abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"

var emails = []string{
	"foo@example.org",
	"spammer@example.com",
	"user@malicious-domain.net",
	"someone@example.mal",
}

var blockEverythingFilter = []string{
	"- .*",
}

var allowEverythingFilter = []string{
	"+ .*",
}

var allowGNodeFilter = []string{
	"+ @g-node.org$",
	"- .*",
}

var blockMalicious = []string{
	"- .*malicious-domain.net$",
	"- spammer@",
}

// Writes the filters to a file and returns the directory that contains the
// filter file (to be set as setting.CustomPath)
func writeFilterFile(t *testing.T, filters []string) string {
	path := t.TempDir()
	fname := filepath.Join(path, "addressfilters")

	if err := ioutil.WriteFile(fname, []byte(strings.Join(filters, "\n")), 0777); err != nil {
		t.Fatalf("Failed to write line filters to file %q: %v", fname, err.Error())
	}
	return path
}

// randAlnum returns a random alphanumeric (lowercase, latin) string of length 'n'.
func randAlnum(n int) string {
	N := len(ALNUM)
	chrs := make([]byte, n)
	for idx := range chrs {
		chrs[idx] = ALNUM[rand.Intn(N)]
	}

	return string(chrs)
}

func randAddress() string {
	user := randAlnum(rand.Intn(20))
	domain := randAlnum(rand.Intn(20)) + "." + randAlnum(rand.Intn(3))

	return string(user) + "@" + string(domain)
}

func TestAllowGNodeFilter(t *testing.T) {
	setting.CustomPath = writeFilterFile(t, allowGNodeFilter)

	for _, address := range emails {
		if isAddressAllowed(address) {
			t.Fatalf("Address %q should be blocked but was allowed", address)
		}
	}

	if !isAddressAllowed("me@g-node.org") {
		t.Fatalf("G-Node address blocked but should be allowed")
	}

	if isAddressAllowed("malicious@g-node.org@piracy.tk") {
		t.Fatalf("Malicious address should be blocked but was allowed")
	}
}

func TestEverythingFilters(t *testing.T) {
	setting.CustomPath = writeFilterFile(t, allowEverythingFilter)
	rand.Seed(time.Now().UnixNano())

	for idx := 0; idx < 100; idx++ {
		randress := randAddress()
		if !isAddressAllowed(randAddress()) {
			t.Fatalf("Address %q should be allowed but was blocked", randress)
		}
	}

	setting.CustomPath = writeFilterFile(t, blockEverythingFilter)

	for idx := 0; idx < 100; idx++ {
		randress := randAddress()
		if isAddressAllowed(randAddress()) {
			t.Fatalf("Address %q should be blocked but was allowed", randress)
		}
	}
}

func TestBlockDomainFilter(t *testing.T) {
	setting.CustomPath = writeFilterFile(t, blockMalicious)

	// 0, 3 should be allowed; 1, 2 should be blocked
	if address := emails[0]; !isAddressAllowed(address) {
		t.Fatalf("Address %q should be allowed but was blocked", address)
	}

	if address := emails[1]; isAddressAllowed(address) {
		t.Fatalf("Address %q should be blocked but was allowed", address)
	}

	if address := emails[2]; isAddressAllowed(address) {
		t.Fatalf("Address %q should be blocked but was allowed", address)
	}

	if address := emails[3]; !isAddressAllowed(address) {
		t.Fatalf("Address %q should be allowed but was blocked", address)
	}
}

func TestFiltersNone(t *testing.T) {
	setting.CustomPath = filepath.Join(t.TempDir(), "does", "not", "exist")
	if address := emails[3]; !isAddressAllowed(address) {
		t.Fatalf("Address %q should be allowed but was blocked", address)
	}
}