User Structs

user.go

@@ -0,0 +1,38 @@
+package mapache
+
+import "time"
+
+type User struct {
+	ID        string    `json:"id" gorm:"primaryKey"`
+	FirstName string    `json:"first_name" gorm:"index"`
+	LastName  string    `json:"last_name"`
+	Email     string    `json:"email" gorm:"index,unique"`
+	Password  string    `json:"password"`
+	Subteam   string    `json:"subteam"`
+	UpdatedAt time.Time `json:"updated_at" gorm:"autoUpdateTime;precision:6"`
+	CreatedAt time.Time `json:"created_at" gorm:"autoCreateTime;precision:6"`
+	Roles     []string  `json:"roles" gorm:"-"`
+}
+
+func (User) TableName() string {
+	return "user"
+}
+
+func (u User) HasRole(role string) bool {
+	for _, r := range u.Roles {
+		if r == role {
+			return true
+		}
+	}
+	return false
+}
+
+type UserRole struct {
+	UserID    string    `json:"user_id" gorm:"primaryKey"`
+	Role      string    `json:"role" gorm:"primaryKey"`
+	CreatedAt time.Time `json:"time" gorm:"autoCreateTime;precision:6"`
+}
+
+func (UserRole) TableName() string {
+	return "user_role"
+}

user_test.go

@@ -0,0 +1,34 @@
+package mapache
+
+import "testing"
+
+func TestUser_TableName(t *testing.T) {
+	u := User{}
+	if u.TableName() != "user" {
+		t.Error("Expected user, got", u.TableName())
+	}
+}
+
+func TestUser_HasRole(t *testing.T) {
+	t.Run("Test HasRole", func(t *testing.T) {
+		u := User{
+			Roles: []string{"admin", "user"},
+		}
+		if !u.HasRole("admin") {
+			t.Error("Expected true, got", u.HasRole("admin"))
+		}
+		if !u.HasRole("user") {
+			t.Error("Expected true, got", u.HasRole("user"))
+		}
+		if u.HasRole("guest") {
+			t.Error("Expected false, got", u.HasRole("guest"))
+		}
+	})
+}
+
+func TestUserRole_TableName(t *testing.T) {
+	ur := UserRole{}
+	if ur.TableName() != "user_role" {
+		t.Error("Expected user_role, got", ur.TableName())
+	}
+}