go1inch_test.go

package go1inch_test

import (
	"context"
	"testing"

	go1inch "github.com/0xbsa/go-1inch"
)

func TestHealthCheckReponse(t *testing.T) {
	client := go1inch.NewDefaultClient()
	networks := []go1inch.Network{go1inch.Eth, go1inch.Bsc, go1inch.Matic, go1inch.Optimism, go1inch.Arbitrum}
	for _, network := range networks {
		res, _, err := client.Healthcheck(context.Background(), network)
		if err != nil {
			t.Fatal(err)
		}
		if res.Status != "OK" {
			t.Errorf("healthcheck returned %s, expected OK", res.Status)
		}
		t.Logf("Network healthcheck %d: %v", network, res)
	}
}

func TestQuoteWithoutOpts(t *testing.T) {
	client := go1inch.NewDefaultClient()

	res, _, _, err := client.Quote(context.Background(), 1,
		"0xeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee",
		"0x6b175474e89094c44da98b954eedeac495271d0f",
		"100000000000000000000",
		nil,
	)
	if err != nil {
		t.Fatal(err)
	}
	t.Log(res)
}

func TestQuoteWithOpts(t *testing.T) {
	client := go1inch.NewDefaultClient()
	res, _, _, err := client.Quote(context.Background(), 1,
		"0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE",
		"0x111111111117dc0aa78b770fa6a738034120c302",
		"100000000000000000000",
		&go1inch.QuoteOpts{
			Fee:             "1",
			GasPrice:        "1",
			ComplexityLevel: "2",
			GasLimit:        "10000000",
			VirtualParts:    "50",
			Parts:           "50",
			MainRouteParts:  "10",
		},
	)
	if err != nil {
		t.Fatal(err)
	}
	t.Log(res)
}

func TestQuoteWithMissingParameter(t *testing.T) {
	client := go1inch.NewDefaultClient()

	_, _, _, err := client.Quote(context.Background(), 1,
		"",
		"0x6b175474e89094c44da98b954eedeac495271d0f",
		"100000000000000000000",
		nil,
	)
	if err == nil {
		t.Error("getting swap should have failed. fromTokenAddress is missing.")
	}
}

func TestSwapWithoutOpts(t *testing.T) {
	client := go1inch.NewDefaultClient()

	res, _, _, err := client.Swap(context.Background(), 1,
		"0xeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee",
		"0x6b175474e89094c44da98b954eedeac495271d0f",
		"100000000000000",
		"0x52bc44d5378309ee2abf1539bf71de1b7d7be3b5",
		1,
		nil,
	)
	if err != nil {
		t.Fatal(err)
	}
	t.Logf("%#v", res)
}

func TestSwapWithOpts(t *testing.T) {
	client := go1inch.NewDefaultClient()

	res, _, _, err := client.Swap(context.Background(), 1,
		"0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE",
		"0x111111111117dc0aa78b770fa6a738034120c302",
		"100000000000000000000",
		"0x52bc44d5378309ee2abf1539bf71de1b7d7be3b5",
		1,
		&go1inch.SwapOpts{
			Receiver:          "0x52bc44d5378309ee2abf1539bf71de1b7d7be3b5",
			ReferrerAddress:   "0xAb5801a7D398351b8bE11C439e05C5B3259aeC9B",
			Fee:               "1",
			GasPrice:          "1",
			BurnChi:           false,
			ComplexityLevel:   "2",
			AllowPartialFill:  true,
			DisableEstimate:   true,
			GasLimit:          "10000000",
			Parts:             "50",
			VirtualParts:      "50",
			MainRouteParts:    "10",
			IncludeTokensInfo: true,
		})
	if err != nil {
		t.Fatal(err)
	}
	t.Logf("%#v", res)
}

func TestSwapWithMissingParameter(t *testing.T) {
	client := go1inch.NewDefaultClient()

	_, _, _, err := client.Swap(context.Background(), 1,
		"",
		"0x6b175474e89094c44da98b954eedeac495271d0f",
		"100000000000000000000",
		"",
		0,
		nil,
	)
	if err == nil {
		t.Error("getting swap should have failed. fromTokenAddress is missing.")
	}
}

func TestGetLiquiditySouces(t *testing.T) {
	client := go1inch.NewDefaultClient()

	res, _, err := client.LiquiditySouces(context.Background(), 137)
	if err != nil {
		t.Fatal(err)
	}
	t.Log(res)
}

func TestGetTokens(t *testing.T) {
	client := go1inch.NewDefaultClient()

	res, _, err := client.Tokens(context.Background(), 137)
	if err != nil {
		t.Fatal(err)
	}
	t.Log(res)
}

func TestApproveSpender(t *testing.T) {
	client := go1inch.NewDefaultClient()
	res, _, err := client.ApproveSpender(context.Background(), 56)
	if err != nil {
		t.Fatal(err)
	}
	t.Log(res)
}

func TestApproveAllowance(t *testing.T) {
	const TOKEN_ADDRESS = "0xbb4CdB9CBd36B01bD1cBaEBF2De08d9173bc095c"  // WBNB
	const WALLET_ADDRESS = "0xC49926C4124cEe1cbA0Ea94Ea31a6c12318df947" // Wallet that has Router V4 approved for WBNB
	client := go1inch.NewDefaultClient()
	res, _, err := client.ApproveAllowance(context.Background(), 56, TOKEN_ADDRESS, WALLET_ADDRESS)
	if err != nil {
		t.Fatal(err)
	}
	t.Log(res)
}

func TestApproveTransactionWithoutOpts(t *testing.T) {
	client := go1inch.NewDefaultClient()
	res, _, err := client.ApproveTransaction(
		context.Background(),
		1,
		"0x6b175474e89094c44da98b954eedeac495271d0f",
		nil,
	)
	if err != nil {
		t.Fatal(err)
	}
	t.Log(res)
}

func TestApproveTransactionWithOpts(t *testing.T) {
	client := go1inch.NewDefaultClient()
	res, _, err := client.ApproveTransaction(
		context.Background(),
		1,
		"0x6b175474e89094c44da98b954eedeac495271d0f",
		&go1inch.ApproveTransactionOpts{
			Amount: "100000",
		},
	)
	if err != nil {
		t.Fatal(err)
	}
	t.Log(res)
}

func TestWBTC(t *testing.T) {
	client := go1inch.NewDefaultClient()
	_, code, _, err := client.Quote(context.Background(), 137, "0x1bfd67037b42cf73acf2047067bd4f2c47d9bfd6", "0x2791bca1f2de4661ed88a30c99a7a9449aa84174", "100000000", nil)
	if err != nil {
		t.Fatal(err)
	}
	if code != 200 {
		t.Error("expected 200, got ", code)
	}
}