Esempio n. 1
0
func TestSessionLifetime(t *testing.T) {
	store := memstore.New(time.Millisecond*10, false)
	ts := NewSessionCache(store, TokenSalt)

	t1 := ts.Add()
	t2 := ts.Add()

	if err := ts.Get(t1, nil); err != nil {
		t.Error("The session t1 was not stored")
	}
	if err := ts.Get(t2, nil); err != nil {
		t.Error("The session t2 was not stored")
	}

	time.Sleep(time.Millisecond * 20)

	if err := ts.Get(t1, nil); err == nil {
		t.Error("The session t1 was not expired")
	}
	if err := ts.Get(t2, nil); err == nil {
		t.Error("The session t2 was not expired")
	}

	if err := ts.Delete(t1); err == nil {
		t.Error("The expired session t1 should not be removable")
	}
	if err := ts.Set(t2, nil); err == nil {
		t.Error("The expired session t2 should not be changeable")
	}
}
Esempio n. 2
0
func BenchmarkSessionCreationFast(b *testing.B) {
	store := memstore.New(time.Millisecond, false)
	ts := NewSessionCacheFast(store, TokenSalt)
	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		ts.Add()
	}
}
Esempio n. 3
0
func BenchmarkSessionCreationFast(b *testing.B) {
	store := memstore.New(time.Millisecond, false)
	ts := NewSessionStore().
		SalterFast([]byte(TokenSalt)).
		Store(store).
		Build()
	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		ts.Add(nil)
	}
}
Esempio n. 4
0
func TestSessionLifetime(t *testing.T) {
	store := memstore.New(time.Millisecond*10, false)
	ts := NewSessionStore().
		SalterFast([]byte(TokenSalt)).
		Store(store).
		Build()

	t1, err := ts.Add(nil)
	if err != nil {
		t.Errorf("The session t1 could not be generated: %v", err)
	}
	t2, err := ts.Add(nil)
	if err != nil {
		t.Errorf("The session t2 could not be generated: %v", err)
	}

	if err := ts.Get(t1, nil); err != nil {
		t.Error("The session t1 was not stored")
	}
	if err := ts.Get(t2, nil); err != nil {
		t.Error("The session t2 was not stored")
	}

	time.Sleep(time.Millisecond * 20)

	if err := ts.Get(t1, nil); err == nil {
		t.Error("The session t1 was not expired")
	}
	if err := ts.Get(t2, nil); err == nil {
		t.Error("The session t2 was not expired")
	}

	if err := ts.Delete(t1); err == nil {
		t.Error("The expired session t1 should not be removable")
	}
	if err := ts.Set(t2, nil); err == nil {
		t.Error("The expired session t2 should not be changeable")
	}
}
Esempio n. 5
0
func TestSessionHandling(t *testing.T) {
	testValues := []struct {
		ref   string
		token string
		value int
	}{
		{"t1", "", 3},
		{"t2", "", 6},
		{"t3", "", 83679},
		{"t4", "", 2748},
		{"t5", "", 54},
		{"t6", "", 6},
		{"t7", "", 2},
		{"t8", "", 8},
		{"t9", "", 7},
		{"t10", "", 8},
	}
	rmTestIndex := 6
	changeValues := map[int]int{
		2: 5062,
		9: 4099,
	}

	store := memstore.New(time.Millisecond*100, false)
	ts := NewSessionCache(store, TokenSalt)
	if _, err := ts.Count(); err != nil {
		t.Fatal("The Count() method should be supported by MemStore")
		return
	}

	if count, _ := ts.Count(); count != 0 {
		t.Errorf(
			"The session cache should be empty, but it has %d items",
			count)
	}

	lastCount, _ := ts.Count()
	for i := range testValues {
		item := &testValues[i]
		item.token = ts.Add()

		if count, _ := ts.Count(); count != lastCount+1 {
			t.Errorf(
				"The new session '%s' was not stored into session cache",
				item.token)
		}
		lastCount, _ = ts.Count()

		err := ts.Set(item.token, item.value)
		if err != nil {
			t.Errorf("The session %s could not be set", item.ref)
		}
	}

	if count, _ := ts.Count(); count != len(testValues) {
		t.Errorf("The session count do not match (%d!=%d)",
			count, len(testValues))
	}

	for _, i := range testValues {
		var v int
		if err := ts.Get(i.token, &v); err != nil {
			t.Errorf("The session %s could not be read", i.ref)
		}
		if v != i.value {
			t.Errorf("The session %s was stored incorrectly", i.ref)
		}
	}

	rmTestKey := testValues[rmTestIndex]
	if err := ts.Delete(rmTestKey.token); err != nil {
		t.Errorf("The session %s could not be removed", rmTestKey.ref)
	}
	if err := ts.Get(rmTestKey.token, nil); err == nil {
		t.Errorf("The removed session %s should not be retrieved", rmTestKey.ref)
	}
	if count, _ := ts.Count(); count == len(testValues) {
		t.Error("The session count should not match")
	}

	for k, v := range changeValues {
		item := testValues[k]
		err := ts.Set(item.token, v)
		if err != nil {
			t.Errorf("The session %s could not be changed", item.ref)
		}
	}
	for k, v := range changeValues {
		item := testValues[k]
		var v2 int
		if err := ts.Get(item.token, &v2); err != nil {
			t.Errorf("The session %s could not be read", item.ref)
		}
		if v2 != v {
			t.Errorf("The session %s was not changed", item.ref)
		}
	}
}