summaryrefslogtreecommitdiffstats
path: root/app/app.go
diff options
context:
space:
mode:
authorPierre de La Morinerie <kemenaran@gmail.com>2018-02-12 22:16:32 +0530
committerChristopher Speller <crspeller@gmail.com>2018-02-12 08:46:32 -0800
commit07fd7aeeb8eb2b198b01b713a4ab57f6352faef2 (patch)
tree002eb93e9258b91553f3784629a1acdff6f6fc49 /app/app.go
parent3e0c3eff9f2ddec241cdb3f7a91230fd7c51a5f6 (diff)
downloadchat-07fd7aeeb8eb2b198b01b713a4ab57f6352faef2.tar.gz
chat-07fd7aeeb8eb2b198b01b713a4ab57f6352faef2.tar.bz2
chat-07fd7aeeb8eb2b198b01b713a4ab57f6352faef2.zip
Add tests for the `platform server` command (#8231)
* Cleanup app state on initialization error When returning an initialization error, the app state was not cleaned up. This is especially visible during tests, as `appCount` is not decremented, and makes the new app initialization fail. * Test the `platform server` command As the `platform server` command only exits when interrupted by a signal, it is not possible to test it as the other cobra commands. Instead we directly test the actual command function. The internal command handler is slighly refactored to take a channel in argument, and registers it as the signal handler. Nothing very different—except than controlling this channel from the outside allows the test to send the system signal itself, thus preventing the server to run forever.
Diffstat (limited to 'app/app.go')
-rw-r--r--app/app.go11
1 files changed, 9 insertions, 2 deletions
diff --git a/app/app.go b/app/app.go
index dd5deb342..636f0a428 100644
--- a/app/app.go
+++ b/app/app.go
@@ -86,7 +86,7 @@ var appCount = 0
// New creates a new App. You must call Shutdown when you're done with it.
// XXX: For now, only one at a time is allowed as some resources are still shared.
-func New(options ...Option) (*App, error) {
+func New(options ...Option) (outApp *App, outErr error) {
appCount++
if appCount > 1 {
panic("Only one App should exist at a time. Did you forget to call Shutdown()?")
@@ -103,6 +103,11 @@ func New(options ...Option) (*App, error) {
clientConfig: make(map[string]string),
licenseListeners: map[string]func(){},
}
+ defer func() {
+ if outErr != nil {
+ app.Shutdown()
+ }
+ }()
for _, option := range options {
option(app)
@@ -182,7 +187,9 @@ func (a *App) Shutdown() {
a.ShutDownPlugins()
a.WaitForGoroutines()
- a.Srv.Store.Close()
+ if a.Srv.Store != nil {
+ a.Srv.Store.Close()
+ }
a.Srv = nil
if a.htmlTemplateWatcher != nil {