test_example

test_example.go

package main

import (
	"fmt"
	"errors"
	"testing"
  "github.com/stretchr/testify/assert"
)

type User struct {
	Id int64
	Email string
	Role string
}

type UserServiceInter interface {
	GetUserInfoById(userId int64) (User, error)
}

type UserServiceImpl struct {}

func (us *UserServiceImpl) GetUserInfoById(userId int64) (User, error) {
	var user User 
	// Get user from db or other services...
	// Get user from db or other services...
	return user, nil // user or error
}

type UserServiceMock struct {}

func (us *UserServiceMock) GetUserInfoById(userId int64) (User, error) {
	if userId <= 0 {
		return nil, errors.New("invalid user id")
	}

	if userId == 1 {
		return User{
			Id: userId,
			Email: "[email protected]",
			Role: "guest",
		}, nil
	}

	return User{
		Id: userId,
		Email: "[email protected]",
		Role: "admin",
	}, nil
}

func isAdminUser(userService UserServiceInter, userId int64) bool {
	user, err := userService.GetUserInfoById(userId)
	if err != nil {
		return false
	}
	return user.Role == 'admin'
}

// TEST
func isAdminUser_Test(t *testing.T) {
	var userServiceMock = UserServiceMock{}
	var userId
	var isAdmin

	// Test case 1
	userId = -1
	isAdmin = isAdminUser(userServiceMock, userId)
	assert.Equal(t, false, isAdmin, "Failed")

	// Test case 1
	userId = 0
	isAdmin = isAdminUser(userServiceMock, userId)
	assert.Equal(t, false, isAdmin, "Failed")

	// Test case 1
	userId = 1
	isAdmin = isAdminUser(userServiceMock, 1)
	assert.Equal(t, false, isAdmin, "Failed")

	// Test case 1
	userId = 3
	isAdmin = isAdminUser(userServiceMock, userId)
	assert.Equal(t, true, isAdmin, "Failed")
}


// Main function
func main() {
	// Do something...
}