Updated functional test with bug fix for Optum#344

AmudaPalani · May 29, 2020 · b004836 · b004836
1 parent 45bd355
commit b004836
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/acceptance/update_lease_status_test.go b/tests/acceptance/update_lease_status_test.go
@@ -253,11 +253,16 @@ func TestUpdateLeaseStatusLambda(t *testing.T) {
 		err = json.Unmarshal(result.Payload, &resp)
 		require.Nil(t, err)
 
-		// Check lease status is active
+		// Check lease status is inactive
 		lease, err := dbSvc.GetLease(accountID, "user")
 		require.Nil(t, err)
 		require.Equal(t, db.LeaseStatus("Inactive"), lease.LeaseStatus)
 		require.Equal(t, db.LeaseStatusReason("Expired"), lease.LeaseStatusReason)
+
+		// Check account status is updated to NotReady
+		account, err := dbSvc.GetAccount(accountID)
+		require.Nil(t, err)
+		require.Equal(t, db.NotReady, account.AccountStatus)
 	})
 
 	t.Run("Exceeded lease budget result in Inactive lease with reason OverBudget.", func(t *testing.T) {