如何在使用Go Buffalo框架的开发环境中，使用TOML fixtures来填充数据库？

huangapple
go评论128阅读模式
英文:

How can I use TOML fixtures to seed my database in a development environment using the Go Buffalo framework?

问题

我正在尝试使用 TOML fixtures 在 Go Buffalo 框架中填充我的开发数据库。然而，我在如何做到这一点方面遇到了困难，找不到清晰的示例或文档。

英文:

I am trying to seed my development database in the Go Buffalo framework using TOML fixtures. However, I am having trouble finding a clear example or documentation on how to do this.

答案1

得分: 4

要种子数据库，您可以使用grifts。在创建新应用程序时，应该在grifts/db.go中生成一个虚拟的占位符grift，看起来像这样：

  1. package grifts
  3. import "github.com/gobuffalo/grift/grift"
  5. var _ = grift.Namespace("db", func() {
  7. 	grift.Desc("seed", "Seeds a database")
  8. 	grift.Add("seed", func(c *grift.Context) error {
  9. 		// 在这里添加数据库种子内容
  10. 		return nil
  11. 	})
  13. })

我尝试在grift中使用model.LoadFixture("seed widgets")，但由于不在测试环境中，所以出现了一些恐慌。我认为直接从toml fixtures中支持数据库种子不是直接支持的，但这将是一个有用的功能。然而，通过查看LoadFixture的代码，我们可以构建自己的fixture加载：

假设您有一个名为name = "seed widgets"的fixture场景：

  1. package grifts
  3. import (
  4. 	"fmt"
  5. 	"os"
  6. 	"strings"
  8. 	"github.com/gobuffalo/grift/grift"
  9. 	"github.com/gobuffalo/suite/v4"
  10. 	"github.com/gobuffalo/suite/v4/fix"
  11. )
  13. var _ = grift.Namespace("db", func() {
  15. 	grift.Desc("seed", "Seeds a database")
  16. 	grift.Add("seed", func(c *grift.Context) error {
  17. 		// DB连接将连接到在`GO_ENV`中设置的环境（默认为`test`）
  18. 		// 在您的`.env`文件中将此环境变量设置为`development`
  20. 		// 注意：最好将种子fixtures放在不同的目录中
  21. 		// 以分离关注点
  22. 		model, err := suite.NewModelWithFixtures(os.DirFS("./fixtures"))
  23. 		if err != nil {
  24. 			return err
  25. 		}
  27. 		sc, err := fix.Find("seed widgets")
  28. 		if err != nil {
  29. 			return err
  30. 		}
  32. 		for _, table := range sc.Tables {
  33. 			for _, row := range table.Row {
  34. 				q := "insert into " + table.Name
  35. 				keys := []string{}
  36. 				skeys := []string{}
  37. 				for k := range row {
  38. 					keys = append(keys, k)
  39. 					skeys = append(skeys, ":"+k)
  40. 				}
  42. 				q = q + fmt.Sprintf(" (%s) values (%s)", strings.Join(keys, ","), strings.Join(skeys, ","))
  43. 				if _, err = model.DB.Store.NamedExec(q, row); err != nil {
  44. 					return err
  45. 				}
  46. 			}
  47. 		}
  49. 		return nil
  50. 	})
  52. })
英文:

To seed a database, you can use grifts. When creating a new application, there should be a dummy placeholder grift generated at grifts/db.go that looks like:

  1. package grifts
  3. import "github.com/gobuffalo/grift/grift"
  5. var _ = grift.Namespace("db", func() {
  7. 	grift.Desc("seed", "Seeds a database")
  8. 	grift.Add("seed", func(c *grift.Context) error {
  9. 		// Add DB seeding stuff here
  10. 		return nil
  11. 	})
  13. })

I attempted to use model.LoadFixture("seed widgets") in a grift, but there is some panic because it is not in a test environment. I think DB seeding from toml fixtures is not supported directly but that would be a useful feature. However looking at the code from LoadFixture, we can construct our own fixture loading:

Assuming you have a fixture with a scenario named name = "seed widgets":

  1. package grifts
  3. import (
  4. 	"fmt"
  5. 	"os"
  6. 	"strings"
  8. 	"github.com/gobuffalo/grift/grift"
  9. 	"github.com/gobuffalo/suite/v4"
  10. 	"github.com/gobuffalo/suite/v4/fix"
  11. )
  13. var _ = grift.Namespace("db", func() {
  15. 	grift.Desc("seed", "Seeds a database")
  16. 	grift.Add("seed", func(c *grift.Context) error {
  17. 		// The DB connection will connect to the environment set in `GO_ENV` (defaults to `test`)
  18. 		// Set this environment variable in your `.env` file to `development`
  20. 		// NOTE: it may be better to put seed fixtures in a different directory
  21. 		//       to seperate concerns
  22. 		model, err := suite.NewModelWithFixtures(os.DirFS("./fixtures"))
  23. 		if err != nil {
  24. 			return err
  25. 		}
  27. 		sc, err := fix.Find("seed widgets")
  28. 		if err != nil {
  29. 			return err
  30. 		}
  32. 		for _, table := range sc.Tables {
  33. 			for _, row := range table.Row {
  34. 				q := "insert into " + table.Name
  35. 				keys := []string{}
  36. 				skeys := []string{}
  37. 				for k := range row {
  38. 					keys = append(keys, k)
  39. 					skeys = append(skeys, ":"+k)
  40. 				}
  42. 				q = q + fmt.Sprintf(" (%s) values (%s)", strings.Join(keys, ","), strings.Join(skeys, ","))
  43. 				if _, err = model.DB.Store.NamedExec(q, row); err != nil {
  44. 					return err
  45. 				}
  46. 			}
  47. 		}
  49. 		return nil
  50. 	})
  52. })

答案2

得分: 1

你可以在这里找到关于测试和夹具的所有所需信息：https://github.com/gobuffalo/suite

这是一个关于操作和模型测试的基本示例：

actions/actions_test.go

  1. package actions
  3. import (
  4.     "os"
  5.     "testing"
  7.     "github.com/gobuffalo/suite/v4"
  8. )
  10. type ActionSuite struct {
  11.     *suite.Action
  12. }
  14. func Test_ActionSuite(t *testing.T) {
  15.     action, err := suite.NewActionWithFixtures(App(), os.DirFS("../fixtures"))
  16.     if err != nil {
  17.         t.Fatal(err)
  18.     }
  20.     as := &ActionSuite{
  21.         Action: action,
  22.     }
  23.     suite.Run(t, as)
  24. }

models/models_test.go

  1. package models
  3. import (
  4.     "os"
  5.     "testing"
  7.     "github.com/gobuffalo/suite/v4"
  8. )
  10. type ModelSuite struct {
  11.     *suite.Model
  12. }
  14. func Test_ModelSuite(t *testing.T) {
  15.     model, err := suite.NewModelWithFixtures(os.DirFS("../fixtures"))
  16.     if err != nil {
  17.         t.Fatal(err)
  18.     }
  20.     as := &ModelSuite{
  21.         Model:  model,
  22.     }
  23.     suite.Run(t, as)
  24. }

fixtures/widgets.toml

  1. [[scenario]]
  2. name = "lots of widgets"
  4.   [[scenario.table]]
  5.     name = "widgets"
  7.     [[scenario.table.row]]
  8.       id = "<%= uuidNamed("widget") %>"
  9.       name = "This is widget #1"
  10.       body = "some widget body"
  11.       created_at = "<%= now() %>"
  12.       updated_at = "<%= now() %>"
  14.     [[scenario.table.row]]
  15.       id = "<%= uuid() %>"
  16.       name = "This is widget #2"
  17.       body = "some widget body"
  18.       created_at = "<%= now() %>"
  19.       updated_at = "<%= now() %>"
  21.   [[scenario.table]]
  22.     name = "users"
  24.     [[scenario.table.row]]
  25.       id = "<%= uuid() %>"
  26.       name = "Mark Bates"
  27.       admin = true
  28.       price = 19.99
  29.       widget_id = "<%= uuidNamed("widget") %>"
  30.       created_at = "<%= now() %>"
  31.       updated_at = "<%= now() %>"

actions/widgets_test.go

  1. func (as *ActionSuite) Test_WidgetsResource_List() {
  2.     as.LoadFixture("lots of widgets")
  3.     res := as.HTML("/widgets").Get()
  5.     body := res.Body.String()
  6.     as.Contains(body, "widget #1")
  7.     as.Contains(body, "widget #2")
  8. }

models/widget_test.go

  1. func (ms *ModelSuite) Test_Widget_SomeModelMethod() {
  2.     ms.LoadFixture("lots of widgets")
  4.     var widgets []Widget
  5.     if err := DB.All(&widgets); err != nil {
  6.         ms.Fail(err.Error())
  7.     }
  9.     for _, widget := range widgets {
  10.         ms.Equal("some value", widget.SomeModelMethod())
  11.     }
  12. }

database.yml

  1. test:
  2.   url: {{envOr "TEST_DATABASE_URL" "postgres://postgres@127.0.0.1:5432/medicare_65_quote_test?sslmode=disable"}}
英文:

You can find all the information you need about testing and fixtures here: https://github.com/gobuffalo/suite

Here is a basic example of actions and model tests:

actions/actions_test.go

  1. package actions
  3. import (
  4.     &quot;os&quot;
  5.     &quot;testing&quot;
  7.     &quot;github.com/gobuffalo/suite/v4&quot;
  8. )
  10. type ActionSuite struct {
  11.     *suite.Action
  12. }
  14. func Test_ActionSuite(t *testing.T) {
  15.     action, err := suite.NewActionWithFixtures(App(), os.DirFS(&quot;../fixtures&quot;))
  16.     if err != nil {
  17.         t.Fatal(err)
  18.     }
  20.     as := &amp;ActionSuite{
  21.         Action: action,
  22.     }
  23.     suite.Run(t, as)
  24. }

models/models_test.go

  1. package models
  3. import (
  4.     &quot;os&quot;
  5.     &quot;testing&quot;
  7.     &quot;github.com/gobuffalo/suite/v4&quot;
  8. )
  10. type ModelSuite struct {
  11.     *suite.Model
  12. }
  14. func Test_ModelSuite(t *testing.T) {
  15.     model, err := suite.NewModelWithFixtures(os.DirFS(&quot;../fixtures&quot;))
  16.     if err != nil {
  17.         t.Fatal(err)
  18.     }
  20. 	as := &amp;ModelSuite{
  21. 		Model:  model,
  22.     }
  23.     suite.Run(t, as)
  24. }

fixtures/widgets.toml

  1. [[scenario]]
  2. name = &quot;lots of widgets&quot;
  4.   [[scenario.table]]
  5.     name = &quot;widgets&quot;
  7.     [[scenario.table.row]]
  8.       id = &quot;&lt;%= uuidNamed(&quot;widget&quot;) %&gt;&quot;
  9.       name = &quot;This is widget #1&quot;
  10.       body = &quot;some widget body&quot;
  11.       created_at = &quot;&lt;%= now() %&gt;&quot;
  12.       updated_at = &quot;&lt;%= now() %&gt;&quot;
  14.     [[scenario.table.row]]
  15.       id = &quot;&lt;%= uuid() %&gt;&quot;
  16.       name = &quot;This is widget #2&quot;
  17.       body = &quot;some widget body&quot;
  18.       created_at = &quot;&lt;%= now() %&gt;&quot;
  19.       updated_at = &quot;&lt;%= now() %&gt;&quot;
  21.   [[scenario.table]]
  22.     name = &quot;users&quot;
  24.     [[scenario.table.row]]
  25.       id = &quot;&lt;%= uuid() %&gt;&quot;
  26.       name = &quot;Mark Bates&quot;
  27.       admin = true
  28.       price = 19.99
  29.       widget_id = &quot;&lt;%= uuidNamed(&quot;widget&quot;) %&gt;&quot;
  30.       created_at = &quot;&lt;%= now() %&gt;&quot;
  31.       updated_at = &quot;&lt;%= now() %&gt;&quot;

actions/widgets_test.go

  1. func (as *ActionSuite) Test_WidgetsResource_List() {
  2.     as.LoadFixture(&quot;lots of widgets&quot;)
  3.     res := as.HTML(&quot;/widgets&quot;).Get()
  5.     body := res.Body.String()
  6.     as.Contains(body, &quot;widget #1&quot;)
  7.     as.Contains(body, &quot;widget #2&quot;)
  8. }

models/widget_test.go

  1. func (ms *ModelSuite) Test_Widget_SomeModelMethod() {
  2. 	ms.LoadFixture(&quot;lots of widgets&quot;)
  4. 	var widgets []Widget
  5. 	if err := DB.All(&amp;widgets); err != nil {
  6. 		ms.Fail(err.Error())
  7. 	}
  9. 	for _, widget := range widgets {
  10. 		ms.Equal(&quot;some value&quot;, widget.SomeModelMethod())
  11. 	}
  12. }

database.yml

  1. test:
  2.   url: {{envOr &quot;TEST_DATABASE_URL&quot; &quot;postgres://postgres@127.0.0.1:5432/medicare_65_quote_test?sslmode=disable&quot;}}

huangapple
  • 本文由 发表于 2023年1月19日 00:56:21
  • 转载请务必保留本文链接：https://go.coder-hub.com/75162669.html
  • buffalo
  • go
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定