package main

import (
	"testing"

	"github.com/matryer/is"
)

func TestSocials(t *testing.T) {
	is := is.New(t)
	u := `https://vk.com/supergoup`
	URL := parseUrlParams(u)
	hit := map[string]string{}
	hit["Referer"] = "https://vk.com"
	socialNetwork(hit, URL)

	val, ok := hit["SocialNetwork"]
	is.True(ok)
	is.Equal(val, "Vkontakte")
}

func TestSocialsAway(t *testing.T) {
	is := is.New(t)
	u := `https://away.vk.com/supergoup`
	URL := parseUrlParams(u)
	hit := map[string]string{}
	hit["Referer"] = "https://vk.com"
	socialNetwork(hit, URL)

	val, ok := hit["SocialNetwork"]
	is.True(ok)
	is.Equal(val, "Vkontakte")
}