mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-24 15:10:19 +00:00
468 lines
13 KiB
Go
468 lines
13 KiB
Go
package portfolio
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/thrasher-corp/gocryptotrader/currency"
|
|
)
|
|
|
|
func TestGetEthereumBalance(t *testing.T) {
|
|
address := "0xb794f5ea0ba39494ce839613fffba74279579268"
|
|
nonsenseAddress := "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA"
|
|
|
|
response, err := GetEthereumBalance(address)
|
|
if err != nil {
|
|
t.Errorf("Portfolio GetEthereumBalance() Error: %s", err)
|
|
}
|
|
|
|
if response.Address != "0xb794f5ea0ba39494ce839613fffba74279579268" {
|
|
t.Error("Portfolio GetEthereumBalance() address invalid")
|
|
}
|
|
|
|
response, err = GetEthereumBalance(nonsenseAddress)
|
|
if response.Error.Message != "" || err == nil {
|
|
t.Errorf("Portfolio GetEthereumBalance() Error: %s",
|
|
response.Error.Message)
|
|
}
|
|
}
|
|
|
|
func TestGetCryptoIDBalance(t *testing.T) {
|
|
ltcAddress := "LX2LMYXtuv5tiYEMztSSoEZcafFPYJFRK1"
|
|
_, err := GetCryptoIDAddress(ltcAddress, currency.LTC)
|
|
if err != nil {
|
|
t.Fatalf("TestGetCryptoIDBalance error: %s", err)
|
|
}
|
|
}
|
|
|
|
func TestGetAddressBalance(t *testing.T) {
|
|
ltcAddress := "LdP8Qox1VAhCzLJNqrr74YovaWYyNBUWvL"
|
|
ltc := currency.LTC
|
|
description := "Description of Wallet"
|
|
balance := float64(1000)
|
|
|
|
portfolio := Base{}
|
|
portfolio.AddAddress(ltcAddress, description, ltc, balance)
|
|
|
|
addBalance, _ := portfolio.GetAddressBalance("LdP8Qox1VAhCzLJNqrr74YovaWYyNBUWvL",
|
|
description,
|
|
ltc)
|
|
|
|
if addBalance != balance {
|
|
t.Error("Portfolio GetAddressBalance() Error: Incorrect value")
|
|
}
|
|
|
|
addBalance, found := portfolio.GetAddressBalance("WigWham",
|
|
description,
|
|
ltc)
|
|
|
|
if addBalance != 0 {
|
|
t.Error("Portfolio GetAddressBalance() Error: Incorrect value")
|
|
}
|
|
if found {
|
|
t.Error("Portfolio GetAddressBalance() Error: Incorrect value")
|
|
}
|
|
}
|
|
|
|
func TestExchangeExists(t *testing.T) {
|
|
newBase := Base{}
|
|
newBase.AddAddress("someaddress",
|
|
currency.LTC.String(),
|
|
currency.NewCode("LTCWALLETTEST"),
|
|
0.02)
|
|
|
|
if !newBase.ExchangeExists("someaddress") {
|
|
t.Error("portfolio_test.go - AddressExists error")
|
|
}
|
|
if newBase.ExchangeExists("bla") {
|
|
t.Error("portfolio_test.go - AddressExists error")
|
|
}
|
|
}
|
|
|
|
func TestAddressExists(t *testing.T) {
|
|
newbase := Base{}
|
|
newbase.AddAddress("someaddress",
|
|
currency.LTC.String(),
|
|
currency.NewCode("LTCWALLETTEST"),
|
|
0.02)
|
|
|
|
if !newbase.AddressExists("someaddress") {
|
|
t.Error("portfolio_test.go - AddressExists error")
|
|
}
|
|
if newbase.AddressExists("bla") {
|
|
t.Error("portfolio_test.go - AddressExists error")
|
|
}
|
|
}
|
|
|
|
func TestExchangeAddressExists(t *testing.T) {
|
|
newbase := Base{}
|
|
newbase.AddAddress("someaddress",
|
|
currency.LTC.String(),
|
|
currency.LTC,
|
|
0.02)
|
|
|
|
if !newbase.ExchangeAddressExists("someaddress", currency.LTC) {
|
|
t.Error("portfolio_test.go - ExchangeAddressExists error")
|
|
}
|
|
if newbase.ExchangeAddressExists("TEST", currency.LTC) {
|
|
t.Error("portfolio_test.go - ExchangeAddressExists error")
|
|
}
|
|
}
|
|
|
|
func TestAddExchangeAddress(t *testing.T) {
|
|
newbase := Base{}
|
|
newbase.AddExchangeAddress("OKEX", currency.BTC, 100)
|
|
newbase.AddExchangeAddress("OKEX", currency.BTC, 200)
|
|
|
|
if !newbase.ExchangeAddressExists("OKEX", currency.BTC) {
|
|
t.Error("TestExchangeAddressExists address doesn't exist")
|
|
}
|
|
}
|
|
|
|
func TestUpdateAddressBalance(t *testing.T) {
|
|
newbase := Base{}
|
|
newbase.AddAddress("someaddress",
|
|
currency.LTC.String(),
|
|
currency.NewCode("LTCWALLETTEST"),
|
|
0.02)
|
|
|
|
newbase.UpdateAddressBalance("someaddress", 0.03)
|
|
|
|
value := newbase.GetPortfolioSummary()
|
|
if value.Totals[0].Coin != currency.LTC &&
|
|
value.Totals[0].Balance != 0.03 {
|
|
t.Error("portfolio_test.go - UpdateUpdateAddressBalance error")
|
|
}
|
|
}
|
|
|
|
func TestRemoveAddress(t *testing.T) {
|
|
var newbase Base
|
|
if err := newbase.RemoveAddress("", "MEOW", currency.LTC); err == nil {
|
|
t.Error("invalid address should throw an error")
|
|
}
|
|
|
|
if err := newbase.RemoveAddress("Gibson", "", currency.NewCode("")); err == nil {
|
|
t.Error("invalid coin type should throw an error")
|
|
}
|
|
|
|
if err := newbase.RemoveAddress("HIDDENERINO", "MEOW", currency.LTC); err == nil {
|
|
t.Error("non-existent address should throw an error")
|
|
}
|
|
|
|
newbase.AddAddress("someaddr",
|
|
currency.LTC.String(),
|
|
currency.NewCode("LTCWALLETTEST"),
|
|
420)
|
|
|
|
if !newbase.AddressExists("someaddr") {
|
|
t.Error("portfolio_test.go - TestRemoveAddress")
|
|
}
|
|
|
|
newbase.RemoveAddress("someaddr",
|
|
currency.LTC.String(),
|
|
currency.NewCode("LTCWALLETTEST"))
|
|
if newbase.AddressExists("someaddr") {
|
|
t.Error("portfolio_test.go - TestRemoveAddress")
|
|
}
|
|
}
|
|
|
|
func TestRemoveExchangeAddress(t *testing.T) {
|
|
newbase := Base{}
|
|
exchangeName := "BallerExchange"
|
|
coinType := currency.LTC
|
|
|
|
newbase.AddExchangeAddress(exchangeName, coinType, 420)
|
|
|
|
if !newbase.ExchangeAddressExists(exchangeName, coinType) {
|
|
t.Error("portfolio_test.go - TestRemoveAddress")
|
|
}
|
|
|
|
newbase.RemoveExchangeAddress(exchangeName, coinType)
|
|
if newbase.ExchangeAddressExists(exchangeName, coinType) {
|
|
t.Error("portfolio_test.go - TestRemoveAddress")
|
|
}
|
|
}
|
|
|
|
func TestUpdateExchangeAddressBalance(t *testing.T) {
|
|
newbase := Base{}
|
|
newbase.AddExchangeAddress("someaddress", currency.LTC, 0.02)
|
|
portfolio := GetPortfolio()
|
|
portfolio.Seed(newbase)
|
|
portfolio.UpdateExchangeAddressBalance("someaddress", currency.LTC, 0.04)
|
|
|
|
value := portfolio.GetPortfolioSummary()
|
|
if value.Totals[0].Coin != currency.LTC && value.Totals[0].Balance != 0.04 {
|
|
t.Error("portfolio_test.go - UpdateExchangeAddressBalance error")
|
|
}
|
|
}
|
|
|
|
func TestAddAddress(t *testing.T) {
|
|
var newbase Base
|
|
if err := newbase.AddAddress("", "MEOW", currency.LTC, 1); err == nil {
|
|
t.Error("invalid address should throw an error")
|
|
}
|
|
|
|
if err := newbase.AddAddress("Gibson", "", currency.NewCode(""), 1); err == nil {
|
|
t.Error("invalid coin type should throw an error")
|
|
}
|
|
|
|
// test adding an exchange address
|
|
err := newbase.AddAddress("COINUT", PortfolioAddressExchange, currency.LTC, 0)
|
|
if err != nil {
|
|
t.Errorf("failed to add address: %v", err)
|
|
}
|
|
|
|
// add a test portfolio address and amount
|
|
newbase.AddAddress("Gibson",
|
|
currency.LTC.String(),
|
|
currency.NewCode("LTCWALLETTEST"),
|
|
0.02)
|
|
|
|
// test updating the balance and make sure it's reflected
|
|
newbase.AddAddress("Gibson", currency.LTC.String(),
|
|
currency.NewCode("LTCWALLETTEST"), 0.05)
|
|
b, _ := newbase.GetAddressBalance("Gibson", "LTC",
|
|
currency.NewCode("LTCWALLETTEST"))
|
|
if b != 0.05 {
|
|
t.Error("invalid portfolio amount")
|
|
}
|
|
|
|
portfolio := GetPortfolio()
|
|
portfolio.Seed(newbase)
|
|
if !portfolio.AddressExists("Gibson") {
|
|
t.Error("portfolio_test.go - AddAddress error")
|
|
}
|
|
|
|
// Test updating balance to <= 0, expected result is to remove the address.
|
|
// Fail if address still exists.
|
|
newbase.AddAddress("Gibson",
|
|
currency.LTC.String(),
|
|
currency.NewCode("LTCWALLETTEST"),
|
|
-1)
|
|
|
|
if newbase.AddressExists("Gibson") {
|
|
t.Error("portfolio_test.go - AddAddress error")
|
|
}
|
|
}
|
|
|
|
func TestUpdatePortfolio(t *testing.T) {
|
|
newbase := Base{}
|
|
newbase.AddAddress("someaddress",
|
|
currency.LTC.String(),
|
|
currency.NewCode("LTCWALLETTEST"),
|
|
0.02)
|
|
|
|
portfolio := GetPortfolio()
|
|
portfolio.Seed(newbase)
|
|
|
|
err := portfolio.UpdatePortfolio(
|
|
[]string{"LdP8Qox1VAhCzLJNqrr74YovaWYyNBUWvL"},
|
|
currency.LTC)
|
|
if err != nil {
|
|
t.Error("portfolio_test.go - UpdatePortfolio error", err)
|
|
}
|
|
err = portfolio.UpdatePortfolio([]string{"Testy"}, currency.LTC)
|
|
if err == nil {
|
|
t.Error("portfolio_test.go - UpdatePortfolio error cannot be nil")
|
|
}
|
|
err = portfolio.UpdatePortfolio(
|
|
[]string{"LdP8Qox1VAhCzLJNqrr74YovaWYyNBUWvL", "LVa8wZ983PvWtdwXZ8viK6SocMENLCXkEy"},
|
|
currency.LTC,
|
|
)
|
|
if err != nil {
|
|
t.Error("portfolio_test.go - UpdatePortfolio error", err)
|
|
}
|
|
err = portfolio.UpdatePortfolio(
|
|
[]string{"LdP8Qox1VAhCzLJNqrr74YovaWYyNBUWvL", "Testy"}, currency.LTC,
|
|
)
|
|
if err == nil {
|
|
t.Error("portfolio_test.go - UpdatePortfolio error cannot be nil")
|
|
}
|
|
|
|
time.Sleep(time.Second * 5)
|
|
err = portfolio.UpdatePortfolio(
|
|
[]string{"0xb794f5ea0ba39494ce839613fffba74279579268",
|
|
"0xe853c56864a2ebe4576a807d26fdc4a0ada51919"}, currency.ETH,
|
|
)
|
|
if err == nil { // eth support seems to have been dropped for cryptoid
|
|
t.Error("portfolio_test.go - UpdatePortfolio error cannot be nil")
|
|
}
|
|
err = portfolio.UpdatePortfolio(
|
|
[]string{"0xb794f5ea0ba39494ce839613fffba74279579268", "TESTY"}, currency.ETH,
|
|
)
|
|
if err == nil {
|
|
t.Error("portfolio_test.go - UpdatePortfolio error cannot be nil")
|
|
}
|
|
|
|
err = portfolio.UpdatePortfolio(
|
|
[]string{PortfolioAddressExchange, PortfolioAddressPersonal}, currency.LTC)
|
|
|
|
if err != nil {
|
|
t.Error("portfolio_test.go - UpdatePortfolio error", err)
|
|
}
|
|
}
|
|
|
|
func TestGetPortfolioByExchange(t *testing.T) {
|
|
newbase := Base{}
|
|
newbase.AddExchangeAddress("OKEX", currency.LTC, 0.07)
|
|
newbase.AddExchangeAddress("Bitfinex", currency.LTC, 0.05)
|
|
newbase.AddAddress("someaddress", "LTC", currency.NewCode(PortfolioAddressPersonal), 0.03)
|
|
portfolio := GetPortfolio()
|
|
portfolio.Seed(newbase)
|
|
value := portfolio.GetPortfolioByExchange("OKEX")
|
|
result, ok := value[currency.LTC]
|
|
if !ok {
|
|
t.Error("portfolio_test.go - GetPortfolioByExchange error")
|
|
}
|
|
|
|
if result != 0.07 {
|
|
t.Error("portfolio_test.go - GetPortfolioByExchange result != 0.10")
|
|
}
|
|
|
|
value = portfolio.GetPortfolioByExchange("Bitfinex")
|
|
result, ok = value[currency.LTC]
|
|
if !ok {
|
|
t.Error("portfolio_test.go - GetPortfolioByExchange error")
|
|
}
|
|
|
|
if result != 0.05 {
|
|
t.Error("portfolio_test.go - GetPortfolioByExchange result != 0.05")
|
|
}
|
|
}
|
|
|
|
func TestGetExchangePortfolio(t *testing.T) {
|
|
newbase := Base{}
|
|
newbase.AddAddress("OKEX", PortfolioAddressExchange, currency.LTC, 0.03)
|
|
newbase.AddAddress("Bitfinex", PortfolioAddressExchange, currency.LTC, 0.05)
|
|
newbase.AddAddress("someaddress", PortfolioAddressPersonal, currency.LTC, 0.03)
|
|
portfolio := GetPortfolio()
|
|
portfolio.Seed(newbase)
|
|
value := portfolio.GetExchangePortfolio()
|
|
|
|
result, ok := value[currency.LTC]
|
|
if !ok {
|
|
t.Error("portfolio_test.go - GetExchangePortfolio error")
|
|
}
|
|
|
|
if result != 0.08 {
|
|
t.Error("portfolio_test.go - GetExchangePortfolio result != 0.08")
|
|
}
|
|
}
|
|
|
|
func TestGetPersonalPortfolio(t *testing.T) {
|
|
newbase := Base{}
|
|
newbase.AddAddress("someaddress", PortfolioAddressPersonal, currency.N2O, 0.02)
|
|
newbase.AddAddress("anotheraddress", PortfolioAddressPersonal, currency.N2O, 0.03)
|
|
newbase.AddAddress("Exchange", PortfolioAddressExchange, currency.N2O, 0.01)
|
|
portfolio := GetPortfolio()
|
|
portfolio.Seed(newbase)
|
|
value := portfolio.GetPersonalPortfolio()
|
|
result, ok := value[currency.N2O]
|
|
if !ok {
|
|
t.Error("portfolio_test.go - GetPersonalPortfolio error")
|
|
}
|
|
|
|
if result != 0.05 {
|
|
t.Error("portfolio_test.go - GetPersonalPortfolio result != 0.05")
|
|
}
|
|
}
|
|
|
|
func TestGetPortfolioSummary(t *testing.T) {
|
|
newbase := Base{}
|
|
// Personal holdings
|
|
newbase.AddAddress("someaddress", PortfolioAddressPersonal, currency.LTC, 1)
|
|
newbase.AddAddress("someaddress2", PortfolioAddressPersonal, currency.LTC, 2)
|
|
newbase.AddAddress("someaddress3", PortfolioAddressPersonal, currency.BTC, 100)
|
|
newbase.AddAddress("0xde0b295669a9fd93d5f28d9ec85e40f4cb697bae",
|
|
PortfolioAddressPersonal, currency.ETH, 865346880000000000)
|
|
newbase.AddAddress("0x9edc81c813b26165f607a8d1b8db87a02f34307f",
|
|
PortfolioAddressPersonal, currency.ETH, 165346880000000000)
|
|
|
|
// Exchange holdings
|
|
newbase.AddExchangeAddress("Bitfinex", currency.LTC, 20)
|
|
newbase.AddExchangeAddress("Bitfinex", currency.BTC, 100)
|
|
newbase.AddExchangeAddress("OKEX", currency.ETH, 42)
|
|
|
|
portfolio := GetPortfolio()
|
|
portfolio.Seed(newbase)
|
|
value := portfolio.GetPortfolioSummary()
|
|
|
|
getTotalsVal := func(c currency.Code) Coin {
|
|
for x := range value.Totals {
|
|
if value.Totals[x].Coin == c {
|
|
return value.Totals[x]
|
|
}
|
|
}
|
|
return Coin{}
|
|
}
|
|
|
|
if getTotalsVal(currency.LTC).Coin != currency.LTC {
|
|
t.Error("portfolio_test.go - TestGetPortfolioSummary error")
|
|
}
|
|
|
|
if getTotalsVal(currency.ETH).Coin == currency.LTC {
|
|
t.Error("portfolio_test.go - TestGetPortfolioSummary error")
|
|
}
|
|
|
|
if getTotalsVal(currency.LTC).Balance != 23 {
|
|
t.Error("portfolio_test.go - TestGetPortfolioSummary error")
|
|
}
|
|
|
|
if getTotalsVal(currency.BTC).Balance != 200 {
|
|
t.Error("portfolio_test.go - TestGetPortfolioSummary error")
|
|
}
|
|
}
|
|
|
|
func TestGetPortfolioGroupedCoin(t *testing.T) {
|
|
newbase := Base{}
|
|
newbase.AddAddress("someaddress", currency.LTC.String(), currency.LTC, 0.02)
|
|
newbase.AddAddress("Exchange", PortfolioAddressExchange, currency.LTC, 0.05)
|
|
portfolio := GetPortfolio()
|
|
portfolio.Seed(newbase)
|
|
value := portfolio.GetPortfolioGroupedCoin()
|
|
if value[currency.LTC][0] != "someaddress" && len(value[currency.LTC][0]) != 1 {
|
|
t.Error("portfolio_test.go - GetPortfolioGroupedCoin error")
|
|
}
|
|
}
|
|
|
|
func TestSeed(t *testing.T) {
|
|
newbase := Base{}
|
|
newbase.AddAddress("someaddress", currency.LTC.String(), currency.LTC, 0.02)
|
|
portfolio := GetPortfolio()
|
|
portfolio.Seed(newbase)
|
|
|
|
if !portfolio.AddressExists("someaddress") {
|
|
t.Error("portfolio_test.go - Seed error")
|
|
}
|
|
}
|
|
|
|
func TestStartPortfolioWatcher(t *testing.T) {
|
|
newBase := Base{}
|
|
newBase.AddAddress("LX2LMYXtuv5tiYEMztSSoEZcafFPYJFRK1",
|
|
currency.LTC.String(),
|
|
currency.NewCode(PortfolioAddressPersonal),
|
|
0.02)
|
|
|
|
newBase.AddAddress("Testy",
|
|
currency.LTC.String(),
|
|
currency.NewCode(PortfolioAddressPersonal),
|
|
0.02)
|
|
|
|
portfolio := GetPortfolio()
|
|
portfolio.Seed(newBase)
|
|
|
|
if !portfolio.AddressExists("LX2LMYXtuv5tiYEMztSSoEZcafFPYJFRK1") {
|
|
t.Error("portfolio_test.go - TestStartPortfolioWatcher")
|
|
}
|
|
|
|
go StartPortfolioWatcher()
|
|
}
|
|
|
|
func TestGetPortfolio(t *testing.T) {
|
|
ptrBASE := GetPortfolio()
|
|
if reflect.TypeOf(ptrBASE).String() != "*portfolio.Base" {
|
|
t.Error("portfolio_test.go - GetoPortfolio error")
|
|
}
|
|
}
|