cleanup session test

slicebit · Jun 4, 2016 · 3ef1506 · 3ef1506
1 parent aa667d0
commit 3ef1506
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/session_test.go b/session_test.go
@@ -32,8 +32,10 @@ func TestSessionAddError(t *testing.T) {
 	type User struct {
 		ID string `qb:"constraints:primary_key"`
 	}
+	session.AddTable(User{})
 	err = session.CreateAll()
 	assert.Nil(t, err)
+
 	session.Close()
 	defer assert.Panics(t, func() {
 		session.Add(&User{ID: "hello"})