From a0a70b7394eb8d6ee7a65b92ec49437860af4426 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tomasz=20Zdyba=C5=82?= <tomek@zdybal.lap.pl>
Date: Wed, 5 Jul 2023 22:02:46 +0200
Subject: [PATCH] tests: round trip test for blob submission/retrieval

---
 client_test.go | 43 +++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 41 insertions(+), 2 deletions(-)

diff --git a/client_test.go b/client_test.go
index 073b91e..0a82e9e 100644
--- a/client_test.go
+++ b/client_test.go
@@ -12,8 +12,12 @@ import (
 	"testing"
 	"time"
 
+	"cosmossdk.io/math"
 	"github.com/ory/dockertest/v3"
 	"github.com/stretchr/testify/suite"
+
+	blob "github.com/rollkit/celestia-openrpc/types/blob"
+	"github.com/rollkit/celestia-openrpc/types/share"
 )
 
 type TestSuite struct {
@@ -87,6 +91,7 @@ func TestIntegrationTestSuite(t *testing.T) {
 	suite.Run(t, new(TestSuite))
 }
 
+// TestClient is a basic smoke test,  ensuring that client can execute simple methods.
 func (t *TestSuite) TestClient() {
 	client, err := NewClient(context.Background(), t.getRPCAddress(), t.token)
 	t.NoError(err)
@@ -94,8 +99,8 @@ func (t *TestSuite) TestClient() {
 
 	t.NotNil(client)
 
-	ctx, closer := context.WithTimeout(context.Background(), 1*time.Second)
-	defer closer()
+	ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
+	defer cancel()
 
 	resp := client.Share.ProbabilityOfAvailability(ctx)
 	t.NotZero(resp)
@@ -105,6 +110,40 @@ func (t *TestSuite) TestClient() {
 	t.NotEmpty(info.APIVersion)
 }
 
+// TestRoundTrip tests
+func (t *TestSuite) TestRoundTrip() {
+	client, err := NewClient(context.Background(), t.getRPCAddress(), t.token)
+	t.Require().NoError(err)
+	defer client.Close()
+
+	ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
+	defer cancel()
+
+	namespace, err := share.NewBlobNamespaceV0([]byte{1, 2, 3, 4, 5, 6, 7, 8})
+	t.Require().NoError(err)
+	t.Require().NotEmpty(namespace)
+
+	data := []byte("hello world")
+	blobBlob, err := blob.NewBlobV0(namespace, data)
+	t.Require().NoError(err)
+
+	// write blob to DA
+	txResponse, err := client.State.SubmitPayForBlob(ctx, math.NewInt(100000000), 200000000, []*blob.Blob{blobBlob})
+	t.Require().NoError(err)
+	t.Require().NotNil(txResponse)
+	t.Zero(txResponse.Code)
+	t.NotZero(txResponse.Height)
+
+	ctx, cancel = context.WithTimeout(context.Background(), 1*time.Second)
+	defer cancel()
+	// retrieve data back from DA
+	blobs, err := client.Blob.GetAll(ctx, uint64(txResponse.Height), []share.Namespace{namespace})
+	t.Require().NoError(err)
+	t.Require().NotEmpty(blobs)
+	t.Len(blobs, 1)
+	t.Equal(data, blobs[0].Data)
+}
+
 func (t *TestSuite) getRPCAddress() string {
 	return fmt.Sprintf("http://localhost:%s", t.resource.GetPort("26658/tcp"))
 }