使用存储过程模拟PostgreSQL

Question

我一直在浏览https://github.com/data-dog/go-sqlmock的测试文件，以找出如何为模拟目的创建存储过程。我有：

_, err = db.Exec(`
    CREATE OR REPLACE FUNCTION val() RETURNS INT AS 
        $$ SELECT 1; $$ 
    LANGUAGE sql;   
`)
if err != nil {
    t.Fatal(err)
}

我得到：

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

如果，我会用

尝试

        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)
        }

我得到：

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

我对如何设置基本存储过程感到非常困惑。

Answer 1

sqlmock库为此效果很好。

但请注意，ExpectExec接收正则表达式以匹配：

// 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

您正在向该功能发送您期望接收的确切字符串而无需任何逃脱的字符串。

要逃脱字符串，请添加以下内容：

import (
   "regexp"
)

，然后在添加期望时，逃脱字符串（注意regexp.QuoteMeta）：

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

这样，逃脱的REGEXP将与您的Exec命令匹配。