"code.cloudfoundry.org/bbs/db/dbfakes"
"code.cloudfoundry.org/bbs/models"
"code.cloudfoundry.org/bbs/models/test/model_helpers"
"code.cloudfoundry.org/bbs/taskworkpool/taskworkpoolfakes"
"code.cloudfoundry.org/lager/lagertest"
"code.cloudfoundry.org/rep"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"github.com/onsi/gomega/gbytes"
)
var _ = Describe("Task Controller", func() {
var (
logger *lagertest.TestLogger
fakeTaskDB *dbfakes.FakeTaskDB
fakeAuctioneerClient *auctioneerfakes.FakeClient
fakeTaskCompletionClient *taskworkpoolfakes.FakeTaskCompletionClient
controller *controllers.TaskController
)
BeforeEach(func() {
fakeTaskDB = new(dbfakes.FakeTaskDB)
fakeAuctioneerClient = new(auctioneerfakes.FakeClient)
fakeTaskCompletionClient = new(taskworkpoolfakes.FakeTaskCompletionClient)
logger = lagertest.NewTestLogger("test")
controller = controllers.NewTaskController(fakeTaskDB, fakeTaskCompletionClient, fakeAuctioneerClient, fakeServiceClient, fakeRepClientFactory)
})
Describe("Tasks", func() {
var (
"github.com/onsi/gomega/gbytes"
)
var _ = Describe("LRP Convergence Controllers", func() {
var (
err error
logger *lagertest.TestLogger
fakeLRPDB *dbfakes.FakeLRPDB
actualHub *eventfakes.FakeHub
fakeAuctioneerClient *auctioneerfakes.FakeClient
keysToAuction []*auctioneer.LRPStartRequest
keysToRetire []*models.ActualLRPKey
keysWithMissingCells []*models.ActualLRPKeyWithSchedulingInfo
retiringActualLRP1 *models.ActualLRP
retiringActualLRP2 *models.ActualLRP
desiredLRP1, desiredLRP2 models.DesiredLRPSchedulingInfo
unclaimingActualLRP1 *models.ActualLRP
unclaimingActualLRP2 *models.ActualLRP
cellID string
cellSet models.CellSet
controller *controllers.LRPConvergenceController
)
BeforeEach(func() {
fakeLRPDB = new(dbfakes.FakeLRPDB)
fakeAuctioneerClient = new(auctioneerfakes.FakeClient)
"strings"
"github.com/pivotal-cf/brokerapi"
"github.com/pivotal-cf/brokerapi/fakes"
"code.cloudfoundry.org/lager"
"code.cloudfoundry.org/lager/lagertest"
"github.com/drewolson/testflight"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)
var _ = Describe("Service Broker API", func() {
var fakeServiceBroker *fakes.FakeServiceBroker
var brokerAPI http.Handler
var brokerLogger *lagertest.TestLogger
var credentials = brokerapi.BrokerCredentials{
Username: "******",
Password: "******",
}
makeInstanceProvisioningRequest := func(instanceID string, details map[string]interface{}, queryString string) *testflight.Response {
response := &testflight.Response{}
testflight.WithServer(brokerAPI, func(r *testflight.Requester) {
path := "/v2/service_instances/" + instanceID + queryString
buffer := &bytes.Buffer{}
json.NewEncoder(buffer).Encode(details)
request, err := http.NewRequest("PUT", path, buffer)
Expect(err).NotTo(HaveOccurred())
"code.cloudfoundry.org/lager/lagertest"
"github.com/cloudfoundry/gunk/command_runner/fake_command_runner"
. "github.com/cloudfoundry/gunk/command_runner/fake_command_runner/matchers"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)
var _ = Describe("RunAndLog", func() {
const logs string = `time="2016-03-02T13:56:38Z" level=warning msg="signal: potato"
time="2016-03-02T13:56:38Z" level=error msg="fork/exec POTATO: no such file or directory"
time="2016-03-02T13:56:38Z" level=fatal msg="Container start failed: [10] System error: fork/exec POTATO: no such file or directory"`
var (
commandRunner *fake_command_runner.FakeCommandRunner
logRunner runrunc.RuncCmdRunner
logger *lagertest.TestLogger
logFile *os.File
)
BeforeEach(func() {
commandRunner = fake_command_runner.New()
logger = lagertest.NewTestLogger("test")
var err error
logFile, err = ioutil.TempFile("", "runandlog")
Expect(err).NotTo(HaveOccurred())
logRunner = runrunc.NewLogRunner(commandRunner, func() (*os.File, error) {
return logFile, nil
})
handler = func(w http.ResponseWriter, r *http.Request) { time.Sleep(10) }
handler = middleware.RequestCountWrap(handler)
})
It("reports call count", func() {
handler.ServeHTTP(nil, nil)
handler.ServeHTTP(nil, nil)
handler.ServeHTTP(nil, nil)
Expect(sender.GetCounter("RequestCount")).To(Equal(uint64(3)))
})
})
Describe("LogWrap", func() {
var (
logger *lagertest.TestLogger
loggableHandlerFunc middleware.LoggableHandlerFunc
)
BeforeEach(func() {
logger = lagertest.NewTestLogger("test-session")
logger.RegisterSink(lager.NewWriterSink(GinkgoWriter, lager.DEBUG))
loggableHandlerFunc = func(logger lager.Logger, w http.ResponseWriter, r *http.Request) {
logger = logger.Session("logger-group")
logger.Info("written-in-loggable-handler")
}
})
It("creates \"request\" session and passes it to LoggableHandlerFunc", func() {
handler := middleware.LogWrap(logger, nil, loggableHandlerFunc)
req, err := http.NewRequest("GET", "http://example.com", nil)
Expect(err).NotTo(HaveOccurred())
"code.cloudfoundry.org/lager/lagertest"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"github.com/tedsuo/ifrit"
_ "github.com/lib/pq"
"testing"
)
var (
db *sql.DB
sqlDB *sqldb.SQLDB
fakeClock *fakeclock.FakeClock
fakeGUIDProvider *guidproviderfakes.FakeGUIDProvider
logger *lagertest.TestLogger
cryptor encryption.Cryptor
serializer format.Serializer
migrationProcess ifrit.Process
dbDriverName, dbBaseConnectionString string
dbFlavor string
)
func TestSql(t *testing.T) {
RegisterFailHandler(Fail)
RunSpecs(t, "SQL DB Suite")
}
var _ = BeforeSuite(func() {
var err error
fakeClock = fakeclock.NewFakeClock(time.Now())
"errors"
"math/rand"
"code.cloudfoundry.org/lager/lagertest"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"github.com/onsi/gomega/gbytes"
"github.com/rosenhouse/cnsim/fakes"
"github.com/rosenhouse/cnsim/models"
"github.com/rosenhouse/cnsim/simulate"
)
var _ = Describe("Steady state simulator", func() {
var (
appSizeDistribution *fakes.MeanParameterizedDiscreteDistribution
sim *simulate.SteadyState
logger *lagertest.TestLogger
req models.SteadyStateRequest
)
BeforeEach(func() {
appSizeDistribution = &fakes.MeanParameterizedDiscreteDistribution{}
appSizeDistribution.SampleStub = func(_ float64) (int, error) {
return req.MeanInstancesPerApp + rand.Intn(2) - 1, nil
}
sim = &simulate.SteadyState{
AppSizeDistribution: appSizeDistribution,
}
logger = lagertest.NewTestLogger("test")
req = models.SteadyStateRequest{
NumHosts: 1000,
NumApps: 10000,
"code.cloudfoundry.org/garden-shed/repository_fetcher"
"code.cloudfoundry.org/garden-shed/rootfs_provider"
fakes "code.cloudfoundry.org/garden-shed/rootfs_provider/rootfs_providerfakes"
"code.cloudfoundry.org/lager"
"code.cloudfoundry.org/lager/lagertest"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)
var _ = Describe("The Cake Co-ordinator", func() {
var (
fakeFetcher *fakes.FakeRepositoryFetcher
fakeLayerCreator *fakes.FakeLayerCreator
fakeCake *fake_cake.FakeCake
fakeGCer *fakes.FakeGCer
fakeMetrics *fakes.FakeMetricser
logger *lagertest.TestLogger
cakeOrdinator *rootfs_provider.CakeOrdinator
)
BeforeEach(func() {
logger = lagertest.NewTestLogger("test")
fakeFetcher = new(fakes.FakeRepositoryFetcher)
fakeLayerCreator = new(fakes.FakeLayerCreator)
fakeCake = new(fake_cake.FakeCake)
fakeGCer = new(fakes.FakeGCer)
fakeMetrics = new(fakes.FakeMetricser)
cakeOrdinator = rootfs_provider.NewCakeOrdinator(fakeCake, fakeFetcher, fakeLayerCreator, fakeMetrics, fakeGCer)
"code.cloudfoundry.org/lager/lagertest"
"github.com/cloudfoundry/dropsonde/metric_sender/fake"
"github.com/cloudfoundry/dropsonde/metrics"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"github.com/tedsuo/ifrit"
"github.com/tedsuo/ifrit/ginkgomon"
)
var _ = Describe("Encryptor", func() {
var (
runner ifrit.Runner
encryptorProcess ifrit.Process
logger *lagertest.TestLogger
cryptor encryption.Cryptor
keyManager encryption.KeyManager
fakeDB *dbfakes.FakeEncryptionDB
sender *fake.FakeMetricSender
)
BeforeEach(func() {
sender = fake.NewFakeMetricSender()
metrics.Initialize(sender, nil)
fakeDB = new(dbfakes.FakeEncryptionDB)
logger = lagertest.NewTestLogger("test")
oldKey, err := encryption.NewKey("old-key", "old-passphrase")
"os"
"path/filepath"
"runtime"
"code.cloudfoundry.org/guardian/metrics"
"code.cloudfoundry.org/lager/lagertest"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)
var _ = Describe("Metrics", func() {
var (
logger *lagertest.TestLogger
backingStorePath string
depotPath string
m metrics.Metrics
)
BeforeEach(func() {
var err error
backingStorePath, err = ioutil.TempDir("", "backing_stores")
Expect(err).NotTo(HaveOccurred())
Expect(ioutil.WriteFile(
filepath.Join(backingStorePath, "bs-1"), []byte("test"), 0660,
)).To(Succeed())
Expect(ioutil.WriteFile(
filepath.Join(backingStorePath, "bs-2"), []byte("test"), 0660,
)).To(Succeed())
"code.cloudfoundry.org/bbs/taskworkpool"
"code.cloudfoundry.org/cfhttp"
"code.cloudfoundry.org/lager"
"code.cloudfoundry.org/lager/lagertest"
"github.com/tedsuo/ifrit"
"github.com/tedsuo/ifrit/ginkgomon"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"github.com/onsi/gomega/ghttp"
)
var _ = Describe("TaskWorker", func() {
var (
fakeServer *ghttp.Server
logger *lagertest.TestLogger
timeout time.Duration
)
BeforeEach(func() {
timeout = 1 * time.Second
cfhttp.Initialize(timeout)
fakeServer = ghttp.NewServer()
logger = lagertest.NewTestLogger("test")
logger.RegisterSink(lager.NewWriterSink(GinkgoWriter, lager.INFO))
})
AfterEach(func() {
fakeServer.Close()
})
