Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

https://github.com/prashantv/gostub

gostub is a library to make stubbing in unit tests easy
https://github.com/prashantv/gostub

Last synced: 17 days ago
JSON representation

gostub is a library to make stubbing in unit tests easy

Lists

README 

        
# gostub



[![Build Status](https://travis-ci.org/prashantv/gostub.svg?branch=master)](https://travis-ci.org/prashantv/gostub)

[![GoDoc](https://godoc.org/github.com/prashantv/gostub?status.svg)](https://godoc.org/github.com/prashantv/gostub)

[![Coverage Status](https://coveralls.io/repos/github/prashantv/gostub/badge.svg?branch=master)](https://coveralls.io/github/prashantv/gostub?branch=master)



gostub is a library to make stubbing in unit tests easy.



## Getting started



Import the following package:

`github.com/prashantv/gostub`



Click [here](https://godoc.org/github.com/prashantv/gostub) to read the [API documentation](https://godoc.org/github.com/prashantv/gostub).



## Package overview



Package gostub is used for stubbing variables in tests, and resetting the

original value once the test has been run.



This can be used to stub static variables as well as static functions. To stub a

static variable, use the Stub function:



```go

var configFile = "config.json"



func GetConfig() ([]byte, error) {

    return ioutil.ReadFile(configFile)

}



// Test code

stubs := gostub.Stub(&configFile, "/tmp/test.config")



data, err := GetConfig()

// data will now return contents of the /tmp/test.config file

```



gostub can also stub static functions in a test by using a variable to reference

the static function, and using that local variable to call the static function:



```go

var timeNow = time.Now



func GetDate() int {

    return timeNow().Day()

}

```



You can test this by using gostub to stub the timeNow variable:



```go

stubs := gostub.Stub(&timeNow, func() time.Time {

    return time.Date(2015, 6, 1, 0, 0, 0, 0, time.UTC)

})

defer stubs.Reset()



// Test can check that GetDate returns 6

```



If you are stubbing a function to return a constant value like in the above

test, you can use StubFunc instead:



```go

stubs := gostub.StubFunc(&timeNow, time.Date(2015, 6, 1, 0, 0, 0, 0, time.UTC))

defer stubs.Reset()

```



StubFunc can also be used to stub functions that return multiple values:



```go

var osHostname = osHostname

// [...] production code using osHostname to call it.



// Test code:

stubs := gostub.StubFunc(&osHostname, "fakehost", nil)

defer stubs.Reset()

```

StubEnv can be used to setup environment variables for tests, and the

environment values are reset to their original values upon Reset:



```go

stubs := gostub.New()

stubs.SetEnv("GOSTUB_VAR", "test_value")

defer stubs.Reset()

```



The Reset method should be deferred to run at the end of the test to reset all

stubbed variables back to their original values.



You can set up multiple stubs by calling Stub again:



```go

stubs := gostub.Stub(&v1, 1)

stubs.Stub(&v2, 2)

defer stubs.Reset()

```



For simple cases where you are only setting up simple stubs, you can condense

the setup and cleanup into a single line:



```go

defer gostub.Stub(&v1, 1).Stub(&v2, 2).Reset()

```



This sets up the stubs and then defers the Reset call.



You should keep the return argument from the Stub call if you need to change

stubs or add more stubs during test execution:



```go

stubs := gostub.Stub(&v1, 1)

defer stubs.Reset()



// Do some testing

stubs.Stub(&v1, 5)



// More testing

stubs.Stub(&b2, 6)

```



The Stub call must be passed a pointer to the variable that should be stubbed,

and a value which can be assigned to the variable.