-
Notifications
You must be signed in to change notification settings - Fork 36
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #10 from vallieres/documentation-move
Documentation Move for Collaboration
- Loading branch information
Showing
1 changed file
with
274 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,274 @@ | ||
# go-mocket Documentation | ||
|
||
## Setting Up Tests | ||
|
||
*** | ||
|
||
To set up tests, you need to register the driver and override the DB instance used across the code base. | ||
```go | ||
import ( | ||
"database/sql" | ||
mocket "github.com/selvatico/go-mocket" | ||
"github.com/jinzhu/gorm" | ||
) | ||
|
||
func SetupTests() *sql.DB { // or *gorm.DB | ||
mocket.Catcher.Register() // Safe register. Allowed multiple calls to save | ||
// GORM | ||
db, err := gorm.Open(mocket.DRIVER_NAME, "connection_string") // Can be any connection string | ||
DB = db | ||
|
||
// OR | ||
// Regular sql package usage | ||
db, err := sql.Open(mocket.DRIVER_NAME, "connection_string") | ||
|
||
return db | ||
} | ||
``` | ||
|
||
In the snippet above, we intentionally skipped assigning to proper variable DB instance. One of the assumptions is that the project has one DB instance at the time, overriding it with FakeDriver will do the job. | ||
|
||
## Simple Chain Usage | ||
|
||
*** | ||
|
||
```go | ||
// Function to tests | ||
func GetUsers(db *sql.DB) []map[string]string { | ||
var res []map[string]string | ||
age := 27 | ||
rows, err := db.Query("SELECT name FROM users WHERE age=?", age) | ||
if err != nil { | ||
log.Fatal(err) | ||
} | ||
defer rows.Close() | ||
for rows.Next() { | ||
var name string | ||
var age string | ||
if err := rows.Scan(&name, &age); err != nil { | ||
log.Fatal(err) | ||
} | ||
row := map[string]string{"name": name, "age": age} | ||
res = append(res, row) | ||
} | ||
if err := rows.Err(); err != nil { | ||
log.Fatal(err) | ||
} | ||
return res | ||
} | ||
|
||
// Test function | ||
func TestResponses(t *testing.T) { | ||
SetupTests() | ||
|
||
t.Run("Simple SELECT caught by query", func(t *testing.T) { | ||
Catcher.Logging = false | ||
commonReply := []map[string]interface{}{{"user_id": 27, "name": "FirstLast", "age": "30"}} | ||
Catcher.Reset().NewMock().WithQuery(`SELECT name FROM users WHERE`).WithReply(commonReply) | ||
result := GetUsers(DB) // Global or local variable | ||
if len(result) != 1 { | ||
t.Errorf("Returned sets is not equal to 1. Received %d", len(result)) | ||
} | ||
if result[0]["age"] != "30" { | ||
t.Errorf("Age is not equal. Got %v", result[0]["age"]) | ||
} | ||
}) | ||
} | ||
``` | ||
In the example above, we create a new mock via `.NewMock()` and attach a query pattern which will be used to catch a matched query. `.WithReply()` specifies which response will be provided during the mock of this request. | ||
As `Catcher` is global variable without calling `.Reset()` this mock will be applied to all subsequent tests and queries if the pattern matches. | ||
|
||
## Usage via `FakeResponse` Object | ||
|
||
We are taking `GetUsers` from the previous example and an example on how it can be using a FakeResponse directly attached to the Catcher object | ||
|
||
```go | ||
t.Run("Simple select with direct object", func(t *testing.T) { | ||
Catcher.Reset() | ||
commonReply := []map[string]interface{}{{"user_id": 27, "name": "FirstLast", "age": "30"}} | ||
Catcher.Attach([]*FakeResponse{ | ||
{ | ||
Pattern:"SELECT name FROM users WHERE", // the same as .WithQuery() | ||
Response: commonReply, // the same as .WithReply | ||
Once: false, // To not use it twice if true | ||
}, | ||
}) | ||
result := GetUsers(DB) | ||
if len(result) != 1 { | ||
t.Errorf("Returned sets is not equal to 1. Received %d", len(result)) | ||
} | ||
if result[0]["age"] != "30" { | ||
t.Errorf("Age is not equal. Got %v", result[0]["age"]) | ||
} | ||
}) | ||
``` | ||
|
||
## GORM Example | ||
|
||
Usage of a mocked GORM is completely transparent. You need to know which query will be generated by GORM and mock them or mock just by using arguments. In this case, you need to pay attention to order of arguments as GORM will not necessarily arrange them in order you provided them. | ||
|
||
*Tip:* To identify the exact query generated by GORM you can look at the console output when running your mocked DB connection. They show up like this: | ||
``` | ||
2018/01/01 12:00:01 mock_catcher: check query: INSERT INTO "users" ("name") VALUES (?) | ||
``` | ||
|
||
|
||
## More Examples | ||
|
||
*** | ||
|
||
### Catch by Arguments | ||
|
||
The query can be caught by provided arguments even you are not specifying a query pattern to match. | ||
Please note these two important facts: | ||
|
||
* Order is very important | ||
* GORM will re-order arguments according to fields in the struct defined to describe your model. | ||
|
||
```go | ||
t.Run("Catch by arguments", func(t *testing.T) { | ||
commonReply := []map[string]interface{}{{"name": "FirstLast", "age": "30"}} | ||
Catcher.Reset().NewMock().WithArgs(int64(27)).WithReply(commonReply) | ||
result := GetUsers(DB) | ||
if len(result) != 1 { | ||
t.Fatalf("Returned sets is not equal to 1. Received %d", len(result)) | ||
} | ||
// all other checks from reply | ||
}) | ||
``` | ||
|
||
### Match Only Once | ||
|
||
Mocks marked as Once, will not be match on subsequent queries. | ||
```go | ||
t.Run("Once", func(t *testing.T) { | ||
Catcher.Reset() | ||
Catcher.Attach([]*FakeResponse{ | ||
{ | ||
Pattern:"SELECT name FROM users WHERE", | ||
Response: commonReply, | ||
Once: true, // could be done via chaining .OneTime() | ||
}, | ||
}) | ||
GetUsers(DB) // Trigger once to use this mock | ||
result := GetUsers(DB) // trigger second time to receive empty results | ||
if len(result) != 0 { | ||
t.Errorf("Returned sets is not equal to 0. Received %d", len(result)) | ||
} | ||
}) | ||
``` | ||
|
||
### Insert ID with `.WithId(int64)` | ||
|
||
In order to emulate `INSERT` requests, we can mock the ID returned from the query with the `.WithId(int64)` method. | ||
|
||
```go | ||
// Somewhere in the code | ||
func InsertRecord(db *sql.DB) int64 { | ||
res, err := db.Exec(`INSERT INTO foo VALUES("bar", ?))`, "value") | ||
if err != nil { | ||
return 0 | ||
} | ||
id, _ := res.LastInsertId() | ||
return id | ||
} | ||
|
||
// Test code | ||
t.Run("Last insert id", func(t *testing.T) { | ||
var mockedId int64 | ||
mockedId = 64 | ||
Catcher.Reset().NewMock().WithQuery("INSERT INTO foo").WithId(mockedId) | ||
returnedId := InsertRecord(DB) | ||
if returnedId != mockedId { | ||
t.Fatalf("Last insert id not returned. Expected: [%v] , Got: [%v]", mockedId, returnedId) | ||
} | ||
}) | ||
``` | ||
|
||
### Emulate Exceptions | ||
|
||
You can emulate exceptions or errors during the request by setting it with a fake `FakeResponse` object. | ||
Please note that to fire an error on `SELECT` you need to use `WithQueryException()`, for other queries (UPDATE, DELETE, etc) which do not return results, you need to use `.WithExecException()`. | ||
|
||
Example: | ||
```go | ||
// Somewhere in the code | ||
func GetUsersWithError(db *sql.DB) error { | ||
age := 27 | ||
_, err := db.Query("SELECT name FROM users WHERE age=?", age) | ||
return err | ||
} | ||
|
||
func CreateUsersWithError(db *sql.DB) error { | ||
age := 27 | ||
_, err := db.Query("INSERT INTO users (age) VALUES (?) ", age) | ||
return err | ||
} | ||
|
||
// Test | ||
t.Run("Fire Query error", func(t *testing.T) { | ||
Catcher.Reset().NewMock().WithArgs(int64(27)).WithQueryException() | ||
err := GetUsersWithError(DB) | ||
if err == nil { | ||
t.Fatal("Error not triggered") | ||
} | ||
}) | ||
|
||
t.Run("Fire Execute error", func(t *testing.T) { | ||
Catcher.Reset().NewMock().WithQuery("INSERT INTO users (age)").WithQueryException() | ||
err := CreateUsersWithError(DB) | ||
if err == nil { | ||
t.Fatal("Error not triggered") | ||
} | ||
}) | ||
``` | ||
|
||
### Callbacks | ||
|
||
Besides that, you can catch and attach callbacks when the mock is used. | ||
|
||
## Code Gotchas | ||
|
||
### Query Matching | ||
When you try to match against a query, you have to make sure that you do so with precision. | ||
|
||
For example, this query : | ||
```sql | ||
SELECT * FROM "users" WHERE ("users"."user_id" = 3) ORDER BY "users"."user_id" ASC LIMIT 1 | ||
``` | ||
|
||
If you try to match it with this: | ||
```go | ||
Catcher.Reset().NewMock().WithQuery(`SELECT * FROM users WHERE`).WithReply(commonReply) | ||
``` | ||
It will not work for two reasons. `users` is missing double-quotes and there are two spaces before `WHERE`. One trick is to actually run the test and look at the mocked DB output to find the exact query being executed. | ||
Here is the right way to match this query: | ||
|
||
```go | ||
Catcher.Reset().NewMock().WithQuery(`SELECT * FROM "users" WHERE`).WithReply(commonReply) | ||
``` | ||
|
||
### Reply Matching | ||
When you provide a Reply to Catcher, your field names must match your database model or else, they will not be updated with the right value. | ||
|
||
Given you have this test code: | ||
```go | ||
// DO NOT USE, CODE NOT WORKING | ||
commonReply := []map[string]interface{}{{"userID": 7, "name": "FirstLast", "age": "30"}} | ||
mocket.Catcher.NewMock().OneTime().WithQuery(`SELECT * FROM "dummies"`).WithReply(commonReply) | ||
|
||
result := GetUsers(DB) | ||
``` | ||
|
||
This will work and not error out, but `result` will have a 0 value in the field `userID`. You must make sure to match the Reply fields with the database fields and not the struct fields or else you might bang your head on your keyboard. | ||
|
||
The following code works: | ||
```go | ||
commonReply := []map[string]interface{}{{"user_id": 7, "name": "FirstLast", "age": "30"}} | ||
mocket.Catcher.NewMock().OneTime().WithQuery(`SELECT * FROM "dummies"`).WithReply(commonReply) | ||
|
||
result := GetUsers(DB) | ||
``` | ||
|
||
__More examples coming....__ | ||
|