So simulieren Sie DB-Ping ohne SQLMock

Bei der Entwicklung von Software stoßen wir häufig auf Situationen, in denen wir Datenbankverbindungen zum Testen simulieren müssen. Manchmal verfügen wir jedoch möglicherweise nicht über ein Tool wie sqlmock, das uns dabei hilft, dieses Ziel zu erreichen. Wie sollen wir also den Ping-Vorgang der Datenbank ohne SQLMock simulieren? In diesem Artikel stellt Ihnen der PHP-Editor Youzi eine einfache und effektive Methode vor, um dieses Ziel zu erreichen, mit der Sie Simulationstests von Datenbankverbindungen erfolgreich durchführen können.

Frageninhalt

Ich möchte wissen, wie man mock.mock 重构我的代码以模拟 db.ping() verwendet. Ich möchte dieses Konzept zunächst ohne Rahmen verstehen.

Das ist der Code, den ich testen möchte:

func Connect() (*sql.DB, error) {

    db, err := sql.Open("mysql", "root:secret@tcp(s-maria-db)/s_db")

    if err != nil {
        return nil, err
    }

    for i := 0; i < 60; i++ {
        // I am hoping to mock this portion
        if err := db.Ping(); err == nil {
            break
        }
        time.Sleep(time.Second)
    }

    return db, nil
}

Ich habe diesen Artikel zur Simulation des Ping-Befehls ausprobiert, aber das beantwortet die Frage nicht wirklich. Wenn ja, verstehe ich die Lösung nicht, da sie für meinen Code funktioniert.

Problemumgehung

Sie müssen eine separate Funktion zum Ping verwenden, und diese muss eine Schnittstelle akzeptieren, damit Sie bedingt ein echtes *sql.db oder eine Scheinfunktion übergeben können. Dies liegt daran, dass Sie Methoden für eine Struktur nicht überschreiben können.

// connect opens a connection to the database.
func connect() (*sql.db, error) {
    return sql.open("mysql", "root:secret@tcp(s-maria-db)/s_db")
}

// pinger defines an interface for pinging.
type pinger interface {
    ping() error
}

// ping attempts to ping the database, trying several times before failing.
func ping(p pinger) error {
    const maxattempts = 60
    var err error
    for i := 0; i < maxattempts; i++ {
        if err = p.ping(); err == nil {
            return nil
        }
        if i < maxattempts - 1 {
            time.sleep(time.second)
        }
    }
    return err
}

func main() {
    if err := run(); err != nil {
        log.fatal(err)
    }
}

func run() error {
    db, err := connect()
    if err != nil {
        return fmt.errorf("connecting to db: %w", err)
    }
    defer db.close()

    if err = ping(db); err != nil {
        return fmt.errorf("pinging db: %w", err)
    }
    
    ...
}

type mockDB struct {
    mock.Mock
}

func (m *mockDB) Ping() error {
    args := m.Called()
    return args.Error(0)
}

func TestPing(t *testing.T) {
    db := &mockDB{}
    db.On("Ping").Return(...)

    err := Ping(db)

    ...
    db.AssertExpectations(t)
}

Das obige ist der detaillierte Inhalt vonSo simulieren Sie DB-Ping ohne SQLMock. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!