Ilia Denisov
124 lines
3.9 KiB
Go
124 lines
3.9 KiB
Go
package game_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/iliadenisov/galaxy/internal/model/game"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
var (
|
|
Cruiser = game.ShipType{
|
|
ShipTypeReport: game.ShipTypeReport{
|
|
Name: "Cruiser",
|
|
Drive: 15,
|
|
Armament: 1,
|
|
Weapons: 15,
|
|
Shields: 15,
|
|
Cargo: 0,
|
|
},
|
|
}
|
|
)
|
|
|
|
func TestBlockUpgradeCost(t *testing.T) {
|
|
assert.Equal(t, 00.0, game.BlockUpgradeCost(1, 1.0, 1.0))
|
|
assert.Equal(t, 25.0, game.BlockUpgradeCost(5, 1.0, 2.0))
|
|
assert.Equal(t, 50.0, game.BlockUpgradeCost(10, 1.0, 2.0))
|
|
}
|
|
|
|
func TestGroupUpgradeCost(t *testing.T) {
|
|
sg := game.ShipGroup{
|
|
Tech: map[game.Tech]float64{
|
|
game.TechDrive: 1.0,
|
|
game.TechWeapons: 1.0,
|
|
game.TechShields: 1.0,
|
|
game.TechCargo: 1.0,
|
|
},
|
|
Number: 1,
|
|
}
|
|
assert.Equal(t, 225.0, game.GroupUpgradeCost(sg, Cruiser, 2.0, 2.0, 2.0, 2.0).UpgradeCost(1))
|
|
}
|
|
|
|
func TestUpgradeMaxShips(t *testing.T) {
|
|
sg := game.ShipGroup{
|
|
Tech: map[game.Tech]float64{
|
|
game.TechDrive: 1.0,
|
|
game.TechWeapons: 1.0,
|
|
game.TechShields: 1.0,
|
|
game.TechCargo: 1.0,
|
|
},
|
|
Number: 10,
|
|
}
|
|
uc := game.GroupUpgradeCost(sg, Cruiser, 2.0, 2.0, 2.0, 2.0)
|
|
assert.Equal(t, uint(4), uc.UpgradeMaxShips(1000))
|
|
}
|
|
|
|
func TestCurrentUpgradingLevel(t *testing.T) {
|
|
sg := &game.ShipGroup{
|
|
StateUpgrade: nil,
|
|
}
|
|
assert.Equal(t, 0.0, game.CurrentUpgradingLevel(*sg, game.TechDrive))
|
|
assert.Equal(t, 0.0, game.CurrentUpgradingLevel(*sg, game.TechWeapons))
|
|
assert.Equal(t, 0.0, game.CurrentUpgradingLevel(*sg, game.TechShields))
|
|
assert.Equal(t, 0.0, game.CurrentUpgradingLevel(*sg, game.TechCargo))
|
|
|
|
sg.StateUpgrade = &game.InUpgrade{
|
|
UpgradeTech: []game.UpgradePreference{
|
|
{Tech: game.TechDrive, Level: 1.5, Cost: 100.1},
|
|
},
|
|
}
|
|
assert.Equal(t, 1.5, game.CurrentUpgradingLevel(*sg, game.TechDrive))
|
|
assert.Equal(t, 0.0, game.CurrentUpgradingLevel(*sg, game.TechWeapons))
|
|
assert.Equal(t, 0.0, game.CurrentUpgradingLevel(*sg, game.TechShields))
|
|
assert.Equal(t, 0.0, game.CurrentUpgradingLevel(*sg, game.TechCargo))
|
|
|
|
sg.StateUpgrade.UpgradeTech = append(sg.StateUpgrade.UpgradeTech, game.UpgradePreference{Tech: game.TechCargo, Level: 2.2, Cost: 200.2})
|
|
assert.Equal(t, 1.5, game.CurrentUpgradingLevel(*sg, game.TechDrive))
|
|
assert.Equal(t, 0.0, game.CurrentUpgradingLevel(*sg, game.TechWeapons))
|
|
assert.Equal(t, 0.0, game.CurrentUpgradingLevel(*sg, game.TechShields))
|
|
assert.Equal(t, 2.2, game.CurrentUpgradingLevel(*sg, game.TechCargo))
|
|
}
|
|
|
|
func TestFutureUpgradeLevel(t *testing.T) {
|
|
assert.Equal(t, 0.0, game.FutureUpgradeLevel(2.0, 2.0, 2.0))
|
|
assert.Equal(t, 0.0, game.FutureUpgradeLevel(2.0, 2.0, 3.0))
|
|
assert.Equal(t, 1.5, game.FutureUpgradeLevel(1.5, 2.0, 3.0))
|
|
assert.Equal(t, 2.0, game.FutureUpgradeLevel(2.5, 1.0, 2.0))
|
|
assert.Equal(t, 2.5, game.FutureUpgradeLevel(2.5, 1.0, 0.0))
|
|
}
|
|
|
|
func TestUpgradeGroupPreference(t *testing.T) {
|
|
sg := game.ShipGroup{
|
|
Number: 4,
|
|
Tech: game.TechSet{
|
|
game.TechDrive: 1.0,
|
|
game.TechWeapons: 1.0,
|
|
game.TechShields: 1.0,
|
|
game.TechCargo: 1.0,
|
|
},
|
|
}
|
|
assert.Nil(t, sg.StateUpgrade)
|
|
sg = game.UpgradeGroupPreference(sg, Cruiser, game.TechDrive, 0)
|
|
assert.Nil(t, sg.StateUpgrade)
|
|
|
|
sg = game.UpgradeGroupPreference(sg, Cruiser, game.TechDrive, 2.0)
|
|
assert.NotNil(t, sg.StateUpgrade)
|
|
assert.Equal(t, 300., sg.StateUpgrade.TechCost(game.TechDrive))
|
|
assert.Equal(t, 300., sg.StateUpgrade.Cost())
|
|
|
|
sg = game.UpgradeGroupPreference(sg, Cruiser, game.TechWeapons, 2.0)
|
|
assert.NotNil(t, sg.StateUpgrade)
|
|
assert.Equal(t, 300., sg.StateUpgrade.TechCost(game.TechWeapons))
|
|
assert.Equal(t, 600., sg.StateUpgrade.Cost())
|
|
|
|
sg = game.UpgradeGroupPreference(sg, Cruiser, game.TechShields, 2.0)
|
|
assert.NotNil(t, sg.StateUpgrade)
|
|
assert.Equal(t, 300., sg.StateUpgrade.TechCost(game.TechShields))
|
|
assert.Equal(t, 900., sg.StateUpgrade.Cost())
|
|
|
|
sg = game.UpgradeGroupPreference(sg, Cruiser, game.TechCargo, 2.0)
|
|
assert.NotNil(t, sg.StateUpgrade)
|
|
assert.Equal(t, 0., sg.StateUpgrade.TechCost(game.TechCargo))
|
|
assert.Equal(t, 900., sg.StateUpgrade.Cost())
|
|
}
|