dshmvqnl98119 2017-09-20 00:30
浏览 144
已采纳

用存储过程模拟Postgresql

I've been going through the test files of https://github.com/DATA-DOG/go-sqlmock to figure out how to create a stored procedure for mocking purposes. I have:

_, err = db.Exec(`
    CREATE OR REPLACE FUNCTION val() RETURNS INT AS 
        $$ SELECT 1; $$ 
    LANGUAGE sql;   
`)

if err != nil {
    t.Fatal(err)
}

I get:

all expectations were already fulfilled, call to exec 'CREATE OR REPLACE FUNCTION val() RETURNS INT AS $$ SELECT 1; $$ LANGUAGE sql;' query with args [] was not expected

If, instead, I try it with

        mock.ExpectExec(`
            CREATE OR REPLACE FUNCTION val() RETURNS INT AS
                $$ SELECT 1; $$
            LANGUAGE sql;
        `,
        ).WillReturnResult(sqlmock.NewResult(0, 0))

        if err := mock.ExpectationsWereMet(); err != nil {
            t.Fatal(err)
        }

I get:

there is a remaining expectation which was not matched: ExpectedExec => expecting Exec which:
              - matches sql: 'CREATE OR REPLACE FUNCTION val() RETURNS INT AS $$ SELECT 1; $$ LANGUAGE sql;'
              - is without arguments
              - should return Result having:
                  LastInsertId: 0
                  RowsAffected: 0

I am really confused on how to setup a basic stored procedure.

  • 写回答

1条回答 默认 最新

  • drsfgwuw61488 2017-09-20 02:38
    关注

    The sqlmock library works pretty well for this.

    But please note that the ExpectExec receives a regular expression in order to match:

    // ExpectExec expects Exec() to be called with sql query
    // which match sqlRegexStr given regexp.
    // the *ExpectedExec allows to mock database response
    ExpectExec(sqlRegexStr string) *ExpectedExec
    

    You are sending that function the exact string you expect to receive without any escaping.

    To escape the string, add this:

    import (
       "regexp"
    )
    

    And then when adding the expectation, escape your string (note the regexp.QuoteMeta):

    mock.ExpectExec(regexp.QuoteMeta(`
        CREATE OR REPLACE FUNCTION val() RETURNS INT AS
            $$
            SELECT 1;
            $$
        LANGUAGE sql;
    `),
    ).WillReturnResult(sqlmock.NewResult(0, 0))
    

    That way, the escaped regexp will match your exec command.

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

悬赏问题

  • ¥15 python:excel数据写入多个对应word文档
  • ¥60 全一数分解素因子和素数循环节位数
  • ¥15 ffmpeg如何安装到虚拟环境
  • ¥188 寻找能做王者评分提取的
  • ¥15 matlab用simulink求解一个二阶微分方程,要求截图
  • ¥30 乘子法解约束最优化问题的matlab代码文件,最好有matlab代码文件
  • ¥15 写论文,需要数据支撑
  • ¥15 identifier of an instance of 类 was altered from xx to xx错误
  • ¥100 反编译微信小游戏求指导
  • ¥15 docker模式webrtc-streamer 无法播放公网rtsp