PSF-LoginServer/src/test/scala/objects/terminal/OrderTerminalTest.scala

// Copyright (c) 2017 PSForever
package objects.terminal

import net.psforever.objects.serverobject.structures.{Building, StructureType}
import net.psforever.objects.serverobject.terminals.Terminal
import net.psforever.objects.zones.Zone
import net.psforever.objects._
import net.psforever.objects.avatar.{Avatar, Certification}
import net.psforever.packet.game.ItemTransactionMessage
import net.psforever.types._
import org.specs2.mutable.Specification

class OrderTerminalTest extends Specification {
  val avatar = Avatar(0, "test", PlanetSideEmpire.TR, CharacterGender.Male, 0, CharacterVoice.Mute)
  val player = Player(avatar)

  val building = new Building(
    "Building",
    building_guid = 0,
    map_id = 0,
    Zone.Nowhere,
    StructureType.Building,
    GlobalDefinitions.building
  )
  building.Faction = PlanetSideEmpire.TR
  val infantryTerminal = Terminal(GlobalDefinitions.order_terminal)
  infantryTerminal.Owner = building

  "General terminal behavior" should {
    "player can not buy equipment from the wrong page ('9mmbullet_AP', page 10)" in {
      val msg = ItemTransactionMessage(PlanetSideGUID(1), TransactionType.Buy, 1, "9mmbullet_AP", 0, PlanetSideGUID(0))

      infantryTerminal.Request(player, msg) mustEqual Terminal.NoDeal()
    }
  }

  "Infantry Order Terminal" should {
    "player can buy a box of ammunition ('9mmbullet_AP')" in {
      val msg   = ItemTransactionMessage(PlanetSideGUID(1), TransactionType.Buy, 0, "9mmbullet_AP", 0, PlanetSideGUID(0))
      val reply = infantryTerminal.Request(player, msg)
      reply.isInstanceOf[Terminal.BuyEquipment] mustEqual true
      val reply2 = reply.asInstanceOf[Terminal.BuyEquipment]
      reply2.item.isInstanceOf[AmmoBox] mustEqual true
      reply2.item.asInstanceOf[AmmoBox].Definition mustEqual GlobalDefinitions.bullet_9mm_AP
    }

    "player can buy a weapon ('suppressor')" in {
      val msg   = ItemTransactionMessage(PlanetSideGUID(1), TransactionType.Buy, 0, "suppressor", 0, PlanetSideGUID(0))
      val reply = infantryTerminal.Request(player, msg)
      reply.isInstanceOf[Terminal.BuyEquipment] mustEqual true
      val reply2 = reply.asInstanceOf[Terminal.BuyEquipment]
      reply2.item.isInstanceOf[Tool] mustEqual true
      reply2.item.asInstanceOf[Tool].Definition mustEqual GlobalDefinitions.suppressor
    }

    "player can buy different armor ('lite_armor')" in {
      val msg = ItemTransactionMessage(PlanetSideGUID(1), TransactionType.Buy, 1, "lite_armor", 0, PlanetSideGUID(0))

      infantryTerminal.Request(player, msg) mustEqual Terminal.BuyExosuit(ExoSuitType.Agile)
    }

    "player can buy a box of ammunition belonging to a special armor type ('dualcycler_ammo')" in {
      val msg =
        ItemTransactionMessage(PlanetSideGUID(1), TransactionType.Buy, 1, "dualcycler_ammo", 0, PlanetSideGUID(0))
      val reply = infantryTerminal.Request(player, msg)
      reply.isInstanceOf[Terminal.BuyEquipment] mustEqual true
      val reply2 = reply.asInstanceOf[Terminal.BuyEquipment]
      reply2.item.isInstanceOf[AmmoBox] mustEqual true
      reply2.item.asInstanceOf[AmmoBox].Definition mustEqual GlobalDefinitions.dualcycler_ammo
    }

    "player can buy a support tool ('bank')" in {
      val msg   = ItemTransactionMessage(PlanetSideGUID(1), TransactionType.Buy, 2, "bank", 0, PlanetSideGUID(0))
      val reply = infantryTerminal.Request(player, msg)
      reply.isInstanceOf[Terminal.BuyEquipment] mustEqual true
      val reply2 = reply.asInstanceOf[Terminal.BuyEquipment]
      reply2.item.isInstanceOf[Tool] mustEqual true
      reply2.item.asInstanceOf[Tool].Definition mustEqual GlobalDefinitions.bank
    }

    "player can buy a box of vehicle ammunition ('105mmbullet')" in {
      val msg   = ItemTransactionMessage(PlanetSideGUID(1), TransactionType.Buy, 3, "105mmbullet", 0, PlanetSideGUID(0))
      val reply = infantryTerminal.Request(player, msg)
      reply.isInstanceOf[Terminal.BuyEquipment] mustEqual true
      val reply2 = reply.asInstanceOf[Terminal.BuyEquipment]
      reply2.item.isInstanceOf[AmmoBox] mustEqual true
      reply2.item.asInstanceOf[AmmoBox].Definition mustEqual GlobalDefinitions.bullet_105mm
    }

    "player can not buy fake equipment ('sabot')" in {
      val msg = ItemTransactionMessage(PlanetSideGUID(1), TransactionType.Buy, 0, "sabot", 0, PlanetSideGUID(0))
      infantryTerminal.Request(player, msg) mustEqual Terminal.NoDeal()
    }

    "player can not retrieve an infantry loadout from the wrong line" in {
      val msg = infantryTerminal.Request(
        player,
        ItemTransactionMessage(PlanetSideGUID(10), TransactionType.Loadout, 4, "", 1, PlanetSideGUID(0))
      )
      msg.isInstanceOf[Terminal.NoDeal] mustEqual true
    }
  }

  "Vehicle Terminal" should {
    val terminal = Terminal(GlobalDefinitions.ground_vehicle_terminal)
    terminal.Owner = building

    "player can spawn a vehicle and its default trunk contents" in {
      val msg =
        ItemTransactionMessage(PlanetSideGUID(1), TransactionType.Buy, 46769, "quadassault", 0, PlanetSideGUID(0))
      terminal.Request(player, msg) match {
        case Terminal.BuyVehicle(vehicle, weapons, trunk) =>
          vehicle.Definition mustEqual GlobalDefinitions.quadassault

          weapons.size mustEqual 0 //note: vehicles never have custom weapons using the default loadout

          trunk.size mustEqual 4
          trunk.head.obj.Definition mustEqual GlobalDefinitions.bullet_12mm
          trunk(1).obj.Definition mustEqual GlobalDefinitions.bullet_12mm
          trunk(2).obj.Definition mustEqual GlobalDefinitions.bullet_12mm
          trunk(3).obj.Definition mustEqual GlobalDefinitions.bullet_12mm
        case _ =>
          ko
      }
    }

    "player can not spawn a fake vehicle ('harasser')" in {
      val msg = ItemTransactionMessage(PlanetSideGUID(1), TransactionType.Buy, 46769, "harasser", 0, PlanetSideGUID(0))
      terminal.Request(player, msg) mustEqual Terminal.NoDeal()
    }
  }

  "Certification Terminal" should {
    val terminal = Terminal(GlobalDefinitions.cert_terminal)
    terminal.Owner = building

    "player can learn a certification ('medium_assault')" in {
      val msg =
        ItemTransactionMessage(PlanetSideGUID(1), TransactionType.Learn, 0, "medium_assault", 0, PlanetSideGUID(0))
      terminal.Request(player, msg) mustEqual Terminal.LearnCertification(Certification.MediumAssault)
    }

    "player can not learn a fake certification ('juggling')" in {
      val msg = ItemTransactionMessage(PlanetSideGUID(1), TransactionType.Learn, 0, "juggling", 0, PlanetSideGUID(0))
      terminal.Request(player, msg) mustEqual Terminal.NoDeal()
    }

    "player can forget a certification ('medium_assault')" in {
      val msg =
        ItemTransactionMessage(PlanetSideGUID(1), TransactionType.Sell, 0, "medium_assault", 0, PlanetSideGUID(0))
      terminal.Request(player, msg) mustEqual Terminal.SellCertification(Certification.MediumAssault)
    }

    "player can not forget a fake certification ('juggling')" in {
      val msg = ItemTransactionMessage(PlanetSideGUID(1), TransactionType.Sell, 0, "juggling", 0, PlanetSideGUID(0))
      terminal.Request(player, msg) mustEqual Terminal.NoDeal()
    }
  }

  "Implant_Terminal_Interface" should {
    val terminal = Terminal(GlobalDefinitions.implant_terminal_interface)
    terminal.Owner = building

    "player can learn an implant ('darklight_vision')" in {
      val msg =
        ItemTransactionMessage(PlanetSideGUID(1), TransactionType.Buy, 0, "darklight_vision", 0, PlanetSideGUID(0))

      val reply = terminal.Request(player, msg)
      reply.isInstanceOf[Terminal.LearnImplant] mustEqual true
      val reply2 = reply.asInstanceOf[Terminal.LearnImplant]
      reply2.implant mustEqual GlobalDefinitions.darklight_vision
    }

    "player can not learn a fake implant ('aimbot')" in {
      val msg = ItemTransactionMessage(PlanetSideGUID(1), TransactionType.Buy, 0, "aimbot", 0, PlanetSideGUID(0))

      terminal.Request(player, msg) mustEqual Terminal.NoDeal()
    }

    "player can un-learn an implant ('darklight_vision')" in {
      val msg =
        ItemTransactionMessage(PlanetSideGUID(1), TransactionType.Sell, 0, "darklight_vision", 0, PlanetSideGUID(0))

      val reply = terminal.Request(player, msg)
      reply.isInstanceOf[Terminal.SellImplant] mustEqual true
      val reply2 = reply.asInstanceOf[Terminal.SellImplant]
      reply2.implant mustEqual GlobalDefinitions.darklight_vision
    }

    "player can not un-learn a fake implant ('aimbot')" in {
      val terminal = Terminal(GlobalDefinitions.implant_terminal_interface)
      val msg      = ItemTransactionMessage(PlanetSideGUID(1), TransactionType.Sell, 0, "aimbot", 0, PlanetSideGUID(0))

      terminal.Request(player, msg) mustEqual Terminal.NoDeal()
    }
  }
}