git.iko.soy/crush
on main help sign in
refactor: make test code a little bit cleaner
133f698Andrey Nering
internal/agent/common_test.go+8−5
compare with main
unifiedsplit
@@ -101,23 +101,26 @@ func zAIBuilder(model string) builderFunc {
101}
101}
102
102
103func testEnv(t *testing.T) env {
103func testEnv(t *testing.T) env {
104− testDir := filepath.Join("/tmp/crush-test/", t.Name())
104+ workingDir := filepath.Join("/tmp/crush-test/", t.Name())
105− os.RemoveAll(testDir)
105+ os.RemoveAll(workingDir)
106+
106− err := os.MkdirAll(testDir, 0o755)
107+ err := os.MkdirAll(workingDir, 0o755)
107 require.NoError(t, err)
108 require.NoError(t, err)
108− workingDir := testDir
109+
109 conn, err := db.Connect(t.Context(), t.TempDir())
110 conn, err := db.Connect(t.Context(), t.TempDir())
110 require.NoError(t, err)
111 require.NoError(t, err)
112+
111 q := db.New(conn)
113 q := db.New(conn)
112 sessions := session.NewService(q)
114 sessions := session.NewService(q)
113 messages := message.NewService(q)
115 messages := message.NewService(q)
116+
114 permissions := permission.NewPermissionService(workingDir, true, []string{})
117 permissions := permission.NewPermissionService(workingDir, true, []string{})
115 history := history.NewService(q, conn)
118 history := history.NewService(q, conn)
116 lspClients := csync.NewMap[string, *lsp.Client]()
119 lspClients := csync.NewMap[string, *lsp.Client]()
117
120
118 t.Cleanup(func() {
121 t.Cleanup(func() {
119 conn.Close()
122 conn.Close()
120− os.RemoveAll(testDir)
123+ os.RemoveAll(workingDir)
121 })
124 })
122
125
123 return env{
126 return env{