Function TestGetLineageProvider

lineage_test.go:31–87

(t *testing.T)

)

func TestGetLineageProvider(t *testing.T) {
	datasource, _, err := newTestDataSource()
	if err != nil {
		t.Fatalf("Error creating test data source: %s", err)
	}

	ledgerforge, err := NewLedgerForge(datasource)
	if err != nil {
		t.Fatalf("Error creating LedgerForge instance: %s", err)
	}

	tests := []struct {
		name     string
		txn      *model.Transaction
		expected string
	}{
		{
			name:     "No metadata",
			txn:      &model.Transaction{},
			expected: "",
		},
		{
			name: "Metadata without provider",
			txn: &model.Transaction{
				MetaData: map[string]interface{}{
					"other_key": "value",
				},
			},
			expected: "",
		},
		{
			name: "Metadata with provider",
			txn: &model.Transaction{
				MetaData: map[string]interface{}{
					"LEDGERFORGE_LINEAGE_PROVIDER": "stripe",
				},
			},
			expected: "stripe",
		},
		{
			name: "Provider with non-string value",
			txn: &model.Transaction{
				MetaData: map[string]interface{}{
					"LEDGERFORGE_LINEAGE_PROVIDER": 123,
				},
			},
			expected: "",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result := ledgerforge.getLineageProvider(tt.txn)
			assert.Equal(t, tt.expected, result)
		})
	}
}