Skipping reservations in testing

bitmaelum · Mar 15, 2021 · be09f44 · be09f44
1 parent 376d50c
commit be09f44
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 6 deletions.
diff --git a/internal/handler/address_test.go b/internal/handler/address_test.go
@@ -29,6 +29,7 @@ import (
 	"github.com/bitmaelum/bitmaelum-suite/pkg/bmcrypto"
 	"github.com/bitmaelum/bitmaelum-suite/pkg/hash"
 	"github.com/bitmaelum/bitmaelum-suite/pkg/proofofwork"
+	"github.com/bitmaelum/key-resolver-go/internal"
 	"github.com/bitmaelum/key-resolver-go/internal/address"
 	"github.com/bitmaelum/key-resolver-go/internal/http"
 	"github.com/bitmaelum/key-resolver-go/internal/organisation"
@@ -713,6 +714,9 @@ func TestHistory(t *testing.T) {
 }
 
 func setupRepo() {
+	// NO reservation checks
+	internal.SkipReservationCheck = true
+
 	sr := address.NewSqliteResolver(":memory:")
 	address.SetDefaultRepository(sr)
 

diff --git a/internal/prometheus.go b/internal/prometheus.go
@@ -62,12 +62,12 @@ func ExportMetric() *events.APIGatewayV2HTTPResponse {
 	body += "# TYPE keyresolver_request counter\n"
 
 	input := &dynamodb.ScanInput{
-	    ExpressionAttributeNames: map[string]*string{
-	        "#path_code": aws.String("path_code"),
-	        "#hits": aws.String("hits"),
-	    },
-	    ProjectionExpression: aws.String("#path_code, #hits"),
-	    TableName:            aws.String("prometheus"),
+		ExpressionAttributeNames: map[string]*string{
+			"#path_code": aws.String("path_code"),
+			"#hits":      aws.String("hits"),
+		},
+		ProjectionExpression: aws.String("#path_code, #hits"),
+		TableName:            aws.String("prometheus"),
 	}
 
 	dyna := getDyna()

diff --git a/internal/reservation.go b/internal/reservation.go
@@ -34,6 +34,8 @@ import (
 // @TODO: Make this dynamic
 const baseReservedUrl = "https://resolver.bitmaelum.org/reserved/"
 
+var SkipReservationCheck = false
+
 func getDomainReservations(hash hash.Hash) ([]string, error) {
 	// call /reserved/<hash>
 	client := http.DefaultClient
@@ -70,6 +72,10 @@ func getDomainReservations(hash hash.Hash) ([]string, error) {
 }
 
 func CheckReservations(h hash.Hash, pk *bmcrypto.PubKey) bool {
+	if SkipReservationCheck {
+		return true
+	}
+
 	domains, err := getDomainReservations(h)
 	if err != nil {
 		// Error while fetching domains