コード例 #1
0
ファイル: iterate_test.go プロジェクト: itsabot/abot
func TestMain(m *testing.M) {
	err := os.Setenv("ABOT_ENV", "test")
	if err != nil {
		log.Fatal("failed to set ABOT_ENV.", err)
	}
	r, err = core.NewServer()
	if err != nil {
		log.Fatal("failed to start abot server.", err)
	}
	p, err = plugin.New("testplugin")
	if err != nil {
		log.Fatal("failed to build test plugin.", err)
	}
	p.Config.Name = "testplugin"
	p.Trigger = &dt.StructuredInput{
		Commands: []string{"get"},
		Objects:  []string{"result"},
	}
	plugin.SetStates(p, [][]dt.State{
		[]dt.State{
			{
				OnEntry: func(in *dt.Msg) string {
					return "entered first"
				},
				OnInput: func(in *dt.Msg) {},
				Complete: func(in *dt.Msg) (bool, string) {
					return true, ""
				},
			},
		},
		Iterate(p, "", OptsIterate{
			IterableMemKey:  keyMem,
			ResultMemKeyIdx: keyRes,
		}),
		[]dt.State{
			{
				OnEntry: func(in *dt.Msg) string {
					return "Great!"
				},
				OnInput: func(in *dt.Msg) {},
				Complete: func(in *dt.Msg) (bool, string) {
					return true, ""
				},
			},
		},
	})
	p.SM.SetOnReset(func(in *dt.Msg) {
		p.DeleteMemory(in, keyMem)
		p.DeleteMemory(in, keyRes)
		ResetIterate(p, in)
	})
	if err = plugin.Register(p); err != nil {
		log.Fatal("failed to register test plugin.", err)
	}
	cleanup()
	exitVal := m.Run()
	cleanup()
	os.Exit(exitVal)
}
コード例 #2
0
ファイル: test.go プロジェクト: itsabot/abot
// TestPrepare sets the ABOT_ENV, cleans out the test DB, and initializes the
// server for testing.
func TestPrepare() *httprouter.Router {
	if err := os.Setenv("ABOT_ENV", "test"); err != nil {
		log.Fatal(err)
	}
	r, err := core.NewServer()
	if err != nil {
		log.Fatal("failed to start abot server.", err)
	}
	TestCleanup()
	return r
}