草庐IT

sql - 删除用于测试套件的所有数据库记录的最佳方法是什么？

coder 2024-07-06 原文

我有一个测试套件使用从 YAML 文件读取的种子污染我的数据库。

我想知道是否有一种方法可以在运行测试后清理我的数据库(删除用于测试套件的所有记录)。 

// Open db and returns pointer and closer func
func prepareMySQLDB(t *testing.T) (db *sql.DB, closer func() error) {
    db, err := sql.Open("mysql", "user:pass@/database")
    if err != nil {
        t.Fatalf("open mysql connection: %s", err)
    }
    return db, db.Close
}
// Pollute my database
func polluteDb(db *sql.DB, t *testing.T) {
    seed, err := os.Open("seed.yml")
    if err != nil {
        t.Fatalf("failed to open seed file: %s", err)
    }
    defer seed.Close()
    p := polluter.New(polluter.MySQLEngine(db))
    if err := p.Pollute(seed); err != nil {
        t.Fatalf("failed to pollute: %s", err)
    }
}

func TestAllUsers(t *testing.T) {
    t.Parallel()

    db, closeDb := prepareMySQLDB(t)
    defer closeDb()

    polluteDb(db, t)

    users, err := AllUsersD(db)
    if err != nil {
        t.Fatal("AllUsers() failed")
    }

    got := users[0].Email
    if got != "myemail@gmail.com" {
        t.Errorf("AllUsers().Email = %s; want myemail@gmail.com", got)
    }

    got1 := len(users)
    if got1 != 1 {
        t.Errorf("len(AllUsers()) = %d; want 1", got1)
    }
}


// Test I'm interested in
func TestAddUser(t *testing.T) {
    t.Parallel()

    db, closeDb := prepareMySQLDB(t)
    defer closeDb()

    polluteDb(db, t)

    user, err := AddUser(...)
    if err != nil {
        t.Fatal("AddUser() failed")
    }

    //how can I clean my database after this? 
}

我应该检索最后插入到 TestAddUser() 中的 ID 并手动删除该行，还是有任何其他方法来保存我的数据库状态并在之后检索它？

正如我所说的，我是 Go 的新手，因此非常感谢对我的代码或任何其他评论的评论。

最佳答案

最好的方法通常是先使用一个事务，然后再使用 ROLLBACK，这样一开始就不会提交它们。

github.com/DATA-DOG/go-txdb package 对此有很大帮助。

最终代码:

import (
    "database/sql"
    "os"
    "testing"

    txdb "github.com/DATA-DOG/go-txdb"

    "github.com/romanyx/polluter"
)

//mostly sql tests
func init() {
    txdb.Register("txdb", "mysql", "root:root@/betell_rest")
}

func TestAddUser(t *testing.T) {
    db, err := sql.Open("txdb", "root:root@/betell_rest")
    if err != nil {
        t.Fatal(err)
    }
    defer db.Close()

    users, _ := AllUsers(db)
    userscount := len(users)
    err = AddUser(db, "bla@gmail.com", "pass")
    if err != nil {
        t.Fatal("AddUser() failed")
    }

    users, _ = AllUsers(db)
    if (userscount + 1) != len(users) {
        t.Fatal("AddUser() failed to write in database")
    }
}

注意:你也可以将 db 传递给你的污染者，这样你就不会影响你的数据库。

关于sql - 删除用于测试套件的所有数据库记录的最佳方法是什么？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/57546743/

sql的34errsectionunit-testinggo

有关sql - 删除用于测试套件的所有数据库记录的最佳方法是什么？的更多相关文章

  1. ruby - 为什么我可以在 Ruby 中使用 Object#send 访问私有(private)/ protected 方法？ - 2

    类classAprivatedeffooputs:fooendpublicdefbarputs:barendprivatedefzimputs:zimendprotecteddefdibputs:dibendendA的实例a=A.new测试a.foorescueputs:faila.barrescueputs:faila.zimrescueputs:faila.dibrescueputs:faila.gazrescueputs:fail测试输出failbarfailfailfail.发送测试[:foo,:bar,:zim,:dib,:gaz].each{|m|a.send(m)resc

  2. ruby-on-rails - 使用 Ruby on Rails 进行自动化测试 - 最佳实践 - 2

    很好奇，就使用ruby​​onrails自动化单元测试而言，你们正在做什么？您是否创建了一个脚本来在cron中运行rake作业并将结果邮寄给您？git中的预提交Hook？只是手动调用？我完全理解测试，但想知道在错误发生之前捕获错误的最佳实践是什么。让我们理所当然地认为测试本身是完美无缺的，并且可以正常工作。下一步是什么以确保他们在正确的时间将可能有害的结果传达给您？ 最佳答案 不确定您到底想听什么，但是有几个级别的自动代码库控制:在处理某项功能时，您可以使用类似autotest的内容获得关于哪些有效，哪些无效的即时反馈。要确保您的提

  3. ruby-on-rails - Rails - 子类化模型的设计模式是什么？ - 2

    我有一个模型:classItem项目有一个属性“商店”基于存储的值，我希望Item对象对特定方法具有不同的行为。Rails中是否有针对此的通用设计模式？如果方法中没有大的if-else语句，这是如何干净利落地完成的？ 最佳答案 通常通过Single-TableInheritance. 关于ruby-on-rails-Rails-子类化模型的设计模式是什么？，我们在StackOverflow上找到一个类似的问题： https://stackoverflow.co

  4. ruby-on-rails - Rails 常用字符串(用于通知和错误信息等) - 2

    大约一年前，我决定确保每个包含非唯一文本的Flash通知都将从模块中的方法中获取文本。我这样做的最初原因是为了避免一遍又一遍地输入相同的字符串。如果我想更改措辞，我可以在一个地方轻松完成，而且一遍又一遍地重复同一件事而出现拼写错误的可能性也会降低。我最终得到的是这样的:moduleMessagesdefformat_error_messages(errors)errors.map{|attribute,message|"Error:#{attribute.to_s.titleize}#{message}."}enddeferror_message_could_not_find(obje

  5. ruby - 如何以所有可能的方式将字符串拆分为长度最多为 3 的连续子字符串？ - 2

    我试图获取一个长度在1到10之间的字符串，并输出将字符串分解为大小为1、2或3的连续子字符串的所有可能方式。例如:输入:123456将整数分割成单个字符，然后继续查找组合。该代码将返回以下所有数组。[1,2,3,4,5,6][12,3,4,5,6][1,23,4,5,6][1,2,34,5,6][1,2,3,45,6][1,2,3,4,56][12,34,5,6][12,3,45,6][12,3,4,56][1,23,45,6][1,2,34,56][1,23,4,56][12,34,56][123,4,5,6][1,234,5,6][1,2,345,6][1,2,3,456][123

  6. ruby - 什么是填充的 Base64 编码字符串以及如何在 ruby​​ 中生成它们？ - 2

    我正在使用的第三方API的文档状态:"[O]urAPIonlyacceptspaddedBase64encodedstrings."什么是“填充的Base64编码字符串”以及如何在Ruby中生成它们。下面的代码是我第一次尝试创建转换为Base64的JSON格式数据。xa=Base64.encode64(a.to_json) 最佳答案 他们说的padding其实就是Base64本身的一部分。它是末尾的“=”和“==”。Base64将3个字节的数据包编码为4个编码字符。所以如果你的输入数据有长度n和n%3=1=>"=="末尾用于填充n%

  7. ruby - 解析 RDFa、微数据等的最佳方式是什么，使用统一的模式/词汇(例如 schema.org)存储和显示信息 - 2

    我主要使用Ruby来执行此操作，但到目前为止我的攻击计划如下:使用gemsrdf、rdf-rdfa和rdf-microdata或mida来解析给定任何URI的数据。我认为最好映射到像schema.org这样的统一模式，例如使用这个yaml文件，它试图描述数据词汇表和opengraph到schema.org之间的转换:#SchemaXtoschema.orgconversion#data-vocabularyDV:name:namestreet-address:streetAddressregion:addressRegionlocality:addressLocalityphoto:i

  8. ruby - 为什么 4.1%2 使用 Ruby 返回 0.0999999999999996？但是 4.2%2==0.2 - 2

    为什么4.1%2返回0.0999999999999996？但是4.2%2==0.2。 最佳答案 参见此处:WhatEveryProgrammerShouldKnowAboutFloating-PointArithmetic实数是无限的。计算机使用的位数有限(今天是32位、64位)。因此计算机进行的浮点运算不能代表所有的实数。0.1是这些数字之一。请注意，这不是与Ruby相关的问题，而是与所有编程语言相关的问题，因为它来自计算机表示实数的方式。 关于ruby-为什么4.1%2使用Ruby返

  9. ruby-on-rails - 如何从 format.xml 中删除 <hash></hash> - 2

    我有一个对象has_many应呈现为xml的子对象。这不是问题。我的问题是我创建了一个Hash包含此数据，就像解析器需要它一样。但是rails自动将整个文件包含在.........我需要摆脱type="array"和我该如何处理？我没有在文档中找到任何内容。 最佳答案 我遇到了同样的问题；这是我的XML:我在用这个:entries.to_xml将散列数据转换为XML，但这会将条目的数据包装到中所以我修改了:entries.to_xml(root:"Contacts")但这仍然将转换后的XML包装在“联系人”中，将我的XML代码修改为

  10. ruby - 我可以使用 Ruby 从 CSV 中删除列吗？ - 2

    查看Ruby的CSV库的文档，我非常确定这是可能且简单的。我只需要使用Ruby删除CSV文件的前三列，但我没有成功运行它。 最佳答案 csv_table=CSV.read(file_path_in,:headers=>true)csv_table.delete("header_name")csv_table.to_csv#=>ThenewCSVinstringformat检查CSV::Table文档:http://ruby-doc.org/stdlib-1.9.2/libdoc/csv/rdoc/CSV/Table.html

随机推荐