drasl/session_test.go

package main

import (
	"encoding/json"
	"github.com/stretchr/testify/assert"
	"net/http"
	"testing"
)

func TestSession(t *testing.T) {
	t.Parallel()
	{
		ts := &TestSuite{}

		config := testConfig()
		ts.Setup(config)
		defer ts.Teardown()

		ts.CreateTestUser(t, ts.App, ts.Server, TEST_USERNAME)

		t.Run("Test /session/minecraft/hasJoined", ts.testSessionHasJoined)
		t.Run("Test /session/minecraft/join", ts.testSessionJoin)
		t.Run("Test /session/minecraft/profile/:id", ts.testSessionProfile)
		t.Run("Test /blockedservers", ts.testSessionBlockedServers)
	}
}

func (ts *TestSuite) testSessionJoin(t *testing.T) {
	// First, authenticate to get a token pair
	authenticateRes := ts.authenticate(t, TEST_PLAYER_NAME, TEST_PASSWORD)
	accessToken := authenticateRes.AccessToken

	var player Player
	result := ts.App.DB.First(&player, "name = ?", TEST_PLAYER_NAME)
	assert.Nil(t, result.Error)

	selectedProfile := Unwrap(UUIDToID(player.UUID))
	serverID := "0000000000000000000000000000000000000000"
	{
		// Successful join
		payload := sessionJoinRequest{
			AccessToken:     accessToken,
			SelectedProfile: selectedProfile,
			ServerID:        serverID,
		}
		rec := ts.PostJSON(t, ts.Server, "/session/minecraft/join", payload, nil, nil)
		assert.Equal(t, http.StatusNoContent, rec.Code)

		// Player ServerID should be set
		result = ts.App.DB.First(&player, "name = ?", TEST_PLAYER_NAME)
		assert.Nil(t, result.Error)

		assert.Equal(t, serverID, *UnmakeNullString(&player.ServerID))
	}
	{
		// Successful joinserver.jsp
		sessionID := "token:" + accessToken + ":" + selectedProfile
		rec := ts.Get(t, ts.Server, "/game/joinserver.jsp?user="+TEST_PLAYER_NAME+"&sessionId="+sessionID+"&serverId="+serverID, nil, nil)

		assert.Equal(t, http.StatusOK, rec.Code)
		assert.Equal(t, "OK", rec.Body.String())
	}
	{
		// Join should fail if we send an invalid access token

		// Start with an invalid ServerID
		assert.Nil(t, ts.App.DB.First(&player, "name = ?", TEST_PLAYER_NAME).Error)
		player.ServerID = MakeNullString(nil)
		assert.Nil(t, ts.App.DB.Save(&player).Error)

		payload := sessionJoinRequest{
			AccessToken:     "invalid",
			SelectedProfile: selectedProfile,
			ServerID:        serverID,
		}
		rec := ts.PostJSON(t, ts.Server, "/session/minecraft/join", payload, nil, nil)
		assert.Equal(t, http.StatusForbidden, rec.Code)

		var response ErrorResponse
		assert.Nil(t, json.NewDecoder(rec.Body).Decode(&response))
		assert.Equal(t, "ForbiddenOperationException", *response.Error)
		assert.Equal(t, "Invalid token.", *response.ErrorMessage)

		// Player ServerID should be invalid
		assert.Nil(t, ts.App.DB.First(&player, "name = ?", TEST_PLAYER_NAME).Error)
		assert.False(t, player.ServerID.Valid)
	}
	{
		// "Bad login" with invalid access token

		// Start with an invalid ServerID
		assert.Nil(t, ts.App.DB.First(&player, "name = ?", TEST_PLAYER_NAME).Error)
		player.ServerID = MakeNullString(nil)
		assert.Nil(t, ts.App.DB.Save(&player).Error)

		accessToken := "invalid"
		sessionID := "token:" + accessToken + ":" + selectedProfile
		rec := ts.Get(t, ts.Server, "/game/joinserver.jsp?user="+TEST_PLAYER_NAME+"&sessionId="+sessionID+"&serverId="+serverID, nil, nil)

		assert.Equal(t, http.StatusOK, rec.Code)
		assert.Equal(t, "Bad login", rec.Body.String())

		// Player ServerID should be invalid
		assert.Nil(t, ts.App.DB.First(&player, "name = ?", TEST_PLAYER_NAME).Error)
		assert.False(t, player.ServerID.Valid)
	}
}

func (ts *TestSuite) testSessionHasJoined(t *testing.T) {
	var player Player
	result := ts.App.DB.First(&player, "name = ?", TEST_PLAYER_NAME)
	assert.Nil(t, result.Error)

	serverID := "0000000000000000000000000000000000000000"
	// Start with a valid ServerID
	assert.Nil(t, ts.App.DB.First(&player, "name = ?", TEST_PLAYER_NAME).Error)
	player.ServerID = MakeNullString(&serverID)
	assert.Nil(t, ts.App.DB.Save(&player).Error)

	{
		// Successful hasJoined
		url := "/session/minecraft/hasJoined?username=" + player.Name + "&serverId=" + serverID + "&ip=" + "127.0.0.1"
		rec := ts.Get(t, ts.Server, url, nil, nil)
		assert.Equal(t, http.StatusOK, rec.Code)

		var response SessionProfileResponse
		assert.Nil(t, json.NewDecoder(rec.Body).Decode(&response))

		assert.Equal(t, Unwrap(UUIDToID(player.UUID)), response.ID)
		assert.Equal(t, player.Name, response.Name)
	}
	{
		// Successful checkserver.jsp

		url := "/game/checkserver.jsp?user=" + player.Name + "&serverId=" + serverID
		rec := ts.Get(t, ts.Server, url, nil, nil)
		assert.Equal(t, http.StatusOK, rec.Code)
		assert.Equal(t, "YES", rec.Body.String())
	}
	{
		// hasJoined should fail if we send an invalid server ID

		// Start with a valid ServerID
		assert.Nil(t, ts.App.DB.First(&player, "name = ?", TEST_PLAYER_NAME).Error)
		player.ServerID = MakeNullString(&serverID)
		assert.Nil(t, ts.App.DB.Save(&player).Error)

		url := "/session/minecraft/hasJoined?username=" + player.Name + "&serverId=" + "invalid" + "&ip=" + "127.0.0.1"
		rec := ts.Get(t, ts.Server, url, nil, nil)
		assert.Equal(t, http.StatusForbidden, rec.Code)
	}
	{
		// Unsuccessful checkserver.jsp

		invalidServerID := "INVALID-SERVER-ID"
		url := "/game/checkserver.jsp?user=" + player.Name + "&serverId=" + invalidServerID
		rec := ts.Get(t, ts.Server, url, nil, nil)
		assert.Equal(t, http.StatusMethodNotAllowed, rec.Code)
		assert.Equal(t, "NO", rec.Body.String())
	}
}

func (ts *TestSuite) testSessionProfile(t *testing.T) {
	var player Player
	result := ts.App.DB.First(&player, "name = ?", TEST_PLAYER_NAME)
	assert.Nil(t, result.Error)
	{
		// Successfully get profile

		url := "/session/minecraft/profile/" + Unwrap(UUIDToID(player.UUID))
		rec := ts.Get(t, ts.Server, url, nil, nil)
		assert.Equal(t, http.StatusOK, rec.Code)

		var response SessionProfileResponse
		assert.Nil(t, json.NewDecoder(rec.Body).Decode(&response))

		assert.Equal(t, Unwrap(UUIDToID(player.UUID)), response.ID)
		assert.Equal(t, player.Name, response.Name)
	}
	{
		// Successfully get profile with dashes in UUID

		url := "/session/minecraft/profile/" + player.UUID
		rec := ts.Get(t, ts.Server, url, nil, nil)
		assert.Equal(t, http.StatusOK, rec.Code)

		var response SessionProfileResponse
		assert.Nil(t, json.NewDecoder(rec.Body).Decode(&response))

		assert.Equal(t, Unwrap(UUIDToID(player.UUID)), response.ID)
		assert.Equal(t, player.Name, response.Name)
	}
	{
		// If the UUID doesn't exist, we should get a StatusNoContent
		url := "/session/minecraft/profile/" + "00000000000000000000000000000000"
		rec := ts.Get(t, ts.Server, url, nil, nil)
		assert.Equal(t, http.StatusNoContent, rec.Code)
	}
	{
		// If the UUID is invalid, we should get a StatusBadRequest with an error message
		url := "/session/minecraft/profile/" + "invalid"
		rec := ts.Get(t, ts.Server, url, nil, nil)
		assert.Equal(t, http.StatusBadRequest, rec.Code)

		var response ErrorResponse
		assert.Nil(t, json.NewDecoder(rec.Body).Decode(&response))
		assert.Equal(t, "Not a valid UUID: "+"invalid", *response.ErrorMessage)
	}
}

func (ts *TestSuite) testSessionBlockedServers(t *testing.T) {
	rec := ts.Get(t, ts.Server, "/blockedservers", nil, nil)
	assert.Equal(t, http.StatusOK, rec.Code)
}