lib_test.go

package dejavu

import (
	"crypto/rand"
	"testing"
)

func TestDeterministic(t *testing.T) {

	d := New(false, 3, 0.0)

	// add entries
	if d.Witness([]byte("foo")) {
		t.Errorf("Incorrect déjà vu: 'foo'!")
	}
	if d.Witness([]byte("bar")) {
		t.Errorf("Incorrect déjà vu: 'bar'!")
	}

	// remembers entry
	if !d.Witness([]byte("bar")) {
		t.Errorf("Expected déjà vu: 'bar'!")
	}

	// remembers oldest entry before overwriting
	if !d.Witness([]byte("foo")) {
		t.Errorf("Expected déjà vu: 'foo'!")
	}

	// add entries
	if d.Witness([]byte("bam")) {
		t.Errorf("Incorrect déjà vu: 'bam'!")
	}
	if d.Witness([]byte("baz")) {
		t.Errorf("Incorrect déjà vu: 'baz'!")
	}

	// forgot oldest
	if d.Witness([]byte("bar")) {
		t.Errorf("Incorrect déjà vu: 'bar'!")
	}
}

func TestProbabilistic(t *testing.T) {

	d := New(true, 1024, 0.000001)

	// add entries
	if d.Witness([]byte("foo")) {
		t.Errorf("Incorrect déjà vu: 'foo'!")
	}
	if d.Witness([]byte("bar")) {
		t.Errorf("Incorrect déjà vu: 'bar'!")
	}

	// remembers entry
	if !d.Witness([]byte("foo")) {
		t.Errorf("Expected déjà vu: 'foo'!")
	}

	// fill memory
	for i := 0; i < 2048; i++ {
		d.Witness([]byte("data"))
	}

	// forgot oldest
	if d.Witness([]byte("bar")) {
		t.Errorf("Incorrect déjà vu: 'bar'!")
	}
}

func TestProbabilisticLoad(t *testing.T) {
	d := NewProbabilistic(65536, 0.00000001)
	for i := 0; i < 65536; i++ {
		data := make([]byte, 20, 20)
		rand.Read(data)
		if d.Witness(data) {
			t.Errorf("Unexpected dejavu: %#X", data)
		}
	}
}

func TestProcess(t *testing.T) {
	// TODO compare know test input files to expected output file
	d := NewProbabilistic(65536, 0.00000001)
	ProcessPaths(d, true, "/dev/null", "LICENSE", "README.md")
}