database: fix notification test (wrong signature)

quay · Feb 24, 2016 · b5d8f99 · b5d8f99
1 parent 45ed80d
commit b5d8f99
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/database/pgsql/notification_test.go b/database/pgsql/notification_test.go
@@ -94,7 +94,7 @@ func TestNotification(t *testing.T) {
 			},
 		},
 	}
-	assert.Nil(t, datastore.insertVulnerability(v1, false))
+	assert.Nil(t, datastore.insertVulnerability(v1, false, true))
 
 	// Get the notification associated to the previously inserted vulnerability.
 	notification, err := datastore.GetAvailableNotification(time.Second)
@@ -158,7 +158,7 @@ func TestNotification(t *testing.T) {
 		},
 	}
 
-	if assert.Nil(t, datastore.insertVulnerability(v1b, false)) {
+	if assert.Nil(t, datastore.insertVulnerability(v1b, false, true)) {
 		notification, err = datastore.GetAvailableNotification(time.Second)
 		assert.Nil(t, err)
 		assert.NotEmpty(t, notification.Name)