Ilia Denisov
201 lines
7.1 KiB
Go
201 lines
7.1 KiB
Go
package controller_test
|
|
|
|
import (
|
|
"math"
|
|
"slices"
|
|
"testing"
|
|
|
|
e "github.com/iliadenisov/galaxy/internal/error"
|
|
"github.com/iliadenisov/galaxy/internal/model/game"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestJoinShipGroupToFleet(t *testing.T) {
|
|
c, g := newCache()
|
|
var groupIndex uint = 1
|
|
|
|
fleetOne := "R0_Fleet_one"
|
|
fleetTwo := "R0_Fleet_two"
|
|
|
|
assert.ErrorContains(t,
|
|
g.JoinShipGroupToFleet(Race_0.Name, BadEntityName, groupIndex, 0),
|
|
e.GenericErrorText(e.ErrInputEntityTypeNameInvalid))
|
|
|
|
assert.ErrorContains(t,
|
|
g.JoinShipGroupToFleet(Race_0.Name, "Unnamed", groupIndex, 0),
|
|
e.GenericErrorText(e.ErrInputEntityNotExists))
|
|
|
|
// creating ShipGroup
|
|
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 5))
|
|
|
|
assert.ErrorContains(t,
|
|
g.JoinShipGroupToFleet(UnknownRace, fleetOne, groupIndex, 0),
|
|
e.GenericErrorText(e.ErrInputUnknownRace))
|
|
assert.ErrorContains(t,
|
|
g.JoinShipGroupToFleet(Race_Extinct.Name, fleetOne, groupIndex, 0),
|
|
e.GenericErrorText(e.ErrRaceExinct))
|
|
|
|
assert.ErrorContains(t,
|
|
g.JoinShipGroupToFleet(Race_0.Name, "Unnamed", groupIndex, 6),
|
|
e.GenericErrorText(e.ErrJoinFleetGroupNumberNotEnough))
|
|
|
|
// ensure race has no Fleets
|
|
assert.Len(t, slices.Collect(c.ListFleets(Race_0_idx)), 0)
|
|
|
|
assert.NoError(t, g.JoinShipGroupToFleet(Race_0.Name, fleetOne, groupIndex, 0))
|
|
fleets := slices.Collect(c.ListFleets(Race_0_idx))
|
|
groups := slices.Collect(c.RaceShipGroups(Race_0_idx))
|
|
assert.Len(t, groups, 1)
|
|
gi := 0
|
|
assert.Len(t, fleets, 1)
|
|
assert.Equal(t, fleets[0].Name, fleetOne)
|
|
fleetState := c.FleetState(fleets[0].ID)
|
|
assert.Equal(t, game.StateInOrbit, fleetState.State)
|
|
|
|
assert.NotNil(t, groups[gi].FleetID)
|
|
assert.Equal(t, fleets[0].ID, *groups[gi].FleetID)
|
|
|
|
// create another ShipGroup
|
|
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 3))
|
|
groupIndex = 2
|
|
assert.NoError(t, g.JoinShipGroupToFleet(Race_0.Name, fleetTwo, groupIndex, 2))
|
|
fleets = slices.Collect(c.ListFleets(Race_0_idx))
|
|
groups = slices.Collect(c.RaceShipGroups(Race_0_idx))
|
|
assert.Len(t, groups, 3)
|
|
assert.Len(t, fleets, 2)
|
|
assert.Equal(t, fleets[1].Name, fleetTwo)
|
|
fleetState = c.FleetState(fleets[1].ID)
|
|
assert.Equal(t, game.StateInOrbit, fleetState.State)
|
|
|
|
gi = 2
|
|
assert.Len(t, groups, 3)
|
|
assert.NotNil(t, groups[gi].FleetID)
|
|
assert.Equal(t, fleets[1].ID, *groups[gi].FleetID)
|
|
assert.Equal(t, uint(2), groups[gi].Number)
|
|
assert.Equal(t, uint(3), groups[gi].Index)
|
|
|
|
gi = 1
|
|
assert.Nil(t, groups[gi].FleetID)
|
|
assert.Equal(t, uint(1), groups[gi].Number)
|
|
assert.Equal(t, uint(2), groups[gi].Index)
|
|
|
|
groupIndex = groups[gi].Index
|
|
assert.NoError(t, g.JoinShipGroupToFleet(Race_0.Name, fleetOne, groupIndex, 0))
|
|
fleets = slices.Collect(c.ListFleets(Race_0_idx))
|
|
assert.Len(t, fleets, 2)
|
|
groups = slices.Collect(c.RaceShipGroups(Race_0_idx))
|
|
assert.NotNil(t, groups[gi].FleetID)
|
|
assert.Equal(t, fleets[0].ID, *groups[gi].FleetID)
|
|
fleetState = c.FleetState(fleets[0].ID)
|
|
assert.Equal(t, game.StateInOrbit, fleetState.State)
|
|
|
|
// group not In_Orbit
|
|
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 7))
|
|
gi = 3
|
|
c.ShipGroup(gi).StateInSpace = &game.InSpace{
|
|
Origin: 2,
|
|
Range: 1,
|
|
}
|
|
assert.ErrorContains(t,
|
|
g.JoinShipGroupToFleet(Race_0.Name, fleetOne, c.ShipGroup(gi).Index, 0),
|
|
e.GenericErrorText(e.ErrShipsBusy))
|
|
c.ShipGroup(gi).StateInSpace = nil
|
|
|
|
// existing fleet not on the same planet or in_orbit
|
|
c.ShipGroup(0).StateInSpace = &game.InSpace{Origin: 2, Range: 1}
|
|
c.ShipGroup(1).StateInSpace = c.ShipGroup(0).StateInSpace
|
|
assert.ErrorContains(t,
|
|
g.JoinShipGroupToFleet(Race_0.Name, fleetOne, c.ShipGroup(gi).Index, 0),
|
|
e.GenericErrorText(e.ErrShipsNotOnSamePlanet))
|
|
}
|
|
|
|
func TestJoinFleets(t *testing.T) {
|
|
c, g := newCache()
|
|
// creating ShipGroup #1 at Planet_0
|
|
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 1)) // group #1
|
|
// creating ShipGroup #2 at Planet_2
|
|
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_2_num, 2)) // group #2
|
|
// creating ShipGroup #3 at Planet_0
|
|
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 3)) // group #3
|
|
|
|
// ensure race has no Fleets
|
|
assert.Len(t, slices.Collect(c.ListFleets(Race_0_idx)), 0)
|
|
|
|
fleetOnPlanet2 := "R0_Fleet_On_Planet_2"
|
|
fleetSourceOne := "R0_Fleet_one"
|
|
fleetTargetTwo := "R0_Fleet_two"
|
|
|
|
assert.ErrorContains(t,
|
|
g.JoinFleets(Race_0.Name, fleetSourceOne, fleetTargetTwo),
|
|
e.GenericErrorText(e.ErrInputEntityNotExists))
|
|
assert.ErrorContains(t,
|
|
g.JoinFleets(UnknownRace, fleetSourceOne, fleetTargetTwo),
|
|
e.GenericErrorText(e.ErrInputUnknownRace))
|
|
assert.ErrorContains(t,
|
|
g.JoinFleets(Race_Extinct.Name, fleetSourceOne, fleetTargetTwo),
|
|
e.GenericErrorText(e.ErrRaceExinct))
|
|
|
|
assert.ErrorContains(t,
|
|
g.JoinShipGroupToFleet(UnknownRace, fleetSourceOne, 1, 0),
|
|
e.GenericErrorText(e.ErrInputUnknownRace))
|
|
assert.ErrorContains(t,
|
|
g.JoinShipGroupToFleet(Race_Extinct.Name, fleetSourceOne, 1, 0),
|
|
e.GenericErrorText(e.ErrRaceExinct))
|
|
|
|
assert.NoError(t, g.JoinShipGroupToFleet(Race_0.Name, fleetSourceOne, 1, 0))
|
|
|
|
assert.ErrorContains(t,
|
|
g.JoinFleets(Race_0.Name, fleetSourceOne, fleetTargetTwo),
|
|
e.GenericErrorText(e.ErrInputEntityNotExists))
|
|
|
|
assert.NoError(t, g.JoinShipGroupToFleet(Race_0.Name, fleetTargetTwo, 3, 0))
|
|
assert.NoError(t, g.JoinFleets(Race_0.Name, fleetSourceOne, fleetTargetTwo))
|
|
assert.NoError(t, g.JoinShipGroupToFleet(Race_0.Name, fleetOnPlanet2, 2, 0))
|
|
|
|
assert.ErrorContains(t,
|
|
g.JoinFleets(Race_0.Name, fleetOnPlanet2, fleetTargetTwo),
|
|
e.GenericErrorText(e.ErrShipsNotOnSamePlanet))
|
|
}
|
|
|
|
func TestFleetSpeedAndMass(t *testing.T) {
|
|
c, g := newCache()
|
|
|
|
c.MustPlanet(R0_Planet_0_num).Material = 100.
|
|
c.MustPlanet(R0_Planet_0_num).Capital = 100.
|
|
fleet := "Fleet"
|
|
var speed, mass float64
|
|
|
|
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 3)) // 1
|
|
|
|
s := c.ShipGroup(0).Speed(c.MustShipClass(Race_0_idx, Race_0_Gunship))
|
|
m := c.ShipGroup(0).FullMass(c.MustShipClass(Race_0_idx, Race_0_Gunship))
|
|
|
|
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 5)) // 2
|
|
assert.NoError(t, g.LoadCargo(Race_0.Name, 2, "MAT", 10., 0))
|
|
|
|
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 7)) // 3
|
|
assert.NoError(t, g.LoadCargo(Race_0.Name, 3, "CAP", 10., 0))
|
|
|
|
assert.NoError(t, g.JoinShipGroupToFleet(Race_0.Name, fleet, 1, 0))
|
|
fleetIndex := 0
|
|
speed, mass = c.FleetSpeedAndMass(fleetIndex)
|
|
assert.Equal(t, s, speed)
|
|
assert.Equal(t, m, mass)
|
|
|
|
s = math.Min(s, c.ShipGroup(1).Speed(c.MustShipClass(Race_0_idx, Race_0_Freighter)))
|
|
m += c.ShipGroup(1).FullMass(c.MustShipClass(Race_0_idx, Race_0_Freighter))
|
|
|
|
assert.NoError(t, g.JoinShipGroupToFleet(Race_0.Name, fleet, 2, 0))
|
|
speed, mass = c.FleetSpeedAndMass(fleetIndex)
|
|
assert.Equal(t, s, speed)
|
|
assert.Equal(t, m, mass)
|
|
|
|
s = math.Min(s, c.ShipGroup(2).Speed(c.MustShipClass(Race_0_idx, Race_0_Freighter)))
|
|
m += c.ShipGroup(2).FullMass(c.MustShipClass(Race_0_idx, Race_0_Freighter))
|
|
|
|
assert.NoError(t, g.JoinShipGroupToFleet(Race_0.Name, fleet, 3, 0))
|
|
speed, mass = c.FleetSpeedAndMass(fleetIndex)
|
|
assert.Equal(t, s, speed)
|
|
assert.Equal(t, m, mass)
|
|
}
|