Серия 3-го одиночного теста — тестирование MySQL и Redis
Серия 3-го одиночного теста — тестирование MySQL и Redis

Это второе из серии руководств по модульному тестированию языка Go с нуля. В нем рассказывается, как использовать инструменты go-sqlmock и miniredis для проведения пробного тестирования MySQL и Redis.

В предыдущей статье «Go Single Test Series 2 — Сетевое тестирование» мы рассказали, как использовать инструменты httptest и gock для сетевого тестирования. Помимо сетевых зависимостей, мы также часто используем в разработке различные базы данных, такие как распространенные MySQL и Redis. В этой статье будут приведены примеры, демонстрирующие, как имитировать MySQL и Redis при написании модульных тестов.

Пример кода «Пройдите одиночный тест от нуля до скользкой серии» загружен на Github.,Нажмите👉🏻https://github.com/go-quiz/golang-unit-test-demo Посмотреть полный исходный код.

go-sqlmock

sqlmock это реализация sql/driver макетная библиотека. Для имитации каких-либо действий не требуется устанавливать реальное соединение с базой данных. sql поведение водителя. Его очень удобно использовать для макетирования при написании модульных тестов. Результат выполнения оператора sql.

Установить

Язык кода:javascript
копировать
go get github.com/DATA-DOG/go-sqlmock

Пример использования

Здесь используетсяgo-sqlmockБазовый пример кода, представленный в официальной документации.。 В коде ниже,Мы реализовалиrecordStatsЭта функция используется для записи соответствующих данных, генерируемых при просмотре продуктов пользователями.。Конкретная реализованная функция заключается в выполнении следующих действий дважды за одну транзакцию.SQLдействовать: - существоватьproductsКоличество просмотров текущего товара в таблице+1 - существоватьproduct_viewersВ таблице фиксируются пользователи, просматривавшие текущий продукт.id

Язык кода:javascript
копировать
// app.go
package main

import "database/sql"

// recordStats Записывайте информацию о продукте, просматриваемом пользователем
func recordStats(db *sql.DB, userID, productID int64) (err error) {
	// Открыть транзакцию
	// Управляйте таблицами представлений и product_viewers.
	tx, err := db.Begin()
	if err != nil {
		return
	}

	defer func() {
		switch err {
		case nil:
			err = tx.Commit()
		default:
			tx.Rollback()
		}
	}()

	// Обновить таблицу продуктов
	if _, err = tx.Exec("UPDATE products SET views = views + 1"); err != nil {
		return
	}
	// Вставьте часть данных в таблицу Product_viewers.
	if _, err = tx.Exec(
		"INSERT INTO product_viewers (user_id, product_id) VALUES (?, ?)",
		userID, productID); err != nil {
		return
	}
	return
}

func main() {
	// ПРИМЕЧАНИЕ. Процесс тестирования не требует реального подключения.
	db, err := sql.Open("mysql", "root@/blog")
	if err != nil {
		panic(err)
	}
	defer db.Close()
	// Пользователь с идентификатором пользователя 1 просмотрел продукт с идентификатором продукта 5.
	if err = recordStats(db, 1 /*some user id*/, 5 /*some product id*/); err != nil {
		panic(err)
	}
}

сейчассуществоватьмы должныкодсерединаизrecordStatsНаписание функций Модульное тестирование,Но не хочусуществоватьтест Во время процесса подключитесь к реальной базе данных.тест。На данный момент мы можем использовать его, как показано в примере кода ниже.sqlmockИнструменты, которые можно использоватьmockданные Библиотекадействовать。

Язык кода:javascript
копировать
package main

import (
	"fmt"
	"testing"

	"github.com/DATA-DOG/go-sqlmock"
)

// TestShouldUpdateStats вариант использования успешного выполнения sql
func TestShouldUpdateStats(t *testing.T) {
	// Имитация объекта *sql.DB без подключения к реальной базе данных
	db, mock, err := sqlmock.New()
	if err != nil {
		t.Fatalf("an error '%s' was not expected when opening a stub database connection", err)
	}
	defer db.Close()

	// Возвращаемый результат, когда макет выполняет указанный оператор SQL
	mock.ExpectBegin()
	mock.ExpectExec("UPDATE products").WillReturnResult(sqlmock.NewResult(1, 1))
	mock.ExpectExec("INSERT INTO product_viewers").WithArgs(2, 3).WillReturnResult(sqlmock.NewResult(1, 1))
	mock.ExpectCommit()

	// Передайте макет объекта БД в нашу функцию.
	if err = recordStats(db, 2, 3); err != nil {
		t.Errorf("error was not expected while updating stats: %s", err)
	}

	// Обеспечить достижение ожидаемых результатов
	if err := mock.ExpectationsWereMet(); err != nil {
		t.Errorf("there were unfulfilled expectations: %s", err)
	}
}

// TestShouldRollbackStatUpdatesOnFailure Вариант использования теста для сбоя выполнения и отката sql
func TestShouldRollbackStatUpdatesOnFailure(t *testing.T) {
	db, mock, err := sqlmock.New()
	if err != nil {
		t.Fatalf("an error '%s' was not expected when opening a stub database connection", err)
	}
	defer db.Close()

	mock.ExpectBegin()
	mock.ExpectExec("UPDATE products").WillReturnResult(sqlmock.NewResult(1, 1))
	mock.ExpectExec("INSERT INTO product_viewers").
		WithArgs(2, 3).
		WillReturnError(fmt.Errorf("some error"))
	mock.ExpectRollback()

	// now we execute our method
	if err = recordStats(db, 2, 3); err == nil {
		t.Errorf("was expecting an error, but there was none")
	}

	// we make sure that all expectations were met
	if err := mock.ExpectationsWereMet(); err != nil {
		t.Errorf("there were unfulfilled expectations: %s", err)
	}
}

В приведенном выше коде определены тестовый пример, который выполняется успешно, и тестовый пример, в котором не удается выполнить откат, чтобы гарантировать возможность тестирования каждой логической ветки в нашем коде, что не только улучшает покрытие модульными тестами, но и обеспечивает стабильность кода. Прочность.

Выполните модульный тест и посмотрите окончательные результаты теста.

Язык кода:javascript
копировать
❯ go test -v
=== RUN   TestShouldUpdateStats
--- PASS: TestShouldUpdateStats (0.00s)
=== RUN   TestShouldRollbackStatUpdatesOnFailure
--- PASS: TestShouldRollbackStatUpdatesOnFailure (0.00s)
PASS
ok      golang-unit-test-demo/sqlmock_demo      0.011s

Вы можете видеть, что результаты обоих тестовых случаев соответствуют ожиданиям, и модульный тест пройден.

Во многих сценариях использования инструментов ORM,Также можно использоватьgo-sqlmockБиблиотекаmockданные Библиотекадействоватьруководитьтест。

miniredis

Помимо частого использования MySQL, Redis также часто используется в повседневной разработке. В следующем разделе мы узнаем, как имитировать операции, связанные с Redis, при модульном тестировании.

miniredisэто чистыйgoреализовано для Модульное тестированиеизredis server。Это простой и удобный в использованиииз、основанный на памятиизredisЗаменять,Он имеет настоящий TCP-интерфейс,Вы можете думать об этом какredisВерсияизnet/http/httptest

Когда мы пишем модульные тесты для некоторого кода, содержащего операции Redis, мы можем использовать его для имитации операций Redis.

Установить

Язык кода:javascript
копировать
go get github.com/alicebob/miniredis/v2

Пример использования

здесь сgithub.com/go-redis/redisБиблиотека Например,Программа, содержащая несколькоRedisдействоватьизDoSomethingWithRedisфункция。

Язык кода:javascript
копировать
// redis_op.go
package miniredis_demo

import (
	"context"
	"github.com/go-redis/redis/v8" // Обратите внимание на импортированную версию
	"strings"
	"time"
)

const (
	KeyValidWebsite = "app:valid:website:list"
)

func DoSomethingWithRedis(rdb *redis.Client, key string) bool {
	// Здесь может быть некоторая логика для операций Redis
	ctx := context.TODO()
	if !rdb.SIsMember(ctx, KeyValidWebsite, key).Val() {
		return false
	}
	val, err := rdb.Get(ctx, key).Result()
	if err != nil {
		return false
	}
	if !strings.HasPrefix(val, "https://") {
		val = "https://" + val
	}
	// настраивать blog key Срок действия истекает через пять секунд
	if err := rdb.Set(ctx, "blog", val, 5*time.Second).Err(); err != nil {
		return false
	}
	return true
}

подизкодда, я используюminiredisБиблиотекадляDoSomethingWithRedisНаписание функцийиз Модульное тестированиекод,вminiredisне только поддержкаmockОбычно используетсяизRedisдействовать,Он также предоставляет множество практических справочных функций.,Например, проверьтеkeyиз Соответствует ли значение ожидаемомуизs.CheckGet()и помогите проверитьkeyСрок годностиизs.FastForward()

Язык кода:javascript
копировать
// redis_op_test.go

package miniredis_demo

import (
	"github.com/alicebob/miniredis/v2"
	"github.com/go-redis/redis/v8"
	"testing"
	"time"
)

func TestDoSomethingWithRedis(t *testing.T) {
	// издеваться над Redis server
	s, err := miniredis.Run()
	if err != nil {
		panic(err)
	}
	defer s.Close()

	// Подготовьте данные
	s.Set("q1mi", "liwenzhou.com")
	s.SAdd(KeyValidWebsite, "q1mi")

	// Подключите Redis макета server
	rdb := redis.NewClient(&redis.Options{
		Addr: s.Addr(), // mock redis адрес сервера
	})

	// функция вызова
	ok := DoSomethingWithRedis(rdb, "q1mi")
	if !ok {
		t.Fatal()
	}

	// Вы можете вручную проверить, составляет ли значение в Redis ожидаемое значение.
	if got, err := s.Get("blog"); err != nil || got != "https://liwenzhou.com" {
		t.Fatalf("'blog' has the wrong value")
	}
	// Вы также можете использовать справочный инструмент для проверки
	s.CheckGet(t, "blog", "https://liwenzhou.com")

	// Просроченный чек
	s.FastForward(5 * time.Second) // Перемотка вперед на 5 секунд
	if s.Exists("blog") {
		t.Fatal("'blog' should not have existed anymore")
	}
}

Выполните тест выполнения и просмотрите результаты модульного теста:

Язык кода:javascript
копировать
❯ go test -v
=== RUN   TestDoSomethingWithRedis
--- PASS: TestDoSomethingWithRedis (0.00s)
PASS
ok      golang-unit-test-demo/miniredis_demo    0.052s

miniredisВ основном поддерживает подавляющее большинствоизRedisЗаказ,Вы можете узнать больше об использовании, просмотрев документацию.

Конечно, помимо использованияminiredisПостроить локальноredis Помимо серверного метода, для укладки отдельных методов также можно использовать различные инструменты для укладки свай. Конкретный метод макетирования, который следует использовать при написании модульных тестов, по-прежнему зависит от реальной ситуации.

Подвести итог

Написание кода для ежедневной работы Модульное тестирование时如何处理данные Библиотекаиз Зависимости являются наиболее распространенными.извопрос,В этой статье объясняется, как использоватьgo-sqlmockиminiredisинструментmockСвязанные зависимости。 В следующей статье мы пойдем еще дальше и подробно представим, как имитировать интерфейсы при написании модульных тестов.

boy illustration
Неразрушающее увеличение изображений одним щелчком мыши, чтобы сделать их более четкими артефактами искусственного интеллекта, включая руководства по установке и использованию.
boy illustration
Копикодер: этот инструмент отлично работает с Cursor, Bolt и V0! Предоставьте более качественные подсказки для разработки интерфейса (создание навигационного веб-сайта с использованием искусственного интеллекта).
boy illustration
Новый бесплатный RooCline превосходит Cline v3.1? ! Быстрее, умнее и лучше вилка Cline! (Независимое программирование AI, порог 0)
boy illustration
Разработав более 10 проектов с помощью Cursor, я собрал 10 примеров и 60 подсказок.
boy illustration
Я потратил 72 часа на изучение курсорных агентов, и вот неоспоримые факты, которыми я должен поделиться!
boy illustration
Идеальная интеграция Cursor и DeepSeek API
boy illustration
DeepSeek V3 снижает затраты на обучение больших моделей
boy illustration
Артефакт, увеличивающий количество очков: на основе улучшения характеристик препятствия малым целям Yolov8 (SEAM, MultiSEAM).
boy illustration
DeepSeek V3 раскручивался уже три дня. Сегодня я попробовал самопровозглашенную модель «ChatGPT».
boy illustration
Open Devin — инженер-программист искусственного интеллекта с открытым исходным кодом, который меньше программирует и больше создает.
boy illustration
Эксклюзивное оригинальное улучшение YOLOv8: собственная разработка SPPF | SPPF сочетается с воспринимаемой большой сверткой ядра UniRepLK, а свертка с большим ядром + без расширения улучшает восприимчивое поле
boy illustration
Популярное и подробное объяснение DeepSeek-V3: от его появления до преимуществ и сравнения с GPT-4o.
boy illustration
9 основных словесных инструкций по доработке академических работ с помощью ChatGPT, эффективных и практичных, которые стоит собрать
boy illustration
Вызовите deepseek в vscode для реализации программирования с помощью искусственного интеллекта.
boy illustration
Познакомьтесь с принципами сверточных нейронных сетей (CNN) в одной статье (суперподробно)
boy illustration
50,3 тыс. звезд! Immich: автономное решение для резервного копирования фотографий и видео, которое экономит деньги и избавляет от беспокойства.
boy illustration
Cloud Native|Практика: установка Dashbaord для K8s, графика неплохая
boy illustration
Краткий обзор статьи — использование синтетических данных при обучении больших моделей и оптимизации производительности
boy illustration
MiniPerplx: новая поисковая система искусственного интеллекта с открытым исходным кодом, спонсируемая xAI и Vercel.
boy illustration
Конструкция сервиса Synology Drive сочетает проникновение в интрасеть и синхронизацию папок заметок Obsidian в облаке.
boy illustration
Центр конфигурации————Накос
boy illustration
Начинаем с нуля при разработке в облаке Copilot: начать разработку с минимальным использованием кода стало проще
boy illustration
[Серия Docker] Docker создает мультиплатформенные образы: практика архитектуры Arm64
boy illustration
Обновление новых возможностей coze | Я использовал coze для создания апплета помощника по исправлению домашних заданий по математике
boy illustration
Советы по развертыванию Nginx: практическое создание статических веб-сайтов на облачных серверах
boy illustration
Feiniu fnos использует Docker для развертывания личного блокнота Notepad
boy illustration
Сверточная нейронная сеть VGG реализует классификацию изображений Cifar10 — практический опыт Pytorch
boy illustration
Начало работы с EdgeonePages — новым недорогим решением для хостинга веб-сайтов
boy illustration
[Зона легкого облачного игрового сервера] Управление игровыми архивами
boy illustration
Развертывание SpringCloud-проекта на базе Docker и Docker-Compose