dorkgen/google_test.go

package dorkgen

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

var dork *GoogleSearch

func TestToUrl(t *testing.T) {
	dork = &GoogleSearch{}

	result := dork.
		Site("example.com").
		ToURL()

	assert.Equal(t, result, "https://www.google.com/search?q=site%3Aexample.com", "they should be equal")
}

func TestSite(t *testing.T) {
	dork = &GoogleSearch{}

	result := dork.
		Site("example.com").
		ToString()

	assert.Equal(t, result, "site:example.com", "they should be equal")
}

func TestIntext(t *testing.T) {
	dork = &GoogleSearch{}

	result := dork.
		Intext("text").
		ToString()

	assert.Equal(t, result, "\"text\"", "they should be equal")
}

func TestInurl(t *testing.T) {
	dork = &GoogleSearch{}

	result := dork.
		Inurl("index.php").
		ToString()

	assert.Equal(t, result, "inurl:\"index.php\"", "they should be equal")
}

func TestFiletype(t *testing.T) {
	dork = &GoogleSearch{}

	result := dork.
		Filetype("pdf").
		ToString()

	assert.Equal(t, result, "filetype:\"pdf\"", "they should be equal")
}

func TestCache(t *testing.T) {
	dork = &GoogleSearch{}

	result := dork.
		Cache("www.google.com").
		ToString()

	assert.Equal(t, result, "cache:\"www.google.com\"", "they should be equal")
}

func TestRelated(t *testing.T) {
	dork = &GoogleSearch{}

	result := dork.
		Related("www.google.com").
		ToString()

	assert.Equal(t, result, "related:\"www.google.com\"", "they should be equal")
}

func TestExt(t *testing.T) {
	dork = &GoogleSearch{}

	result := dork.
		Ext("(doc | pdf | xls | txt | xml)").
		ToString()

	assert.Equal(t, result, "ext:(doc | pdf | xls | txt | xml)", "they should be equal")
}

func TestExclude(t *testing.T) {
	dork = &GoogleSearch{}

	result := dork.
		Exclude("html").
		Exclude("htm").
		Exclude("php").
		Exclude("md5sums").
		ToString()

	assert.Equal(t, result, "-html -htm -php -md5sums", "they should be equal")
}

func TestOr(t *testing.T) {
	dork = &GoogleSearch{}

	result := dork.
		Site("facebook.com").
		Or().
		Site("twitter.com").
		ToString()

	assert.Equal(t, result, "site:facebook.com OR site:twitter.com", "they should be equal")
}

func TestGroup(t *testing.T) {
	dork = &GoogleSearch{}

	result := dork.
		Site("linkedin.com").
		Group((&GoogleSearch{}).Intext("1").Or().Intext("2").ToString()).
		ToString()

	assert.Equal(t, result, "site:linkedin.com (\"1\" OR \"2\")", "they should be equal")
}