Adding RateLimitSleep() to all tests

This commit is contained in:
Winni Neessen 2022-05-07 14:55:33 +02:00
parent 0ded55e697
commit af15c66865
Signed by: wneessen
GPG key ID: 5F3AF39B820C119D
3 changed files with 6 additions and 6 deletions

View file

@ -41,7 +41,7 @@ func TestBreachesWithDomain(t *testing.T) {
{"example.com is not breached", "example.com", false}, {"example.com is not breached", "example.com", false},
} }
hc := New() hc := New(WithRateLimitSleep())
for _, tc := range testTable { for _, tc := range testTable {
t.Run(tc.testName, func(t *testing.T) { t.Run(tc.testName, func(t *testing.T) {
breachList, _, err := hc.BreachApi.Breaches(WithDomain(tc.domain)) breachList, _, err := hc.BreachApi.Breaches(WithDomain(tc.domain))
@ -80,7 +80,7 @@ func TestBreachesWithoutUnverified(t *testing.T) {
{"xiaomi.cn is breached but not verified", "xiaomi.cn", true, false}, {"xiaomi.cn is breached but not verified", "xiaomi.cn", true, false},
} }
hc := New() hc := New(WithRateLimitSleep())
for _, tc := range testTable { for _, tc := range testTable {
t.Run(tc.testName, func(t *testing.T) { t.Run(tc.testName, func(t *testing.T) {
breachList, _, err := hc.BreachApi.Breaches(WithDomain(tc.domain), WithoutUnverified()) breachList, _, err := hc.BreachApi.Breaches(WithDomain(tc.domain), WithoutUnverified())
@ -108,7 +108,7 @@ func TestBreachByName(t *testing.T) {
{"Example is not a known breach", "Example", false, true}, {"Example is not a known breach", "Example", false, true},
} }
hc := New() hc := New(WithRateLimitSleep())
for _, tc := range testTable { for _, tc := range testTable {
t.Run(tc.testName, func(t *testing.T) { t.Run(tc.testName, func(t *testing.T) {
breachDetails, _, err := hc.BreachApi.BreachByName(tc.breachName) breachDetails, _, err := hc.BreachApi.BreachByName(tc.breachName)
@ -159,7 +159,7 @@ func TestBreachedAccount(t *testing.T) {
if apiKey == "" { if apiKey == "" {
t.SkipNow() t.SkipNow()
} }
hc := New(WithApiKey(apiKey)) hc := New(WithApiKey(apiKey), WithRateLimitSleep())
for _, tc := range testTable { for _, tc := range testTable {
t.Run(tc.testName, func(t *testing.T) { t.Run(tc.testName, func(t *testing.T) {
breachDetails, _, err := hc.BreachApi.BreachedAccount( breachDetails, _, err := hc.BreachApi.BreachedAccount(

View file

@ -44,7 +44,7 @@ func TestNewWithPwnedPadding(t *testing.T) {
// TestNewWithApiKey tests the New() function with the API key set // TestNewWithApiKey tests the New() function with the API key set
func TestNewWithApiKey(t *testing.T) { func TestNewWithApiKey(t *testing.T) {
apiKey := os.Getenv("HIBP_API_KEY") apiKey := os.Getenv("HIBP_API_KEY")
hc := New(WithApiKey(apiKey)) hc := New(WithApiKey(apiKey), WithRateLimitSleep())
if hc.ak != apiKey { if hc.ak != apiKey {
t.Errorf("hibp client API key was not set properly. Expected %s, got: %s", t.Errorf("hibp client API key was not set properly. Expected %s, got: %s",
apiKey, hc.ak) apiKey, hc.ak)

View file

@ -21,7 +21,7 @@ func TestPasteAccount(t *testing.T) {
if apiKey == "" { if apiKey == "" {
t.SkipNow() t.SkipNow()
} }
hc := New(WithApiKey(apiKey)) hc := New(WithApiKey(apiKey), WithRateLimitSleep())
for _, tc := range testTable { for _, tc := range testTable {
t.Run(tc.testName, func(t *testing.T) { t.Run(tc.testName, func(t *testing.T) {
pasteDetails, _, err := hc.PasteApi.PastedAccount( pasteDetails, _, err := hc.PasteApi.PastedAccount(