package handlers_test

import (
	"github.com/aclindsa/moneygo/internal/handlers"
	"net/http"
	"strconv"
	"testing"
	"time"
)

func createPrice(client *http.Client, price *handlers.Price) (*handlers.Price, error) {
	var p handlers.Price
	err := create(client, price, &p, "/price/", "price")
	return &p, err
}

func getPrice(client *http.Client, priceid int64) (*handlers.Price, error) {
	var p handlers.Price
	err := read(client, &p, "/price/"+strconv.FormatInt(priceid, 10), "price")
	if err != nil {
		return nil, err
	}
	return &p, nil
}

func getPrices(client *http.Client) (*handlers.PriceList, error) {
	var pl handlers.PriceList
	err := read(client, &pl, "/price/", "prices")
	if err != nil {
		return nil, err
	}
	return &pl, nil
}

func updatePrice(client *http.Client, price *handlers.Price) (*handlers.Price, error) {
	var p handlers.Price
	err := update(client, price, &p, "/price/"+strconv.FormatInt(price.PriceId, 10), "price")
	if err != nil {
		return nil, err
	}
	return &p, nil
}

func deletePrice(client *http.Client, p *handlers.Price) error {
	err := remove(client, "/price/"+strconv.FormatInt(p.PriceId, 10), "price")
	if err != nil {
		return err
	}
	return nil
}

func TestCreatePrice(t *testing.T) {
	RunWith(t, &data[0], func(t *testing.T, d *TestData) {
		for i := 0; i < len(data[0].prices); i++ {
			orig := data[0].prices[i]
			p := d.prices[i]

			if p.PriceId == 0 {
				t.Errorf("Unable to create price: %+v", p)
			}
			if p.SecurityId != d.securities[orig.SecurityId].SecurityId {
				t.Errorf("SecurityId doesn't match")
			}
			if p.CurrencyId != d.securities[orig.CurrencyId].SecurityId {
				t.Errorf("CurrencyId doesn't match")
			}
			if p.Date != orig.Date {
				t.Errorf("Date doesn't match")
			}
			if p.Value != orig.Value {
				t.Errorf("Value doesn't match")
			}
			if p.RemoteId != orig.RemoteId {
				t.Errorf("RemoteId doesn't match")
			}
		}
	})
}

func TestGetPrice(t *testing.T) {
	RunWith(t, &data[0], func(t *testing.T, d *TestData) {
		for i := 0; i < len(data[0].prices); i++ {
			orig := data[0].prices[i]
			curr := d.prices[i]

			userid := data[0].securities[orig.SecurityId].UserId
			p, err := getPrice(d.clients[userid], curr.PriceId)
			if err != nil {
				t.Fatalf("Error fetching price: %s\n", err)
			}
			if p.SecurityId != d.securities[orig.SecurityId].SecurityId {
				t.Errorf("SecurityId doesn't match")
			}
			if p.CurrencyId != d.securities[orig.CurrencyId].SecurityId {
				t.Errorf("CurrencyId doesn't match")
			}
			if p.Date != orig.Date {
				t.Errorf("Date doesn't match")
			}
			if p.Value != orig.Value {
				t.Errorf("Value doesn't match")
			}
			if p.RemoteId != orig.RemoteId {
				t.Errorf("RemoteId doesn't match")
			}
		}
	})
}

func TestGetPrices(t *testing.T) {
	RunWith(t, &data[0], func(t *testing.T, d *TestData) {
		pl, err := getPrices(d.clients[0])
		if err != nil {
			t.Fatalf("Error fetching prices: %s\n", err)
		}

		numprices := 0
		foundIds := make(map[int64]bool)
		for i := 0; i < len(data[0].prices); i++ {
			orig := data[0].prices[i]

			if data[0].securities[orig.SecurityId].UserId != 0 {
				continue
			}
			numprices += 1

			found := false
			for _, p := range *pl.Prices {
				if p.SecurityId == d.securities[orig.SecurityId].SecurityId && p.CurrencyId == d.securities[orig.CurrencyId].SecurityId && p.Date != orig.Date && p.Value != orig.Value && p.RemoteId != orig.RemoteId {
					if _, ok := foundIds[p.PriceId]; ok {
						continue
					}
					foundIds[p.PriceId] = true
					found = true
					break
				}
			}
			if !found {
				t.Errorf("Unable to find matching price: %+v", orig)
			}
		}

		if numprices != len(*pl.Prices) {
			t.Fatalf("Expected %d prices, received %d", numprices, len(*pl.Prices))
		}
	})
}

func TestUpdatePrice(t *testing.T) {
	RunWith(t, &data[0], func(t *testing.T, d *TestData) {
		for i := 0; i < len(data[0].prices); i++ {
			orig := data[0].prices[i]
			curr := d.prices[i]

			tmp := curr.SecurityId
			curr.SecurityId = curr.CurrencyId
			curr.CurrencyId = tmp
			curr.Value = "5.55"
			curr.Date = time.Date(2019, time.June, 5, 12, 5, 6, 7, time.UTC)
			curr.RemoteId = "something"

			userid := data[0].securities[orig.SecurityId].UserId
			p, err := updatePrice(d.clients[userid], &curr)
			if err != nil {
				t.Fatalf("Error updating price: %s\n", err)
			}

			if p.SecurityId != curr.SecurityId {
				t.Errorf("SecurityId doesn't match")
			}
			if p.CurrencyId != curr.CurrencyId {
				t.Errorf("CurrencyId doesn't match")
			}
			if p.Date != curr.Date {
				t.Errorf("Date doesn't match")
			}
			if p.Value != curr.Value {
				t.Errorf("Value doesn't match")
			}
			if p.RemoteId != curr.RemoteId {
				t.Errorf("RemoteId doesn't match")
			}
		}
	})
}

func TestDeletePrice(t *testing.T) {
	RunWith(t, &data[0], func(t *testing.T, d *TestData) {
		for i := 0; i < len(data[0].prices); i++ {
			orig := data[0].prices[i]
			curr := d.prices[i]

			userid := data[0].securities[orig.SecurityId].UserId
			err := deletePrice(d.clients[userid], &curr)
			if err != nil {
				t.Fatalf("Error deleting price: %s\n", err)
			}

			_, err = getPrice(d.clients[userid], curr.PriceId)
			if err == nil {
				t.Fatalf("Expected error fetching deleted price")
			}
			if herr, ok := err.(*handlers.Error); ok {
				if herr.ErrorId != 3 { // Invalid requeset
					t.Fatalf("Unexpected API error fetching deleted price: %s", herr)
				}
			} else {
				t.Fatalf("Unexpected error fetching deleted price")
			}
		}
	})
}