more unit tests

Author: Richard Patel

accounts_test.go

@@ -32,40 +32,40 @@ var (
 	caseMappingAccount []byte
 )
 
-func TestProductAccount(t *testing.T) {
-	expected := ProductAccount{
-		AccountHeader: AccountHeader{
-			Magic:       Magic,
-			Version:     V2,
-			AccountType: AccountTypeProduct,
-			Size:        161,
-		},
-		FirstPrice: solana.MustPublicKeyFromBase58("E36MyBbavhYKHVLWR79GiReNNnBDiHj6nWA7htbkNZbh"),
-		AttrsData: [464]byte{
-			0x06, 0x73, 0x79, 0x6d, 0x62, 0x6f, 0x6c, 0x0a,
-			0x46, 0x58, 0x2e, 0x45, 0x55, 0x52, 0x2f, 0x55,
-			0x53, 0x44, 0x0a, 0x61, 0x73, 0x73, 0x65, 0x74,
-			0x5f, 0x74, 0x79, 0x70, 0x65, 0x02, 0x46, 0x58,
-			0x0e, 0x71, 0x75, 0x6f, 0x74, 0x65, 0x5f, 0x63,
-			0x75, 0x72, 0x72, 0x65, 0x6e, 0x63, 0x79, 0x03,
-			0x55, 0x53, 0x44, 0x0b, 0x64, 0x65, 0x73, 0x63,
-			0x72, 0x69, 0x70, 0x74, 0x69, 0x6f, 0x6e, 0x07,
-			0x45, 0x55, 0x52, 0x2f, 0x55, 0x53, 0x44, 0x0e,
-			0x67, 0x65, 0x6e, 0x65, 0x72, 0x69, 0x63, 0x5f,
-			0x73, 0x79, 0x6d, 0x62, 0x6f, 0x6c, 0x06, 0x45,
-			0x55, 0x52, 0x55, 0x53, 0x44, 0x04, 0x62, 0x61,
-			0x73, 0x65, 0x03, 0x45, 0x55, 0x52, 0x05, 0x74,
-			0x65, 0x6e, 0x6f, 0x72, 0x04, 0x53, 0x70, 0x6f,
-			0x74, 0x53, 0x44, 0x00, 0x00, 0x00, 0x00, 0x00,
-			0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
-			0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
-			0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
-		},
-	}
+var productAccount_EWxGfxoPQSNA2744AYdAKmsQZ8F9o9M7oKkvL3VM1dko = ProductAccount{
+	AccountHeader: AccountHeader{
+		Magic:       Magic,
+		Version:     V2,
+		AccountType: AccountTypeProduct,
+		Size:        161,
+	},
+	FirstPrice: solana.MustPublicKeyFromBase58("E36MyBbavhYKHVLWR79GiReNNnBDiHj6nWA7htbkNZbh"),
+	AttrsData: [464]byte{
+		0x06, 0x73, 0x79, 0x6d, 0x62, 0x6f, 0x6c, 0x0a,
+		0x46, 0x58, 0x2e, 0x45, 0x55, 0x52, 0x2f, 0x55,
+		0x53, 0x44, 0x0a, 0x61, 0x73, 0x73, 0x65, 0x74,
+		0x5f, 0x74, 0x79, 0x70, 0x65, 0x02, 0x46, 0x58,
+		0x0e, 0x71, 0x75, 0x6f, 0x74, 0x65, 0x5f, 0x63,
+		0x75, 0x72, 0x72, 0x65, 0x6e, 0x63, 0x79, 0x03,
+		0x55, 0x53, 0x44, 0x0b, 0x64, 0x65, 0x73, 0x63,
+		0x72, 0x69, 0x70, 0x74, 0x69, 0x6f, 0x6e, 0x07,
+		0x45, 0x55, 0x52, 0x2f, 0x55, 0x53, 0x44, 0x0e,
+		0x67, 0x65, 0x6e, 0x65, 0x72, 0x69, 0x63, 0x5f,
+		0x73, 0x79, 0x6d, 0x62, 0x6f, 0x6c, 0x06, 0x45,
+		0x55, 0x52, 0x55, 0x53, 0x44, 0x04, 0x62, 0x61,
+		0x73, 0x65, 0x03, 0x45, 0x55, 0x52, 0x05, 0x74,
+		0x65, 0x6e, 0x6f, 0x72, 0x04, 0x53, 0x70, 0x6f,
+		0x74, 0x53, 0x44, 0x00, 0x00, 0x00, 0x00, 0x00,
+		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+		0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+	},
+}
 
+func TestProductAccount(t *testing.T) {
 	var actual ProductAccount
 	require.NoError(t, actual.UnmarshalBinary(caseProductAccount))
-	assert.Equal(t, &expected, &actual)
+	assert.Equal(t, &productAccount_EWxGfxoPQSNA2744AYdAKmsQZ8F9o9M7oKkvL3VM1dko, &actual)
 
 	t.Run("GetAttrsMap", func(t *testing.T) {
 		expected := map[string]string{
@@ -193,8 +193,20 @@ var priceAccount_E36MyBbavhYKHVLWR79GiReNNnBDiHj6nWA7htbkNZbh = PriceAccount{
 func TestPriceAccount(t *testing.T) {
 	var actual PriceAccount
 	require.NoError(t, actual.UnmarshalBinary(casePriceAccount))
-
 	assert.Equal(t, &priceAccount_E36MyBbavhYKHVLWR79GiReNNnBDiHj6nWA7htbkNZbh, &actual)
+
+	t.Run("GetComponent_Exists", func(t *testing.T) {
+		pubkey := solana.MustPublicKeyFromBase58("EevTjv14eGHqsxKvgpastHsuLr9FNPfzkP23wG61pT2U")
+		comp := actual.GetComponent(&pubkey)
+		assert.NotNil(t, comp)
+		assert.Equal(t, pubkey, comp.Publisher)
+	})
+
+	t.Run("GetComponent_NotExists", func(t *testing.T) {
+		pubkey := solana.StakeProgramID
+		comp := actual.GetComponent(&pubkey)
+		assert.Nil(t, comp)
+	})
 }
 
 func TestMappingAccount(t *testing.T) {

products_test.go

@@ -0,0 +1,73 @@
+//  Copyright 2022 Blockdaemon Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package pyth
+
+import (
+	"context"
+	"encoding/base64"
+	"io"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/gagliardetto/solana-go"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestClient_GetProductAccount(t *testing.T) {
+	server := httptest.NewServer(http.HandlerFunc(func(wr http.ResponseWriter, req *http.Request) {
+		buf, err := io.ReadAll(req.Body)
+		require.NoError(t, err)
+		assert.JSONEq(t, `{
+			"jsonrpc": "2.0",
+			"id": 0,
+			"method": "getAccountInfo",
+			"params": [
+				"EWxGfxoPQSNA2744AYdAKmsQZ8F9o9M7oKkvL3VM1dko",
+				{"encoding": "base64"}
+			]
+		}`, string(buf))
+
+		_, err = wr.Write([]byte(`{
+			"jsonrpc": "2.0",
+			"id": 0,
+			"result": {
+				"context": {
+					"slot": 118773287
+				},
+				"value": {
+					"data": [
+						"` + base64.StdEncoding.EncodeToString(caseProductAccount) + `",
+						"base64"
+					],
+					"executable": false,
+					"lamports": 23942400,
+					"owner": "gSbePebfvPy7tRqimPoVecS2UsBvYv46ynrzWocc92s",
+					"rentEpoch": 274
+				}
+			}
+		}`))
+		require.NoError(t, err)
+	}))
+	defer server.Close()
+
+	c := NewClient(ProgramIDDevnet, server.URL, server.URL)
+	acc, err := c.GetProductAccount(context.Background(), solana.MustPublicKeyFromBase58("EWxGfxoPQSNA2744AYdAKmsQZ8F9o9M7oKkvL3VM1dko"))
+	require.NoError(t, err)
+	require.NotNil(t, acc)
+
+	assert.Equal(t, &productAccount_EWxGfxoPQSNA2744AYdAKmsQZ8F9o9M7oKkvL3VM1dko, acc)
+}