Fix tests broken by #4144 (#4147)

hypermodeinc · Oct 9, 2019 · fce1819 · fce1819
1 parent 9961fe3
commit fce1819
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/dgraph/cmd/counter/increment_test.go b/dgraph/cmd/counter/increment_test.go
@@ -224,10 +224,10 @@ func TestBestEffortTs(t *testing.T) {
 	incrementInLoop(t, dg, 1)
 	readBestEffort(t, dg, pred, 1)
 	txn := dg.NewReadOnlyTxn().BestEffort()
-	_, err := queryCounter(txn, pred)
+	_, err := queryCounter(context.Background(), txn, pred)
 	require.NoError(t, err)
 
-	incrementInLoop(t, dg, 1)        // Increment the MaxAssigned ts at Alpha.
-	_, err = queryCounter(txn, pred) // The timestamp here shouldn't change.
+	incrementInLoop(t, dg, 1)                              // Increment the MaxAssigned ts at Alpha.
+	_, err = queryCounter(context.Background(), txn, pred) // The timestamp here shouldn't change.
 	require.NoError(t, err)
 }