Merge pull request #207 from sljeff/fix-close-stubext

fix close stubext
shanbay · Sep 8, 2020 · d940c72 · d940c72
2 parents f7efc4b + 176fd55
commit d940c72
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/extensions/stubext/ext.go b/extensions/stubext/ext.go
@@ -55,7 +55,12 @@ func (d *StubExt) Application() *gobay.Application { return d.app }
 
 func (d *StubExt) Object() interface{} { return d }
 
-func (d *StubExt) Close() error { return d.conn.Close() }
+func (d *StubExt) Close() error {
+	if d.Mocked {
+		return nil
+	}
+	return d.conn.Close()
+}
 
 func (d *StubExt) Init(app *gobay.Application) error {
 	if d.NS == "" {