mirror of
https://github.com/docker/compose.git
synced 2025-07-04 20:34:25 +02:00
11339761ca
Initially we stored the context data in the `Metadata` of the context
but in hindsight this data would be better of in the `Endpoints` because
that's what it is used for.
Before:
```json
{
"Name": "aci",
"Metadata": {
"Type": "aci",
"Data": {
"key": "value"
}
},
"Endpoints": {
"docker": {}
}
}
```
After:
```json
{
"Name": "aci",
"Type": "aci",
"Metadata": {},
"Endpoints": {
"aci": {
"key": "value"
},
"docker": {}
}
}
```
With this change the contexts that we create are more in line with the contexts the docker cli creates.
It also makes the code less complicated since we don't need to marsal twice any more. The API is nicer too:
```go
// Get a context:
c, err := store.Get(contextName)
// Get the stored endpoint:
var aciContext store.AciContext
if err := contextStore.GetEndpoint(currentContext, &aciContext); err != nil {
return nil, err
}
```
76 lines
1.7 KiB
Go
76 lines
1.7 KiB
Go
package framework
|
|
|
|
import (
|
|
"context"
|
|
"io/ioutil"
|
|
"os"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
apicontext "github.com/docker/api/context"
|
|
"github.com/docker/api/context/store"
|
|
)
|
|
|
|
// CliSuite is a helper struct that creates a configured context
|
|
// and captures the output of a command. it should be used in the
|
|
// same way as testify.suite.Suite
|
|
type CliSuite struct {
|
|
suite.Suite
|
|
ctx context.Context
|
|
writer *os.File
|
|
reader *os.File
|
|
OriginalStdout *os.File
|
|
storeRoot string
|
|
}
|
|
|
|
// BeforeTest is called by testify.suite
|
|
func (sut *CliSuite) BeforeTest(suiteName, testName string) {
|
|
ctx := context.Background()
|
|
ctx = apicontext.WithCurrentContext(ctx, "example")
|
|
dir, err := ioutil.TempDir("", "store")
|
|
require.Nil(sut.T(), err)
|
|
s, err := store.New(
|
|
store.WithRoot(dir),
|
|
)
|
|
require.Nil(sut.T(), err)
|
|
|
|
err = s.Create("example", "example", "", store.ContextMetadata{})
|
|
require.Nil(sut.T(), err)
|
|
|
|
sut.storeRoot = dir
|
|
|
|
ctx = store.WithContextStore(ctx, s)
|
|
sut.ctx = ctx
|
|
|
|
sut.OriginalStdout = os.Stdout
|
|
r, w, err := os.Pipe()
|
|
require.Nil(sut.T(), err)
|
|
|
|
os.Stdout = w
|
|
sut.writer = w
|
|
sut.reader = r
|
|
}
|
|
|
|
// Context returns a configured context
|
|
func (sut *CliSuite) Context() context.Context {
|
|
return sut.ctx
|
|
}
|
|
|
|
// GetStdOut returns the output of the command
|
|
func (sut *CliSuite) GetStdOut() string {
|
|
err := sut.writer.Close()
|
|
require.Nil(sut.T(), err)
|
|
|
|
out, _ := ioutil.ReadAll(sut.reader)
|
|
|
|
return string(out)
|
|
}
|
|
|
|
// AfterTest is called by testify.suite
|
|
func (sut *CliSuite) AfterTest(suiteName, testName string) {
|
|
os.Stdout = sut.OriginalStdout
|
|
err := os.RemoveAll(sut.storeRoot)
|
|
require.Nil(sut.T(), err)
|
|
}
|