From 7f78dcdc67f476c186c4f19a9f0e0f109b9f8ed7 Mon Sep 17 00:00:00 2001 From: William Perron Date: Mon, 1 Jan 2024 11:40:44 -0500 Subject: [PATCH] fix tests --- fmt_test.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/fmt_test.go b/fmt_test.go index 3af3cd2..ad1f9f8 100644 --- a/fmt_test.go +++ b/fmt_test.go @@ -1,6 +1,7 @@ package themis import ( + "context" "fmt" "testing" @@ -14,7 +15,7 @@ func TestFormatRows(t *testing.T) { rows, err := store.db.Query("SELECT provinces.name, provinces.region, provinces.area, provinces.trade_node FROM provinces WHERE area = 'Gascony'") assert.NoError(t, err) - fmtd, err := FormatRows(rows) + fmtd, err := FormatRows(context.Background(), rows) assert.NoError(t, err) assert.Equal(t, `| name | region | area | trade_node | | -------- | ------ | ------- | ---------- | @@ -32,7 +33,7 @@ func TestFormatRowsAggregated(t *testing.T) { rows, err := store.db.Query("SELECT count(1) as total, trade_node from provinces where region = 'France' group by trade_node") assert.NoError(t, err) - fmtd, err := FormatRows(rows) + fmtd, err := FormatRows(context.Background(), rows) assert.NoError(t, err) assert.Equal(t, `| total | trade_node | | ----- | --------------- |