mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-17 22:15:27 +00:00
31 lines
915 B
Go
31 lines
915 B
Go
package hosterrorscache
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCacheCheckMarkFailed(t *testing.T) {
|
|
cache := New(3, DefaultMaxHostsCount)
|
|
|
|
cache.MarkFailed("http://example.com:80")
|
|
if value, err := cache.failedTargets.Get("http://example.com:80"); err == nil && value != nil {
|
|
require.Equal(t, 1, value, "could not get correct markfailed")
|
|
}
|
|
cache.MarkFailed("example.com:80")
|
|
if value, err := cache.failedTargets.Get("example.com:80"); err == nil && value != nil {
|
|
require.Equal(t, 2, value, "could not get correct markfailed")
|
|
}
|
|
cache.MarkFailed("example.com")
|
|
if value, err := cache.failedTargets.Get("example.com"); err == nil && value != nil {
|
|
require.Equal(t, 1, value, "could not get correct markfailed")
|
|
}
|
|
for i := 0; i < 3; i++ {
|
|
cache.MarkFailed("test")
|
|
}
|
|
|
|
value := cache.Check("test")
|
|
require.Equal(t, true, value, "could not get checked value")
|
|
}
|