go sqlgen/integration_test: Add a test that covers Count functionality

samsarahq · Mar 29, 2019 · 8416f8c · 8416f8c
1 parent 922395e
commit 8416f8c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/sqlgen/integration_test.go b/sqlgen/integration_test.go
@@ -101,6 +101,11 @@ func TestIntegrationBasic(t *testing.T) {
 			Mood: &mood,
 		},
 	}, users)
+
+	var numBobs int64
+	err = db.Count(context.Background(), &User{}, &numBobs, Filter{"name": "Bob"})
+	assert.NoError(t, err)
+	assert.Equal(t, int64(1), numBobs)
 }
 
 // TestContextCancelBeforeRowsScan demonstrates we don't