103 lines
3.5 KiB
Go
103 lines
3.5 KiB
Go
package pump_parser
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"testing"
|
|
|
|
agbinary "github.com/gagliardetto/binary"
|
|
"github.com/gagliardetto/solana-go"
|
|
"github.com/mr-tron/base58"
|
|
)
|
|
|
|
type legacyPumpTradeEvent struct {
|
|
Mint solana.PublicKey
|
|
SolAmount uint64
|
|
TokenAmount uint64
|
|
IsBuy bool
|
|
User solana.PublicKey
|
|
Timestamp int64
|
|
VirtualSolReserves uint64
|
|
VirtualTokenReserves uint64
|
|
RealSolReserves uint64
|
|
RealTokenReserves uint64
|
|
FeeRecipient solana.PublicKey
|
|
FeeBasisPoints uint64
|
|
Fee uint64
|
|
Creator solana.PublicKey
|
|
CreatorFeeBasisPoints uint64
|
|
CreatorFee uint64
|
|
TrackVolume bool
|
|
TotalUnclaimedTokens uint64
|
|
TotalClaimedTokens uint64
|
|
CurrentSolVolume uint64
|
|
LastUpdateTimestamp int64
|
|
IxName string
|
|
}
|
|
|
|
func TestTradeEvent(t *testing.T) {
|
|
hexData := "e445a52e51cb9a1dbddb7fd34ee661ee051d1834b36cc6f04cc5bd998d53ab2a566a0ca2415bcfad5f9ed6941a851d3f84ecb200000000006c267d17170000000190d2c525ef0ea205f4b4abfdb6eaaf37fcb5a1b1dec2e2689448eecab6ba93b6c922246900000000c314f11a0d000000be71bf9e22080200c368cd1e06000000bed9ac52910901004ac2f8d0dd5cbc97e3289c197cb5062a54f3d956b9ce6e5115f96567aa5cb3e65f000000000000002c6a010000000000c9e17c171227a50a5b62e3a4a3f8ff4fafe0bca9c332bdf7f32eedbc4229604d1e000000000000005f72000000000000010000000000000000000000000000000000000000000000000000000000000000100000006275795f65786163745f736f6c5f696e"
|
|
d, err := hex.DecodeString(hexData)
|
|
if err != nil {
|
|
t.Errorf("Failed to decode base64 data: %v", err)
|
|
}
|
|
|
|
var tradeEvent legacyPumpTradeEvent
|
|
|
|
err = agbinary.NewBorshDecoder(d[16:]).Decode(&tradeEvent)
|
|
if err != nil {
|
|
t.Fatalf("Failed to deserialize trade event: %v", err)
|
|
}
|
|
if tradeEvent.IxName != "buy_exact_sol_in" {
|
|
t.Fatalf("IxName = %q, want buy_exact_sol_in", tradeEvent.IxName)
|
|
}
|
|
if tradeEvent.SolAmount != 11725956 {
|
|
t.Fatalf("SolAmount = %d, want 11725956", tradeEvent.SolAmount)
|
|
}
|
|
if !tradeEvent.IsBuy {
|
|
t.Fatalf("IsBuy = false, want true")
|
|
}
|
|
t.Logf("Trade Event: %+v", tradeEvent)
|
|
|
|
xx, err := base58.Decode("3Bxs48EzTZB4tzRd")
|
|
fmt.Println(len(xx), err)
|
|
|
|
}
|
|
func TestCal(t *testing.T) {
|
|
//e445a52e51cb9a1db94afc7d1bd7bc6f5e99e54b
|
|
// . b94afc7d1bd7bc6f
|
|
s := calculateDiscriminator("global:initialize_with_permission")
|
|
fmt.Println(hex.EncodeToString(s[:]))
|
|
|
|
s2, _ := base58.Decode("6ApXSNCamGdm")
|
|
s3 := binary.LittleEndian.Uint64(s2[1:])
|
|
fmt.Println(s2, s3)
|
|
|
|
fmt.Println(solana.MustPublicKeyFromBase58("BM9CcyErJcu2mjrFvUsRRrD3snGeHDDVirJLvL6EjvMN").IsOnCurve())
|
|
}
|
|
|
|
func TestPumpCompleteMatchesTradeEvent(t *testing.T) {
|
|
mint := solana.MustPublicKeyFromBase58("8GNGkNnfBuoTP3QRnmdNzSYuuE15M8tvcNvxNsV4pump")
|
|
user := solana.MustPublicKeyFromBase58("DS95KxqUCCjwQaXhD7fhKatXbivwWDNrJdNV5ZcubGdz")
|
|
bondingCurve := solana.MustPublicKeyFromBase58("Gz5EX3X7kUDS48baijJKubQDKy3BBKpnMJQ3f3W1e9jA")
|
|
|
|
tradeEvent := PumpTradeEvent{
|
|
Mint: mint,
|
|
User: user,
|
|
}
|
|
completeEvent := CompleteEvent{
|
|
Mint: mint,
|
|
User: user,
|
|
BondingCurve: bondingCurve,
|
|
}
|
|
if !pumpCompleteMatchesTradeEvent(completeEvent, tradeEvent, bondingCurve) {
|
|
t.Fatal("pumpCompleteMatchesTradeEvent() = false, want true")
|
|
}
|
|
|
|
completeEvent.User = solana.MustPublicKeyFromBase58("3g89wLRwJ5P22fkCdPJBAP7iiYAo6yY96geQvMYj6tYm")
|
|
if pumpCompleteMatchesTradeEvent(completeEvent, tradeEvent, bondingCurve) {
|
|
t.Fatal("pumpCompleteMatchesTradeEvent() = true for mismatched user")
|
|
}
|
|
}
|