{"id":206989,"date":"2025-07-08T09:32:44","date_gmt":"2025-07-08T01:32:44","guid":{"rendered":"https:\/\/server.hk\/cnblog\/206989\/"},"modified":"2025-07-08T09:32:44","modified_gmt":"2025-07-08T01:32:44","slug":"%e4%bd%bf%e7%94%a8-go-sqlmock-%e5%b9%b6%e5%b0%86%e5%8f%82%e6%95%b0%e6%8f%92%e5%85%a5%e6%a8%a1%e6%8b%9f%e6%9f%a5%e8%af%a2%e7%9a%84%e9%97%ae%e9%a2%98","status":"publish","type":"post","link":"https:\/\/server.hk\/cnblog\/206989\/","title":{"rendered":"\u4f7f\u7528 go-sqlmock \u5e76\u5c06\u53c2\u6570\u63d2\u5165\u6a21\u62df\u67e5\u8be2\u7684\u95ee\u9898"},"content":{"rendered":"<p><b><\/b> <\/p>\n<p>\u5f53\u524d\u4f4d\u7f6e\uff1a <span>&gt;<\/span> <span>&gt;<\/span> <span>&gt;<\/span> <span>&gt;<\/span> <span>\u4f7f\u7528 go-sqlmock \u5e76\u5c06\u53c2\u6570\u63d2\u5165\u6a21\u62df\u67e5\u8be2\u7684\u95ee\u9898<\/span><\/p>\n<p><span>\u6765\u6e90\uff1astackoverflow<\/span><br \/>\n<span>2024-04-19 12:06:32<\/span><br \/>\n<span><i><\/i>0\u6d4f\u89c8<\/span><br \/>\n<span style=\"cursor: pointer\"><i><\/i>\u6536\u85cf<\/span> <\/p>\n<p>\u4ece\u73b0\u5728\u5f00\u59cb\uff0c\u52aa\u529b\u5b66\u4e60\u5427\uff01\u672c\u6587<span style=\"color: #FF6600;, Helvetica, Arial, sans-serif;font-size: 14px;background-color: #FFFFFF\">\u300a\u4f7f\u7528 go-sqlmock \u5e76\u5c06\u53c2\u6570\u63d2\u5165\u6a21\u62df\u67e5\u8be2\u7684\u95ee\u9898\u300b<\/span>\u4e3b\u8981\u8bb2\u89e3\u4e86<span style=\"color: #FF6600;, Helvetica, Arial, sans-serif;font-size: 14px;background-color: #FFFFFF\"><\/span>\u7b49\u7b49\u76f8\u5173\u77e5\u8bc6\u70b9\uff0c\u6211\u4f1a\u5728\u4e2d\u6301\u7eed\u66f4\u65b0\u76f8\u5173\u7684\u7cfb\u5217\u6587\u7ae0\uff0c\u6b22\u8fce\u5927\u5bb6\u5173\u6ce8\u5e76\u79ef\u6781\u7559\u8a00\u5efa\u8bae\u3002\u4e0b\u9762\u5c31\u5148\u4e00\u8d77\u6765\u770b\u4e00\u4e0b\u672c\u7bc7\u6b63\u6587\u5185\u5bb9\u5427\uff0c\u5e0c\u671b\u80fd\u5e2e\u5230\u4f60\uff01<\/p>\n<p> \u95ee\u9898\u5185\u5bb9<br \/>\n <\/p>\n<p>\u6211\u6b63\u5728\u5c1d\u8bd5\u4f7f\u7528 go-sqlmock \u6a21\u62df\u6211\u7684\u67e5\u8be2\u51fd\u6570\u5e76\u7c7b\u4f3c\u5730\u590d\u5236\u6570\u636e\u5e93\u8868\u3002\u4f46\u662f\uff0c\u6211\u6ca1\u6709\u5f97\u5230\u6211\u671f\u671b\u7684\u7ed3\u679c\u3002\u67e5\u8be2\u7684\u884c\u4e3a\u4e0d\u6b63\u5e38\uff0c\u53c2\u6570\u6ca1\u6709\u63d2\u5165\u5230\u67e5\u8be2\u4e2d\u5e76\u4e14\u5b9e\u9645\u7ed3\u679c\u4e0d\u6b63\u786e\u3002\u6211\u5728\u8fd9\u91cc\u505a\u9519\u4e86\u4ec0\u4e48\uff1f<\/p>\n<p>\u8fd9\u662f\u6211\u6b63\u5728\u5632\u7b11\u7684\u51fd\u6570\u548c\u67e5\u8be2\uff1a<\/p>\n<pre>func (y *yumdatabase) gettransactionid(pkg string) (int, error) {\n    var id int\n\n    queryfortid := \"select tid from trans_cmdline where cmdline like '%install \" + pkg + \"%' order by tid desc limit 1\"\n    row := y.db.queryrow(queryfortid)\n    switch err := row.scan(&amp;id); err {\n    case sql.errnorows:\n        fmt.println(\"no rows were returned\")\n        return 0, err\n    case nil:\n        return id, nil\n    default:\n        return 0, err\n    }\n}<\/pre>\n<p>\u8fd9\u662f\u6a21\u62df\u6d4b\u8bd5\u529f\u80fd\uff1a<\/p>\n<pre>func testgettransactionid(t *testing.t) {\n    db, mock, err := sqlmock.new()\n    if err != nil {\n        t.fatalf(\"err not expected: %v\", err)\n    }\n    pkg := \"tcpdump\"\n    rows := sqlmock.newrows([]string{\"tid\"}).addrow(\"1\").addrow(\"3\")\n    mock.expectquery(\"select tid from trans_cmdline where cmdline like '%install \" + pkg + \"%' order by tid desc limit 1\").willreturnrows(rows)\n\n    mockdb := &amp;yumdatabase{\n        db: db,\n    }\n\n    got, err := mockdb.gettransactionid(\"tcpdump\")\n    assert.equal(t, 3, got)\n}<\/pre>\n<p>\u5982\u679c\u4e0a\u8ff0\u5de5\u4f5c\u6309\u9884\u671f\u8fdb\u884c\uff0c\u6211\u4f1a\u5728\u201cgot\u201d\u4e2d\u8fd4\u56de\u201c3\u201d\uff0c\u4f46\u6211\u4f1a\u8fd4\u56de\u201c1\u201d<\/p>\n<p>\u5176\u6b21\uff0c\u662f\u5426\u53ef\u4ee5\u5c06 rows \u66f4\u6539\u4e3a\u4ee5\u4e0b\u5185\u5bb9\uff1a<\/p>\n<pre>rows := sqlmock.newrows([]string{\"tid\", \"cmdline\"}).addrow(\"1\", \"install test\").addrow(\"3\", \"delete test2\")<\/pre>\n<p>\u5b9e\u9645\u4e0a\u8fdb\u884c\u4e86\u6bd4\u8f83\u201cwhere cmdline like &#8216;%install xyz%&#8217;\u201d\uff0c\u56e0\u4e3a\u6211\u5c1d\u8bd5\u4e86\u8fd9\u4e2a\uff0c\u5e76\u4e14\u6536\u5230\u4e86\u4ee5\u4e0b\u9519\u8bef\uff08\u6240\u6709\u4e3b\u8981\u4ee3\u7801\u90fd\u6784\u5efa\u5e76\u5de5\u4f5c\uff0c\u5305\u62ec\u67e5\u8be2\uff0c\u6240\u4ee5\u8fd9\u662f\u4e00\u4e2a\u95ee\u9898\u6211\u731c\u662f\u6211\u5199\u7684\u6a21\u62df\u4ee3\u7801\uff09\uff1a<\/p>\n<pre>error sql: expected 2 destination arguments in Scan, not 1<\/pre>\n<p>\u6211\u5e0c\u671b\u770b\u5230\u4ece sql \u67e5\u8be2\u8fd4\u56de\u7684\u6700\u9ad8 tid\uff0c\u800c\u4e0d\u662f\u201caddrow\u201d\u4e2d\u6307\u5b9a\u7684\u7b2c\u4e00\u4e2a tid\uff0c\u5e76\u4e14\u6211\u5e0c\u671b\u67e5\u8be2\u5b9e\u73b0\u5bf9\u6a21\u62df\u4e2d\u7684\u201ccmdline\u201d\u884c\u7684\u68c0\u67e5\u3002<\/p>\n<p> <\/p>\n<h2>\u6b63\u786e\u7b54\u6848<\/h2>\n<p> <\/p>\n<p>\u6211\u5c31\u662f\u8fd9\u6837\u7ba1\u7406\u4f60\u7684\u9700\u6c42\u7684\u3002\u9996\u5148\uff0c\u8ba9\u6211\u5206\u4eab\u4ee3\u7801\uff0c\u7136\u540e\uff0c\u6211\u5c06\u5f15\u5bfc\u60a8\u5b8c\u6210\u6240\u6709\u76f8\u5173\u66f4\u6539\u3002\u8be5\u4ee3\u7801\u5305\u542b\u5728\u4e24\u4e2a\u6587\u4ef6\u4e2d\uff1a<code>repo.go<\/code> \u548c <code>repo_test.go<\/code>\u3002<\/p>\n<h3><code>repo.go<\/code> \u6587\u4ef6<\/h3>\n<pre>package repo\n\nimport (\n    \"database\/sql\"\n    \"fmt\"\n)\n\nfunc gettransactionid(db *sql.db, pkg string) (int, error) {\n    var id int\n    row := db.queryrow(\"select tid from trans_cmdline where cmdline like '%install $1%' order by tid desc limit 1\", pkg)\n    switch err := row.scan(&amp;id); err {\n    case sql.errnorows:\n        fmt.println(\"no rows were returned\")\n        return 0, err\n    case nil:\n        return id, nil\n    default:\n        return 0, err\n    }\n}\n<\/pre>\n<p>\u8fd9\u91cc\u6709\u4e24\u4e2a\u5c0f\u6539\u8fdb\uff1a<\/p>\n<ol>\n<li><code>*sql.db<\/code> \u4f5c\u4e3a\u53c2\u6570\u4f20\u5165\u3002\u6b63\u5982\u6700\u4f73\u5b9e\u8df5\u6240\u5efa\u8bae\u7684\uff0c<strong>\u51fd\u6570\u662f\u4e00\u7b49\u516c\u6c11<\/strong>\u3002\u8fd9\u5c31\u662f\u4e3a\u4ec0\u4e48\u6211\u66f4\u613f\u610f\u5c3d\u53ef\u80fd\u575a\u6301\u4f7f\u7528\u5b83\u4eec\u3002<\/li>\n<li>\u6211\u4f7f\u7528\u51c6\u5907\u597d\u7684\u8bed\u53e5\u6765\u4f20\u9012\u67e5\u8be2\u7684\u53c2\u6570\u3002\u4e0d\u662f\u7b80\u5355\u7684\u5b57\u7b26\u4e32\u8fde\u63a5\u3002\u56e0\u6b64\uff0c\u53ef\u4ee5\u66f4\u8f7b\u677e\u5730\u62e6\u622a\u4f20\u9012\u7ed9\u67e5\u8be2\u7684\u53c2\u6570\u5e76\u5bf9\u5176\u8bbe\u7f6e\u671f\u671b<\/li>\n<\/ol>\n<p>\u73b0\u5728\u8ba9\u6211\u4eec\u5207\u6362\u5230\u6d4b\u8bd5\u4ee3\u7801\u3002<\/p>\n<h3><code>repo_test.go<\/code> \u6587\u4ef6<\/h3>\n<pre>package repo\n\nimport (\n    \"database\/sql\"\n    \"testing\"\n\n    \"github.com\/DATA-DOG\/go-sqlmock\"\n    \"github.com\/stretchr\/testify\/assert\"\n)\n\nfunc TestGetTransactionId(t *testing.T) {\n    db, mock, err := sqlmock.New(sqlmock.QueryMatcherOption(sqlmock.QueryMatcherEqual))\n    if err != nil {\n        t.Fatalf(\"err not expected while opening mock db, %v\", err)\n    }\n    t.Run(\"HappyPath\", func(t *testing.T) {\n        rows := sqlmock.NewRows([]string{\"tid\"}).AddRow(\"1\")\n        mock.ExpectQuery(\"SELECT tid FROM trans_cmdline WHERE cmdline LIKE '%install $1%' ORDER BY tid DESC LIMIT 1\").\n            WithArgs(\"tcpdump\").\n            WillReturnRows(rows)\n\n        got, err := GetTransactionId(db, \"tcpdump\")\n\n        assert.Equal(t, 1, got)\n        assert.Nil(t, err)\n    })\n\n    t.Run(\"NoRowsReturned\", func(t *testing.T) {\n        mock.ExpectQuery(\"SELECT tid FROM trans_cmdline WHERE cmdline LIKE '%install $1%' ORDER BY tid DESC LIMIT 1\").\n            WithArgs(\"tcpdump\").\n            WillReturnError(sql.ErrNoRows)\n\n        got, err := GetTransactionId(db, \"tcpdump\")\n\n        assert.Equal(t, 0, got)\n        assert.Equal(t, sql.ErrNoRows, err)\n    })\n}\n<\/pre>\n<p>\u8fd9\u91cc\uff0c\u60a8\u9700\u8981\u6ce8\u610f\u66f4\u591a\u66f4\u6539\uff1a<\/p>\n<ol>\n<li>\u5728\u5b9e\u4f8b\u5316 <code>db<\/code> \u548c <code>mock<\/code> \u65f6\uff0c\u60a8\u5e94\u8be5\u5c06 <code>sqlmock.querymatcherequal<\/code> \u4f5c\u4e3a\u53c2\u6570\u4f20\u9012\u7ed9\u3002\u56e0\u6b64\uff0c\u5b83\u5c06\u5b8c\u5168\u5339\u914d\u67e5\u8be2\u3002<\/li>\n<li><code>expectquery<\/code> \u65b9\u6cd5\u73b0\u5728\u4f7f\u7528\u51c6\u5907\u597d\u7684\u8bed\u53e5\u529f\u80fd\u5e76\u9700\u8981\u4e00\u4e2a\u53c2\u6570\uff08\u4f8b\u5982\u672c\u4f8b\u4e2d\u7684 <code>tcpdump<\/code>\uff09\u3002<\/li>\n<li>\u91cd\u6784\u4e86\u65ad\u8a00\u4ee5\u5229\u7528 <code>github.com\/stretchr\/testify\/assert<\/code> \u5305\u3002<\/li>\n<\/ol>\n<p>\u6211\u5e0c\u671b\u8fd9\u53ef\u4ee5\u5e2e\u52a9\u60a8\u89e3\u51b3\u95ee\u9898\uff0c\u8bf7\u544a\u8bc9\u6211\uff01<\/p>\n<p>\u4ee5\u4e0a\u5c31\u662f\u672c\u6587\u7684\u5168\u90e8\u5185\u5bb9\u4e86\uff0c\u662f\u5426\u6709\u987a\u5229\u5e2e\u52a9\u4f60\u89e3\u51b3\u95ee\u9898\uff1f\u82e5\u662f\u80fd\u7ed9\u4f60\u5e26\u6765\u5b66\u4e60\u4e0a\u7684\u5e2e\u52a9\uff0c\u8bf7\u5927\u5bb6\u591a\u591a\u652f\u6301\uff01\u66f4\u591a\u5173\u4e8eGolang\u7684\u76f8\u5173\u77e5\u8bc6\uff0c\u4e5f\u53ef\u5173\u6ce8\u516c\u4f17\u53f7\u3002<\/p>\n","protected":false},"excerpt":{"rendered":"<p>\u5f53\u524d\u4f4d\u7f6e\uff1a &gt; &gt; &#46;&#46;&#46;<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"closed","ping_status":"","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[4925],"tags":[],"class_list":["post-206989","post","type-post","status-publish","format-standard","hentry","category-4925"],"_links":{"self":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/206989","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/comments?post=206989"}],"version-history":[{"count":0,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/206989\/revisions"}],"wp:attachment":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/media?parent=206989"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/categories?post=206989"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/tags?post=206989"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}