func (s *managedStorageSuite) SetUpTest(c *gc.C) { s.BaseSuite.SetUpTest(c) s.MgoSuite.SetUpTest(c) s.db = s.Session.DB("juju") s.txnRunner = statetxn.NewRunner(txn.NewRunner(s.db.C("txns"))) s.resourceStorage = storage.NewGridFS("storage", "test", s.Session) s.managedStorage = storage.NewManagedStorage(s.db, s.txnRunner, s.resourceStorage) }
func (s *benchmarkSuite) BenchmarkManagedStorageCreate(c *gc.C) { db := s.Session.DB("a-database") txnRunner := jujuTxn.NewRunner(txn.NewRunner(db.C("txns"))) rstore := storage.NewGridFS(db.Name, "prefix", s.Session) store := storage.NewManagedStorage(db, txnRunner, rstore) c.ResetTimer() const fileSize = 30 * 1024 for i := 0; i < c.N; i++ { src := newDataSource(int64(i), fileSize) err := store.PutForEnvironment("env-uuid", fmt.Sprintf("file%d", i), src, fileSize) c.Assert(err, gc.IsNil) } }