Adjust + Test

models/bill/tests/bill_test.go

@@ -0,0 +1,95 @@
+package bill_test
+
+import (
+	"testing"
+
+	"cloud.o-forge.io/core/oc-lib/models/bill"
+	"cloud.o-forge.io/core/oc-lib/models/common/enum"
+	"cloud.o-forge.io/core/oc-lib/models/order"
+	"cloud.o-forge.io/core/oc-lib/models/utils"
+	"cloud.o-forge.io/core/oc-lib/tools"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+// ---- Bill model ----
+
+func TestBill_StoreDraftDefault(t *testing.T) {
+	b := &bill.Bill{}
+	b.StoreDraftDefault()
+	assert.True(t, b.IsDraft)
+}
+
+func TestBill_CanDelete_Draft(t *testing.T) {
+	b := &bill.Bill{}
+	b.IsDraft = true
+	assert.True(t, b.CanDelete())
+}
+
+func TestBill_CanDelete_NonDraft(t *testing.T) {
+	b := &bill.Bill{}
+	b.IsDraft = false
+	assert.False(t, b.CanDelete())
+}
+
+func TestBill_CanUpdate_StatusChange_NonDraft(t *testing.T) {
+	b := &bill.Bill{Status: enum.PENDING}
+	b.IsDraft = false
+	set := &bill.Bill{Status: enum.PAID}
+	ok, returned := b.CanUpdate(set)
+	assert.True(t, ok)
+	assert.Equal(t, enum.PAID, returned.(*bill.Bill).Status)
+}
+
+func TestBill_CanUpdate_SameStatus_NonDraft(t *testing.T) {
+	b := &bill.Bill{Status: enum.PENDING}
+	b.IsDraft = false
+	set := &bill.Bill{Status: enum.PENDING}
+	ok, _ := b.CanUpdate(set)
+	assert.False(t, ok)
+}
+
+func TestBill_CanUpdate_Draft(t *testing.T) {
+	b := &bill.Bill{Status: enum.PENDING}
+	b.IsDraft = true
+	set := &bill.Bill{Status: enum.PAID}
+	ok, _ := b.CanUpdate(set)
+	assert.True(t, ok)
+}
+
+func TestBill_GetAccessor(t *testing.T) {
+	b := &bill.Bill{}
+	acc := b.GetAccessor(&tools.APIRequest{})
+	assert.NotNil(t, acc)
+}
+
+func TestBill_GetAccessor_NilRequest(t *testing.T) {
+	b := &bill.Bill{}
+	acc := b.GetAccessor(nil)
+	assert.NotNil(t, acc)
+}
+
+// ---- GenerateBill ----
+
+func TestGenerateBill_Basic(t *testing.T) {
+	o := &order.Order{
+		AbstractObject: utils.AbstractObject{UUID: "order-uuid-1"},
+	}
+	req := &tools.APIRequest{PeerID: "peer-abc"}
+	b, err := bill.GenerateBill(o, req)
+	require.NoError(t, err)
+	assert.NotNil(t, b)
+	assert.Equal(t, "order-uuid-1", b.OrderID)
+	assert.Equal(t, enum.PENDING, b.Status)
+	assert.False(t, b.IsDraft)
+	assert.Contains(t, b.Name, "peer-abc")
+}
+
+// ---- SumUpBill ----
+
+func TestBill_SumUpBill_NoSubOrders(t *testing.T) {
+	b := &bill.Bill{Total: 0}
+	result, err := b.SumUpBill(nil)
+	require.NoError(t, err)
+	assert.Equal(t, 0.0, result.Total)
+}