@@ -41,7 +41,7 @@ func TestWithComment_NoContext(t *testing.T) {
4141 Config : core.CommenterConfig {EnableDBDriver : true },
4242 },
4343 query : "SELECT 1;" ,
44- wantQuery : "SELECT 1/*db_driver=database%2Fsql%3A*/;" ,
44+ wantQuery : "SELECT 1/*db_driver=' database%2Fsql%3A' */;" ,
4545 },
4646 {
4747 desc : "enable DBDriver and pass static tag driver name" ,
@@ -50,7 +50,7 @@ func TestWithComment_NoContext(t *testing.T) {
5050 Tags : core.StaticTags {DriverName : "postgres" },
5151 },
5252 query : "SELECT 1;" ,
53- wantQuery : "SELECT 1/*db_driver=database%2Fsql%3Apostgres*/;" ,
53+ wantQuery : "SELECT 1/*db_driver=' database%2Fsql%3Apostgres' */;" ,
5454 },
5555 {
5656 desc : "enable DBDriver and pass all static tags" ,
@@ -59,7 +59,7 @@ func TestWithComment_NoContext(t *testing.T) {
5959 Tags : core.StaticTags {DriverName : "postgres" , Application : "app-1" },
6060 },
6161 query : "SELECT 1;" ,
62- wantQuery : "SELECT 1/*db_driver=database%2Fsql%3Apostgres*/;" ,
62+ wantQuery : "SELECT 1/*db_driver=' database%2Fsql%3Apostgres' */;" ,
6363 },
6464 {
6565 desc : "enable other tags and pass all static tags" ,
@@ -68,7 +68,7 @@ func TestWithComment_NoContext(t *testing.T) {
6868 Tags : core.StaticTags {DriverName : "postgres" , Application : "app-1" },
6969 },
7070 query : "SELECT 1;" ,
71- wantQuery : "SELECT 1/*application=app-1,db_driver=database%2Fsql%3Apostgres*/;" ,
71+ wantQuery : "SELECT 1/*application=' app-1' ,db_driver=' database%2Fsql%3Apostgres' */;" ,
7272 },
7373 }
7474 for _ , tc := range testCases {
@@ -122,7 +122,7 @@ func TestWithComment_WithContext(t *testing.T) {
122122 },
123123 ),
124124 query : "SELECT 1;" ,
125- wantQuery : "SELECT 1/*application=app-1,db_driver=database%2Fsql%3Apostgres,framework=custom-golang,route=listData*/;" ,
125+ wantQuery : "SELECT 1/*application=' app-1' ,db_driver=' database%2Fsql%3Apostgres' ,framework=' custom-golang' ,route=' listData' */;" ,
126126 },
127127 {
128128 desc : "only all options but context contains all tags" ,
@@ -147,7 +147,7 @@ func TestWithComment_WithContext(t *testing.T) {
147147 },
148148 ),
149149 query : "SELECT 1;" ,
150- wantQuery : "SELECT 1/*action=any+action,application=app-1,db_driver=database%2Fsql%3Apostgres,framework=custom-golang,route=listData*/;" ,
150+ wantQuery : "SELECT 1/*action=' any+action' ,application=' app-1' ,db_driver=' database%2Fsql%3Apostgres' ,framework=' custom-golang' ,route=' listData' */;" ,
151151 },
152152 }
153153 for _ , tc := range testCases {
0 commit comments