Safe Fake driver Register

Author: Dmitriy Seredenko

README.md

@@ -22,7 +22,7 @@ List of features in the library:
 #### Install
 
 ```
-go get github.com/selvatico/go-mocket
+go get github.com/Selvatico/go-mocket
 ```
 
 #### Usage
@@ -39,18 +39,18 @@ Somewhere in you code to setup a tests
 ```go
 import (
     "database/sql"
-    mocket "github.com/selvatico/go-mocket"
+    mocket "github.com/Selvatico/go-mocket"
     "github.com/jinzhu/gorm"
 )
 
 func SetupTests() {
-    sql.Register("fake_test", mocket.FakeDriver{})
+    mocket.Catcher.Register()
     // GORM
-    db, err := gorm.Open("fake_test", "connection_string") // Could be any connection string
+    db, err := gorm.Open(mocket.DRIVER_NAME, "any_string") // Could be any connection string
     app.DB = db // assumption that it will be used everywhere the same
     //OR 
     // Regular sql package usage
-    db, err := sql.Open(driver, source)
+    db, err := sql.Open(mocket.DRIVER_NAME, "any_string")
 }
 ```
 
@@ -61,7 +61,7 @@ Now if use singleton instance of DB it will use everywhere mocked connection.
 ###### Example of mocking by pattern
 
 ```go
-import mocket "github.com/selvatico/go-mocket"
+import mocket "github.com/Selvatico/go-mocket"
 import "net/http/httptest"
 
 func TestHandler(t *testing.T) {

doc.go

@@ -5,14 +5,14 @@ First you need to activate package somewhere in your tests code like this
 
 	import (
     	"database/sql"
-    	mocket "github.com/selvatico/go-mocket"
+    	mocket "github.com/Selvatico/go-mocket"
     	"github.com/jinzhu/gorm"
 	)
 
 	var DB *gorm.DB
 
 	func SetupTests() {
-    	sql.Register("fake_test", mocket.FakeDriver{})
+    	mocket.Catcher.Register()
     	// GORM
     	db, err := gorm.Open("fake_test", "connection_string") // Could be any connection string
     	DB = db
@@ -55,11 +55,11 @@ Somewhere in you tests:
 		"log"
 		"testing"
 		"database/sql"
-		mocket "github.com/selvatico/go-mocket"
+		mocket "github.com/Selvatico/go-mocket"
 	)
 
 	func TestResponses (t *testing.T) {
-		sql.Register("fake_test", FakeDriver{})
+		mocket.Catcher.Register()
 		db, _ := sql.Open("fake_test", "connection_string") // Could be any connection string
 		DB = db
 

response.go

@@ -6,6 +6,10 @@ import (
 	"log"
 	"reflect"
 	"strings"
+	"database/sql"
+)
+const (
+	DRIVER_NAME = "MOCK_FAKE_DRIVER"
 )
 
 //Catcher is global instance of Catcher used for attaching all mocks to connection
@@ -18,6 +22,17 @@ type MockCatcher struct {
 	PanicOnEmptyResponse bool            // If not response matches - do we need to panic?
 }
 
+// Register safely register FakeDriver
+func (mc *MockCatcher) Register() {
+	driversList := sql.Drivers()
+	for _, name := range driversList {
+		if name == DRIVER_NAME {
+			return
+		}
+	}
+	sql.Register(DRIVER_NAME, FakeDriver{})
+}
+
 // Attach several mocks to MockCather. Could be useful to attach mocks from some factories of mocks
 func (mc *MockCatcher) Attach(fr []*FakeResponse) {
 	mc.Mocks = append(mc.Mocks, fr...)