Use custom implementation

Author: Widcket

management/anomaly.go

@@ -1,5 +1,9 @@
 package management
 
+import (
+	"net/http"
+)
+
 type AnomalyManager struct {
 	*Management
 }
@@ -12,16 +16,34 @@ func newAnomalyManager(m *Management) *AnomalyManager {
 // trigger due to multiple failed logins.
 //
 // See: https://auth0.com/docs/api/management/v2#!/Anomaly/get_ips_by_id
-func (m *AnomalyManager) CheckIP(ip string, opts ...RequestOption) (err error) {
-	err = m.Request("GET",  m.URI("anomaly", "blocks", "ips", ip), nil, opts...)
-	return
+func (m *AnomalyManager) CheckIP(ip string, opts ...RequestOption) (isBlocked bool, err error) {
+	req, err := m.NewRequest("GET", m.URI("anomaly", "blocks", "ips", ip), nil, opts...)
+	if err != nil {
+		return false, err
+	}
+
+	res, err := m.Do(req)
+	if err != nil {
+		return false, err
+	}
+
+	// 200: IP address specified is currently blocked.
+	if res.StatusCode == http.StatusOK {
+		return true, nil
+	}
+
+	// 404: IP address specified is not currently blocked.
+	if res.StatusCode == http.StatusNotFound {
+		return false, nil
+	}
+
+	return false, newError(res.Body)
 }
 
 // Unblock an IP address currently blocked by the multiple user accounts
 // trigger due to multiple failed logins.
 //
 // See: https://auth0.com/docs/api/management/v2#!/Anomaly/delete_ips_by_id
 func (m *AnomalyManager) UnblockIP(ip string, opts ...RequestOption) (err error) {
-	err = m.Request("DELETE", m.URI("anomaly", "blocks", "ips", ip), nil, opts...)
-	return
+	return m.Request("DELETE", m.URI("anomaly", "blocks", "ips", ip), nil, opts...)
 }

management/anomaly_test.go

@@ -1,13 +1,24 @@
 package management
 
 import (
+	"fmt"
 	"testing"
 )
 
 func TestAnomaly(t *testing.T) {
 
 	t.Run("CheckIP", func(t *testing.T) {
-		err := m.Anomaly.CheckIP("1.1.1.1")
+		isBlocked, err := m.Anomaly.CheckIP("1.1.1.1")
+		if err != nil {
+			t.Error(err)
+		}
+		if isBlocked {
+			t.Error(fmt.Errorf("Ip should not be blocked"))
+		}
+	})
+
+	t.Run("UnblockIP", func(t *testing.T) {
+		err := m.Anomaly.UnblockIP("1.1.1.1")
 		if err != nil {
 			t.Error(err)
 		}

management/management.go

@@ -156,6 +156,9 @@ type Management struct {
 	// SigningKey manages Auth0 Application Signing Keys.
 	SigningKey *SigningKeyManager
 
+	// Anomaly manages the IP blocks
+	Anomaly *AnomalyManager
+
 	url         *url.URL
 	basePath    string
 	userAgent   string
@@ -223,6 +226,7 @@ func New(domain string, options ...ManagementOption) (*Management, error) {
 	m.Prompt = newPromptManager(m)
 	m.Blacklist = newBlacklistManager(m)
 	m.SigningKey = newSigningKeyManager(m)
+	m.Anomaly = newAnomalyManager(m)
 
 	return m, nil
 }