Function TestInMemoryRefreshTokenStore_Cleanup

store/memory_test.go:191–250

(t *testing.T)

}

func TestInMemoryRefreshTokenStore_Cleanup(t *testing.T) {
	store := NewInMemoryRefreshTokenStore()

	// Add some tokens: 2 valid, 2 expired
	validExpiry := time.Now().Add(time.Hour)
	expiredExpiry := time.Now().Add(-time.Hour)

	err := store.Set(context.Background(), "valid1", &User{ID: "1"}, validExpiry)
	assert.NoError(t, err)
	err = store.Set(context.Background(), "valid2", &User{ID: "2"}, validExpiry)
	assert.NoError(t, err)
	err = store.Set(context.Background(), "expired1", &User{ID: "3"}, expiredExpiry)
	assert.NoError(t, err)
	err = store.Set(context.Background(), "expired2", &User{ID: "4"}, expiredExpiry)
	assert.NoError(t, err)

	// Verify initial count
	count, _ := store.Count(context.Background())
	if count != 4 {
		t.Fatalf("Expected initial count to be 4, got %d", count)
	}

	// Cleanup expired tokens
	cleaned, err := store.Cleanup(context.Background())
	if err != nil {
		t.Fatalf("Cleanup() returned error: %v", err)
	}

	if cleaned != 2 {
		t.Fatalf("Expected 2 tokens to be cleaned up, got %d", cleaned)
	}

	// Verify final count
	count, _ = store.Count(context.Background())
	if count != 2 {
		t.Fatalf("Expected final count to be 2, got %d", count)
	}

	// Verify valid tokens still exist
	_, err = store.Get(context.Background(), "valid1")
	if err != nil {
		t.Fatalf("valid1 token should still exist: %v", err)
	}

	_, err = store.Get(context.Background(), "valid2")
	if err != nil {
		t.Fatalf("valid2 token should still exist: %v", err)
	}

	// Verify expired tokens are gone
	_, err = store.Get(context.Background(), "expired1")
	if err != ErrRefreshTokenNotFound {
		t.Fatalf("expired1 token should be gone: %v", err)
	}

	_, err = store.Get(context.Background(), "expired2")
	if err != ErrRefreshTokenNotFound {
		t.Fatalf("expired2 token should be gone: %v", err)