graphql-engine/cli/commands/seed_create_test.go
Mohd Bilal 8d18a55212 cli: move AddDatabaseToHasura to testutil package
PR-URL: https://github.com/hasura/graphql-engine-mono/pull/6069
GitOrigin-RevId: f7a3b98685c16fec613719eaa05024cd3c9cd3d4
2022-09-27 11:16:22 +00:00

92 lines
3.1 KiB
Go

package commands
import (
"fmt"
"os"
"path/filepath"
"github.com/Pallinder/go-randomdata"
"github.com/hasura/graphql-engine/cli/v2/internal/testutil"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
. "github.com/onsi/gomega/gexec"
)
var testSeedCreate = func(projectDirectory string, globalFlags []string) {
upSql := `CREATE TABLE "public"."table1" ("id" serial NOT NULL, PRIMARY KEY ("id") );
INSERT INTO public.table1 (id) VALUES (1);
INSERT INTO public.table1 (id) VALUES (2);
INSERT INTO public.table1 (id) VALUES (3);
INSERT INTO public.table1 (id) VALUES (4);`
downSql := `DROP TABLE "public"."table1";`
testutil.RunCommandAndSucceed(testutil.CmdOpts{
Args: append([]string{"migrate", "create", "table1", "--up-sql", upSql, "--down-sql", downSql}, globalFlags...),
WorkingDirectory: projectDirectory,
})
testutil.RunCommandAndSucceed(testutil.CmdOpts{
Args: append([]string{"migrate", "apply"}, globalFlags...),
WorkingDirectory: projectDirectory,
})
session := testutil.Hasura(testutil.CmdOpts{
Args: append([]string{"seed", "create", "table_seed", "--from-table", "table1"}, globalFlags...),
WorkingDirectory: projectDirectory,
})
Eventually(session, timeout).Should(Exit(0))
Expect(session.Err.Contents()).Should(ContainSubstring("created seed file successfully"))
}
var _ = Describe("hasura seed create", func() {
Context("config v3", func() {
var teardown func()
var projectDirectory string
var sourceName string
BeforeEach(func() {
projectDirectory = testutil.RandDirName()
hgeEndPort, teardownHGE := testutil.StartHasuraWithMetadataDatabase(GinkgoT(), testutil.HasuraDockerImage)
hgeEndpoint := fmt.Sprintf("http://0.0.0.0:%s", hgeEndPort)
sourceName = randomdata.SillyName()
_, teardownPG := testutil.AddDatabaseToHasura(GinkgoT(), hgeEndpoint, sourceName, "postgres")
testutil.RunCommandAndSucceed(testutil.CmdOpts{
Args: []string{"init", projectDirectory},
})
editEndpointInConfig(filepath.Join(projectDirectory, defaultConfigFilename), hgeEndpoint)
teardown = func() {
teardownPG()
teardownHGE()
os.RemoveAll(projectDirectory)
}
})
AfterEach(func() { teardown() })
It("can create seeds in config v3", func() {
testSeedCreate(projectDirectory, []string{"--database-name", sourceName})
})
})
Context("config v2", func() {
var teardown func()
var projectDirectoryConfigV2 string
BeforeEach(func() {
projectDirectoryConfigV2 = testutil.RandDirName()
hgeEndPort, teardownHGE := testutil.StartHasura(GinkgoT(), testutil.HasuraDockerImage)
hgeEndpoint := fmt.Sprintf("http://0.0.0.0:%s", hgeEndPort)
testutil.RunCommandAndSucceed(testutil.CmdOpts{
Args: []string{"init", projectDirectoryConfigV2, "--version", "2"},
})
editEndpointInConfig(filepath.Join(projectDirectoryConfigV2, defaultConfigFilename), hgeEndpoint)
teardown = func() {
teardownHGE()
os.RemoveAll(projectDirectoryConfigV2)
}
})
AfterEach(func() { teardown() })
It("can create seeds in config v2", func() {
testSeedCreate(projectDirectoryConfigV2, nil)
})
})
})