func setUp() { cfg := config.NewDefaultServiceConfig() cfg.GlobalDefaultBucket = config.NewDefaultBucketConfig(config.DefaultBucketName) // Namespace "dyn" ns := config.NewDefaultNamespaceConfig("dyn") tpl := config.NewDefaultBucketConfig("") tpl.MaxTokensPerRequest = 5 tpl.MaxIdleMillis = -1 config.SetDynamicBucketTemplate(ns, tpl) ns.MaxDynamicBuckets = 2 config.AddNamespace(cfg, ns) // Namespace "dyn_gc" ns = config.NewDefaultNamespaceConfig("dyn_gc") tpl = config.NewDefaultBucketConfig("") tpl.MaxTokensPerRequest = 5 tpl.MaxIdleMillis = 100 config.SetDynamicBucketTemplate(ns, tpl) ns.MaxDynamicBuckets = 3 config.AddNamespace(cfg, ns) // Namespace "nodyn" ns = config.NewDefaultNamespaceConfig("nodyn") b := config.NewDefaultBucketConfig("b") b.MaxTokensPerRequest = 10 config.AddBucket(ns, b) config.AddNamespace(cfg, ns) mbf = &MockBucketFactory{} me := &MockEndpoint{} p := config.NewMemoryConfig(cfg) s = New(mbf, p, me) eventsChan = make(chan events.Event, 100) s.SetListener(func(e events.Event) { eventsChan <- e }, 100) s.Start() qs = me.QuotaService // EVENTS_BUCKET_CREATED event _ = <-eventsChan }
func TestUnmarshalNamespaceConfig(t *testing.T) { n := config.NewDefaultNamespaceConfig("Blah Namespace 123") n.MaxDynamicBuckets = 8000 config.SetDynamicBucketTemplate(n, config.NewDefaultBucketConfig("")) c1 := config.NewDefaultBucketConfig("Blah 123") c1.FillRate = 12345 c1.MaxDebtMillis = 54321 c1.MaxIdleMillis = 67890 c1.MaxTokensPerRequest = 9876 c1.Size = 50000 c2 := config.NewDefaultBucketConfig("Blah 456") c2.FillRate = 123450 c2.MaxDebtMillis = 543210 c2.MaxIdleMillis = 678900 c2.MaxTokensPerRequest = 98760 c2.Size = 5000 c3 := config.NewDefaultBucketConfig("Blah 789") c3.FillRate = 1234500 c3.MaxDebtMillis = 5432100 c3.MaxIdleMillis = 6789000 c3.MaxTokensPerRequest = 987600 c3.Size = 500 config.AddBucket(n, c1) config.AddBucket(n, c2) config.AddBucket(n, c3) b, e := json.Marshal(n) if e != nil { t.Fatal("Unable to JSONify proto", e) } reRead, err := getNamespaceConfig(bytes.NewReader(b)) if err != nil { t.Fatal("Unable to unmarshal JSON", err) } if !reflect.DeepEqual(n, reRead) { t.Fatalf("Two representations aren't equal: %+v != %+v", n, reRead) } }
func TestGC(t *testing.T, factory quotaservice.BucketFactory, impl string) { cfg := config.NewDefaultServiceConfig() nsCfg := config.NewDefaultNamespaceConfig("n") tpl := config.NewDefaultBucketConfig("") // Times out every 250 millis. tpl.MaxIdleMillis = 250 config.SetDynamicBucketTemplate(nsCfg, tpl) config.AddNamespace(cfg, nsCfg) eventsEmitter := "aservice.MockEmitter{Events: make(chan events.Event, 100)} container := quotaservice.NewBucketContainer(cfg, factory, eventsEmitter) // No GC should happen here as long as we are in use. for i := 0; i < 10; i++ { for j := 0; j < 10; j++ { bName := strconv.Itoa(j) b, _ := container.FindBucket("n", bName) if b == nil { t.Fatalf("Failed looking for bucket %v on impl %v", bName, impl) } // Check that the bucket hasn't been GC'd if !container.Exists("n", bName) { t.Fatalf("Bucket %v was GC'd when it shouldn't have on impl %v", bName, impl) } } } bucketNames := make([]string, 10) for i := 0; i < 10; i++ { bucketNames[i] = strconv.Itoa(i) } waitForGC(eventsEmitter.Events, "n", bucketNames) for _, bName := range bucketNames { // Check that the bucket has been GC'd if container.Exists("n", bName) { t.Fatalf("Bucket %v wasn't GC'd when it should have on impl %v", bName, impl) } } }