package store

import (
	"crypto/sha1"
	"errors"
	"testing"

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

func TestUser_EncodeID(t *testing.T) {
	tbl := []struct {
		id   string
		hash string
	}{
		{"myid", "6e34471f84557e1713012d64a7477c71bfdac631"},
		{"", "da39a3ee5e6b4b0d3255bfef95601890afd80709"},
		{"blah blah", "135a1e01bae742c4a576b20fd41a683f6483ca43"},
		{"da39a3ee5e6b4b0d3255bfef95601890afd80709", "da39a3ee5e6b4b0d3255bfef95601890afd80709"},
	}

	for i, tt := range tbl {
		assert.Equal(t, tt.hash, EncodeID(tt.id), "case #%d", i)
	}
}

func TestUser_HashIP(t *testing.T) {
	tbl := []struct {
		ip           string
		hash1, hash2 string
	}{
		{"127.0.0.1", "ae12fe3b5f129b5cc4cdd2b136b7b7947c4d2741", "dbc7c999343f003f189f70aaf52cc04443f90790"},
		{"8.8.8.8", "8cee77c27e32a2b5aec95c29888ac9946618d9a2", "70a46afce9633f010b06e129b8ad08243a1c4da9"},
		{"8cee77c27e32a2b5aec95c29888ac9946618d9a2", "8cee77c27e32a2b5aec95c29888ac9946618d9a2", "8cee77c27e32a2b5aec95c29888ac9946618d9a2"},
		{"", "fbdb1d1b18aa6c08324b7d64b71fb76370690e1d", "823688dafca7393d24c871a2da98a84d8732e927"},
	}

	for i, tt := range tbl {
		u := User{IP: tt.ip}
		u.HashIP("")
		assert.Equal(t, tt.hash1, u.IP, "case #%d", i)

		u = User{IP: tt.ip}
		u.HashIP("123456")
		assert.Equal(t, tt.hash2, u.IP, "case #%d", i)
	}
}

func TestUser_HashFailed(t *testing.T) {
	r := hashWithFallback(mockHash{}, "123456789")
	assert.Equal(t, "995dc9bbdf1939fa", r)

	r = hashWithFallback(mockHash{}, "995dc9bbdf1939fa")
	assert.Equal(t, "995dc9bbdf1939fa", r)

	r = hashWithFallback(sha1.New(), "123456789")
	assert.Equal(t, "f7c3bc1d808e04732adf679965ccc34ca7ae3441", r)

}

type mockHash struct{}

func (mock mockHash) Sum(b []byte) []byte               { return nil }
func (mock mockHash) Reset()                            {}
func (mock mockHash) Size() int                         { return 0 }
func (mock mockHash) BlockSize() int                    { return 0 }
func (mock mockHash) Write(p []byte) (n int, err error) { return 0, errors.New("error") }