test validator

api/server.go

@@ -30,7 +30,7 @@ func NewServer(config util.Config, store db.Store) (*Server, error) {
 	server := &Server{store: store, tokenMaker: tokenMaker, config: config}
 
 	if v, ok := binding.Validator.Engine().(*validator.Validate); ok {
-		v.RegisterValidation("currency", validCurrency)
+		v.RegisterValidation("unit", validUnit)
 	}
 
 	server.setupRouter()

api/validator.go

@@ -6,7 +6,7 @@ import (
 	"github.com/go-playground/validator/v10"
 )
 
-var validCurrency validator.Func = func(fl validator.FieldLevel) bool {
+var validUnit validator.Func = func(fl validator.FieldLevel) bool {
 	if unit, ok := fl.Field().Interface().(string); ok {
 		// check if unit is supported
 		return util.IsSupportedUnit(unit)

api/validator_test.go

@@ -0,0 +1,32 @@
+package api
+
+import (
+	"testing"
+
+	"github.com/aradwann/eenergy/util"
+	"github.com/go-playground/validator/v10"
+	"github.com/stretchr/testify/require"
+)
+
+var unitItems = []struct {
+	have string
+	want bool
+}{
+	{util.KWH, true},
+	{"1234567890QWERTYUIOP", false},
+}
+
+// TestValidateUnit
+func TestValidateUnit(t *testing.T) {
+	validate := validator.New()
+	validate.RegisterValidation("unit", validUnit)
+
+	for _, item := range unitItems {
+		err := validate.Var(item.have, "unit")
+		if item.want {
+			require.Nil(t, err)
+		} else {
+			require.Error(t, err)
+		}
+	}
+}

db/migrations/000001_init_schema.up.sql

@@ -2,7 +2,7 @@ CREATE TABLE "accounts" (
   "id" bigserial PRIMARY KEY,
   "owner" varchar NOT NULL,
   "balance" bigint NOT NULL,
-  "currency" varchar NOT NULL,
+  "unit" varchar NOT NULL,
   "created_at" timestamptz NOT NULL DEFAULT (now())
 );
 

db/migrations/000002_add_users.down.sql

@@ -1,7 +1,7 @@
 ALTER TABLE
 IF EXISTS "accounts"
 DROP CONSTRAINT
-IF EXISTS "owner_currency_key";
+IF EXISTS "owner_unit_key";
 
 ALTER TABLE
 IF EXISTS "accounts"

db/migrations/000002_add_users.up.sql

@@ -9,5 +9,5 @@ CREATE TABLE "users" (
 
 ALTER TABLE "accounts" ADD FOREIGN KEY ("owner") REFERENCES "users" ("username");
 
--- CREATE UNIQUE INDEX ON "accounts" ("owner", "currency");
-ALTER TABLE "accounts" ADD CONSTRAINT "owner_currency_key" UNIQUE ("owner", "currency");
+-- CREATE UNIQUE INDEX ON "accounts" ("owner", "unit");
+ALTER TABLE "accounts" ADD CONSTRAINT "owner_unit_key" UNIQUE ("owner", "unit");

token/paseto_maker_test.go

@@ -50,3 +50,11 @@ func TestExpiredPASETOToken(t *testing.T) {
 	require.Nil(t, payload)
 
 }
+
+func TestPASETOMakerInvalidKeySize(t *testing.T) {
+	symmetricKey := "shortKey" // Invalid key size
+
+	_, err := NewPASETOMaker(symmetricKey)
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "invalid key size")
+}