feat(x/ecocredit): query Batches ORM

x/ecocredit/server/core/query_batches.go

@@ -0,0 +1,45 @@
+package core
+
+import (
+	"context"
+	"github.com/cosmos/cosmos-sdk/orm/model/ormlist"
+	ecocreditv1beta1 "github.com/regen-network/regen-ledger/api/regen/ecocredit/v1beta1"
+	"github.com/regen-network/regen-ledger/x/ecocredit/v1beta1"
+)
+
+// Batches queries for all batches in the given credit class.
+func (k Keeper) Batches(ctx context.Context, request *v1beta1.QueryBatchesRequest) (*v1beta1.QueryBatchesResponse, error) {
+	pg, err := GogoPageReqToPulsarPageReq(request.Pagination)
+	if err != nil {
+		return nil, err
+	}
+	project, err := k.stateStore.ProjectInfoStore().GetByName(ctx, request.ProjectId)
+	if err != nil {
+		return nil, err
+	}
+	it, err := k.stateStore.BatchInfoStore().List(ctx, ecocreditv1beta1.BatchInfoProjectIdIndexKey{}.WithProjectId(project.Id), ormlist.Paginate(pg))
+	if err != nil {
+		return nil, err
+	}
+
+	batches := make([]*v1beta1.BatchInfo, 0)
+	for it.Next() {
+		batch, err := it.Value()
+		if err != nil {
+			return nil, err
+		}
+		var bi v1beta1.BatchInfo
+		if err = PulsarToGogoSlow(batch, &bi); err != nil {
+			return nil, err
+		}
+		batches = append(batches, &bi)
+	}
+	pr, err := PulsarPageResToGogoPageRes(it.PageResponse())
+	if err != nil {
+		return nil, err
+	}
+	return &v1beta1.QueryBatchesResponse{
+		Batches:    batches,
+		Pagination: pr,
+	}, nil
+}

x/ecocredit/server/core/query_batches_test.go

@@ -0,0 +1,56 @@
+package core
+
+import (
+	"github.com/cosmos/cosmos-sdk/orm/types/ormerrors"
+	"github.com/cosmos/cosmos-sdk/types/query"
+	ecocreditv1beta1 "github.com/regen-network/regen-ledger/api/regen/ecocredit/v1beta1"
+	"github.com/regen-network/regen-ledger/x/ecocredit/v1beta1"
+	"gotest.tools/v3/assert"
+	"testing"
+)
+
+func TestQuery_Batches(t *testing.T) {
+	t.Parallel()
+	s := setupBase(t)
+
+	// make a project and two batches
+	assert.NilError(t, s.stateStore.ProjectInfoStore().Insert(s.ctx, &ecocreditv1beta1.ProjectInfo{
+		Name:            "P01",
+		ClassId:         1,
+		ProjectLocation: "US-CA",
+		Metadata:        nil,
+	}))
+	assert.NilError(t, s.stateStore.BatchInfoStore().Insert(s.ctx, &ecocreditv1beta1.BatchInfo{
+		ProjectId:  1,
+		BatchDenom: "C01-20200101-20220101-001",
+		Metadata:   nil,
+		StartDate:  nil,
+		EndDate:    nil,
+	}))
+	assert.NilError(t, s.stateStore.BatchInfoStore().Insert(s.ctx, &ecocreditv1beta1.BatchInfo{
+		ProjectId:  1,
+		BatchDenom: "C01-20200101-20220101-002",
+		Metadata:   nil,
+		StartDate:  nil,
+		EndDate:    nil,
+	}))
+
+	// invalid query
+	_, err := s.k.Batches(s.ctx, &v1beta1.QueryBatchesRequest{ProjectId: "F01"})
+	assert.ErrorContains(t, err, ormerrors.NotFound.Error())
+
+	//  base query
+	res, err := s.k.Batches(s.ctx, &v1beta1.QueryBatchesRequest{ProjectId: "P01"})
+	assert.NilError(t, err)
+	assert.Equal(t, 2, len(res.Batches))
+	assert.Equal(t, "C01-20200101-20220101-001", res.Batches[0].BatchDenom)
+
+	// paginated query
+	res, err = s.k.Batches(s.ctx, &v1beta1.QueryBatchesRequest{
+		ProjectId: "P01",
+		Pagination: &query.PageRequest{Limit: 1, CountTotal: true},
+	})
+	assert.NilError(t, err)
+	assert.Equal(t, 1, len(res.Batches))
+	assert.Equal(t, uint64(2), res.Pagination.Total)
+}