Rumah >pembangunan bahagian belakang >Golang >Gorm padam klausa ujian sqlmock

Gorm padam klausa ujian sqlmock

WBOY
WBOYke hadapan
2024-02-08 22:30:31864semak imbas

Gorm 删除子句 sqlmock 测试

Editor PHP Xinyi memperkenalkan kepada anda ujian sqlmock klausa pemadaman Gorm. Gorm ialah rangka kerja ORM yang sangat baik dalam bahasa Go, dan sqlmock ialah alat ujian untuk Gorm, digunakan untuk mensimulasikan operasi pangkalan data. Apabila membangunkan Gorm, kita sering perlu menguji operasi pemadaman pangkalan data Pada masa ini, kita boleh menggunakan sqlmock untuk mensimulasikan operasi pemadaman pangkalan data untuk ujian unit dan ujian integrasi. Artikel ini akan memperkenalkan anda secara terperinci cara menggunakan Gorm dan sqlmock untuk menguji klausa pemadaman, membantu anda menjalankan kerja pembangunan berkaitan pangkalan data dengan lebih baik.

Kandungan soalan

Saya ada gorm delete, hasil pulangan ialah:

expirationdate := time.now().utc().add(-(48 * time.hour))
var deletedusers users
res := gormdb.withcontext(ctx).
    table("my_users").
    clauses(clause.returning{columns: []clause.column{{name: "email"}}}).
    where("created_at < ?", expirationdate).
    delete(&deletedusers)

Ujian dengan klausa kini sentiasa gagal. Contohnya:

sqlMock.ExpectExec(`DELETE`)
    .WithArgs(expirationDate)
    .WillReturnResult(sqlmock.NewResult(1, 1))

Ralat terima:

"Memanggil pertanyaan 'padam daripada "pengguna_saya" di mana create_at df31941f1ef2ef55d47de4941e260273 menjangkakan exec atau execcontext di mana: n - sepadan dengan sql: 'delete'n - mengambil hujah: n 0 - 2023-01-18 06:15:34.694274 +0000 utc n - harus kembali dengan Keputusan untuk: n lastinsertid: 1n rowsaffected: 1"

Saya mencuba banyak jangkaan sqlmock lain tetapi mereka mempunyai masalah yang sama. Selain itu, kami tidak mempunyai nilai pulangan dalam expectexec, hanya dalam expectquery... Adakah sesiapa perlu menguji pertanyaan gorm dengan klausa?

Penyelesaian

Saya berjaya menguruskan keperluan anda. Mula-mula, izinkan saya berkongsi fail yang saya tulis, dan kemudian saya akan membimbing anda melalui semua perubahan yang berkaitan. Fail ini adalah untuk kegunaan pengeluaran repo.go 和用于测试代码的 repo_test.go.

repo.go

package gormdelete

import (
    "context"
    "time"

    "gorm.io/gorm"
    "gorm.io/gorm/clause"
)

type users struct {
    email string
}

func delete(ctx context.context, gormdb *gorm.db) error {
    expirationdate := time.now().utc().add(-(48 * time.hour))

    var deletedusers users
    res := gormdb.withcontext(ctx).table("my_users").clauses(clause.returning{columns: []clause.column{{name: "email"}}}).where("created_at < ?", expirationdate).delete(&deletedusers)
    if res.error != nil {
        return res.error
    }
    return nil
}

Memandangkan anda tidak memberikan fail lengkap, saya cuba meneka apa yang hilang.

repo_test.go

package gormdelete

import (
    "context"
    "database/sql/driver"
    "testing"
    "time"

    "github.com/DATA-DOG/go-sqlmock"
    "github.com/stretchr/testify/assert"
    "gorm.io/driver/postgres"
    "gorm.io/gorm"
)

// this is taken directly from the docs
// https://github.com/DATA-DOG/go-sqlmock#matching-arguments-like-timetime
type AnyTime struct{}

// Match satisfies sqlmock.Argument interface
func (a AnyTime) Match(v driver.Value) bool {
    _, ok := v.(time.Time)
    return ok
}

func TestDelete(t *testing.T) {
    db, mock, err := sqlmock.New()
    if err != nil {
        t.Fatalf("an error was not expected: %v", err)
    }

    conn, _ := db.Conn(context.Background())
    gormDb, err := gorm.Open(postgres.New(postgres.Config{
        Conn: conn,
    }))

    row := sqlmock.NewRows([]string{"email"}).AddRow("<a href="https://www.php.cn/link/89fee0513b6668e555959f5dc23238e9" class="__cf_email__" data-cfemail="9febfaecebdffae7fef2eff3fab1fcf0f2">[email&#160;protected]</a>")
    mock.ExpectBegin()
    mock.ExpectQuery("DELETE FROM \"my_users\" WHERE created_at < ?").WithArgs(AnyTime{}).WillReturnRows(row)
    mock.ExpectCommit()

    err = Delete(context.Background(), gormDb)

    assert.Nil(t, err)
    if err = mock.ExpectationsWereMet(); err != nil {
        t.Errorf("not all expectations were met: %v", err)
    }
}

Berikut adalah lebih banyak perubahan yang patut disebut:

  1. Saya instantiated anytime mengikut dokumentasi (anda boleh lihat pautan dalam komen).
  2. Saya meneka tetapan untuk dbmockgormdb tetapi saya fikir ia sepatutnya lebih kurang sama.
  3. Saya menukar penggunaan expectexec kepada expectquery kerana kami akan mengembalikan set hasil yang ditentukan oleh kaedah expectexec 的用法切换为 expectquery,因为我们将返回 clauses 文件中 repo.go dalam fail clauses.
  4. Anda mesti memasukkan expectquery 包装在 expectbeginexpectcommit.
  5. Akhir sekali, perhatikan perbezaan dalam cara pemandu mengharapkan parameter dalam pernyataan sql. Dalam kod pengeluaran anda mempunyai pilihan untuk menggunakan ?。但在测试代码中,只能使用?, jika tidak, ia tidak berkelakuan seperti yang diharapkan.

Semoga ia membantu, jika tidak sila beritahu saya!

Atas ialah kandungan terperinci Gorm padam klausa ujian sqlmock. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!

Kenyataan:
Artikel ini dikembalikan pada:stackoverflow.com. Jika ada pelanggaran, sila hubungi admin@php.cn Padam