Add test

Ilhasoft · Jun 17, 2020 · c743fa1 · c743fa1
1 parent 6b47ed4
commit c743fa1
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/models/topups_test.go b/models/topups_test.go
@@ -4,6 +4,7 @@ import (
 	"testing"
 
 	"github.com/nyaruka/mailroom/testsuite"
+
 	"github.com/stretchr/testify/assert"
 )
 
@@ -61,4 +62,11 @@ func TestTopups(t *testing.T) {
 		assert.Equal(t, tc.TopupID, topup)
 		tx.MustExec(`INSERT INTO orgs_topupcredits(is_squashed, used, topup_id) VALUES(TRUE, 1, $1)`, tc.OrgID)
 	}
+
+	// topups can be disabled for orgs
+	tx.MustExec(`UPDATE orgs_org SET uses_topups = FALSE WHERE id = $1`, Org1)
+	org, err := loadOrg(ctx, tx, Org1)
+	topup, err := AllocateTopups(ctx, tx, rc, org, 1)
+	assert.NoError(t, err)
+	assert.Equal(t, NilTopupID, topup)
 }