| // Copyright 2017, OpenCensus Authors |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| // |
| |
| package view |
| |
| import ( |
| "context" |
| "errors" |
| "sync" |
| "testing" |
| "time" |
| |
| "go.opencensus.io/stats" |
| "go.opencensus.io/tag" |
| ) |
| |
| func Test_Worker_ViewRegistration(t *testing.T) { |
| someError := errors.New("some error") |
| |
| sc1 := make(chan *Data) |
| |
| type registration struct { |
| c chan *Data |
| vID string |
| err error |
| } |
| type testCase struct { |
| label string |
| registrations []registration |
| } |
| tcs := []testCase{ |
| { |
| "register v1ID", |
| []registration{ |
| { |
| sc1, |
| "v1ID", |
| nil, |
| }, |
| }, |
| }, |
| { |
| "register v1ID+v2ID", |
| []registration{ |
| { |
| sc1, |
| "v1ID", |
| nil, |
| }, |
| }, |
| }, |
| { |
| "register to v1ID; ??? to v1ID and view with same ID", |
| []registration{ |
| { |
| sc1, |
| "v1ID", |
| nil, |
| }, |
| { |
| sc1, |
| "v1SameNameID", |
| someError, |
| }, |
| }, |
| }, |
| } |
| |
| mf1 := stats.Float64("MF1/Test_Worker_ViewSubscription", "desc MF1", "unit") |
| mf2 := stats.Float64("MF2/Test_Worker_ViewSubscription", "desc MF2", "unit") |
| |
| for _, tc := range tcs { |
| t.Run(tc.label, func(t *testing.T) { |
| restart() |
| |
| views := map[string]*View{ |
| "v1ID": { |
| Name: "VF1", |
| Measure: mf1, |
| Aggregation: Count(), |
| }, |
| "v1SameNameID": { |
| Name: "VF1", |
| Description: "desc duplicate name VF1", |
| Measure: mf1, |
| Aggregation: Sum(), |
| }, |
| "v2ID": { |
| Name: "VF2", |
| Measure: mf2, |
| Aggregation: Count(), |
| }, |
| "vNilID": nil, |
| } |
| |
| for _, r := range tc.registrations { |
| v := views[r.vID] |
| err := Register(v) |
| if (err != nil) != (r.err != nil) { |
| t.Errorf("%v: Register() = %v, want %v", tc.label, err, r.err) |
| } |
| } |
| }) |
| } |
| } |
| |
| func Test_Worker_RecordFloat64(t *testing.T) { |
| restart() |
| |
| someError := errors.New("some error") |
| m := stats.Float64("Test_Worker_RecordFloat64/MF1", "desc MF1", "unit") |
| |
| k1, _ := tag.NewKey("k1") |
| k2, _ := tag.NewKey("k2") |
| ctx, err := tag.New(context.Background(), |
| tag.Insert(k1, "v1"), |
| tag.Insert(k2, "v2"), |
| ) |
| if err != nil { |
| t.Fatal(err) |
| } |
| |
| v1 := &View{"VF1", "desc VF1", []tag.Key{k1, k2}, m, Count()} |
| v2 := &View{"VF2", "desc VF2", []tag.Key{k1, k2}, m, Count()} |
| |
| type want struct { |
| v *View |
| rows []*Row |
| err error |
| } |
| type testCase struct { |
| label string |
| registrations []*View |
| records []float64 |
| wants []want |
| } |
| |
| tcs := []testCase{ |
| { |
| label: "0", |
| registrations: []*View{}, |
| records: []float64{1, 1}, |
| wants: []want{{v1, nil, someError}, {v2, nil, someError}}, |
| }, |
| { |
| label: "1", |
| registrations: []*View{v1}, |
| records: []float64{1, 1}, |
| wants: []want{ |
| { |
| v1, |
| []*Row{ |
| { |
| []tag.Tag{{Key: k1, Value: "v1"}, {Key: k2, Value: "v2"}}, |
| &CountData{Value: 2}, |
| }, |
| }, |
| nil, |
| }, |
| {v2, nil, someError}, |
| }, |
| }, |
| { |
| label: "2", |
| registrations: []*View{v1, v2}, |
| records: []float64{1, 1}, |
| wants: []want{ |
| { |
| v1, |
| []*Row{ |
| { |
| []tag.Tag{{Key: k1, Value: "v1"}, {Key: k2, Value: "v2"}}, |
| &CountData{Value: 2}, |
| }, |
| }, |
| nil, |
| }, |
| { |
| v2, |
| []*Row{ |
| { |
| []tag.Tag{{Key: k1, Value: "v1"}, {Key: k2, Value: "v2"}}, |
| &CountData{Value: 2}, |
| }, |
| }, |
| nil, |
| }, |
| }, |
| }, |
| } |
| |
| for _, tc := range tcs { |
| for _, v := range tc.registrations { |
| if err := Register(v); err != nil { |
| t.Fatalf("%v: Register(%v) = %v; want no errors", tc.label, v.Name, err) |
| } |
| } |
| |
| for _, value := range tc.records { |
| stats.Record(ctx, m.M(value)) |
| } |
| |
| for _, w := range tc.wants { |
| gotRows, err := RetrieveData(w.v.Name) |
| if (err != nil) != (w.err != nil) { |
| t.Fatalf("%s: RetrieveData(%v) = %v; want error = %v", tc.label, w.v.Name, err, w.err) |
| } |
| for _, got := range gotRows { |
| if !containsRow(w.rows, got) { |
| t.Errorf("%s: got row %#v; want none", tc.label, got) |
| break |
| } |
| } |
| for _, want := range w.rows { |
| if !containsRow(gotRows, want) { |
| t.Errorf("%s: got none; want %#v'", tc.label, want) |
| break |
| } |
| } |
| } |
| |
| // Cleaning up. |
| Unregister(tc.registrations...) |
| } |
| } |
| |
| func TestReportUsage(t *testing.T) { |
| ctx := context.Background() |
| |
| m := stats.Int64("measure", "desc", "unit") |
| |
| tests := []struct { |
| name string |
| view *View |
| wantMaxCount int64 |
| }{ |
| { |
| name: "cum", |
| view: &View{Name: "cum1", Measure: m, Aggregation: Count()}, |
| wantMaxCount: 8, |
| }, |
| { |
| name: "cum2", |
| view: &View{Name: "cum1", Measure: m, Aggregation: Count()}, |
| wantMaxCount: 8, |
| }, |
| } |
| |
| for _, tt := range tests { |
| restart() |
| SetReportingPeriod(25 * time.Millisecond) |
| |
| if err := Register(tt.view); err != nil { |
| t.Fatalf("%v: cannot register: %v", tt.name, err) |
| } |
| |
| e := &countExporter{} |
| RegisterExporter(e) |
| |
| stats.Record(ctx, m.M(1)) |
| stats.Record(ctx, m.M(1)) |
| stats.Record(ctx, m.M(1)) |
| stats.Record(ctx, m.M(1)) |
| |
| time.Sleep(50 * time.Millisecond) |
| |
| stats.Record(ctx, m.M(1)) |
| stats.Record(ctx, m.M(1)) |
| stats.Record(ctx, m.M(1)) |
| stats.Record(ctx, m.M(1)) |
| |
| time.Sleep(50 * time.Millisecond) |
| |
| e.Lock() |
| count := e.count |
| e.Unlock() |
| if got, want := count, tt.wantMaxCount; got > want { |
| t.Errorf("%v: got count data = %v; want at most %v", tt.name, got, want) |
| } |
| } |
| |
| } |
| |
| func Test_SetReportingPeriodReqNeverBlocks(t *testing.T) { |
| t.Parallel() |
| |
| worker := newWorker() |
| durations := []time.Duration{-1, 0, 10, 100 * time.Millisecond} |
| for i, duration := range durations { |
| ackChan := make(chan bool, 1) |
| cmd := &setReportingPeriodReq{c: ackChan, d: duration} |
| cmd.handleCommand(worker) |
| |
| select { |
| case <-ackChan: |
| case <-time.After(500 * time.Millisecond): // Arbitrarily using 500ms as the timeout duration. |
| t.Errorf("#%d: duration %v blocks", i, duration) |
| } |
| } |
| } |
| |
| func TestWorkerStarttime(t *testing.T) { |
| restart() |
| |
| ctx := context.Background() |
| m := stats.Int64("measure/TestWorkerStarttime", "desc", "unit") |
| v := &View{ |
| Name: "testview", |
| Measure: m, |
| Aggregation: Count(), |
| } |
| |
| SetReportingPeriod(25 * time.Millisecond) |
| if err := Register(v); err != nil { |
| t.Fatalf("cannot register to %v: %v", v.Name, err) |
| } |
| |
| e := &vdExporter{} |
| RegisterExporter(e) |
| defer UnregisterExporter(e) |
| |
| stats.Record(ctx, m.M(1)) |
| stats.Record(ctx, m.M(1)) |
| stats.Record(ctx, m.M(1)) |
| stats.Record(ctx, m.M(1)) |
| |
| time.Sleep(50 * time.Millisecond) |
| |
| stats.Record(ctx, m.M(1)) |
| stats.Record(ctx, m.M(1)) |
| stats.Record(ctx, m.M(1)) |
| stats.Record(ctx, m.M(1)) |
| |
| time.Sleep(50 * time.Millisecond) |
| |
| e.Lock() |
| if len(e.vds) == 0 { |
| t.Fatal("Got no view data; want at least one") |
| } |
| |
| var start time.Time |
| for _, vd := range e.vds { |
| if start.IsZero() { |
| start = vd.Start |
| } |
| if !vd.Start.Equal(start) { |
| t.Errorf("Cumulative view data start time = %v; want %v", vd.Start, start) |
| } |
| } |
| e.Unlock() |
| } |
| |
| func TestUnregisterReportsUsage(t *testing.T) { |
| restart() |
| ctx := context.Background() |
| |
| m1 := stats.Int64("measure", "desc", "unit") |
| view1 := &View{Name: "count", Measure: m1, Aggregation: Count()} |
| m2 := stats.Int64("measure2", "desc", "unit") |
| view2 := &View{Name: "count2", Measure: m2, Aggregation: Count()} |
| |
| SetReportingPeriod(time.Hour) |
| |
| if err := Register(view1, view2); err != nil { |
| t.Fatalf("cannot register: %v", err) |
| } |
| |
| e := &countExporter{} |
| RegisterExporter(e) |
| |
| stats.Record(ctx, m1.M(1)) |
| stats.Record(ctx, m2.M(1)) |
| stats.Record(ctx, m2.M(1)) |
| |
| Unregister(view2) |
| |
| // Unregister should only flush view2, so expect the count of 2. |
| want := int64(2) |
| |
| e.Lock() |
| got := e.totalCount |
| e.Unlock() |
| if got != want { |
| t.Errorf("got count data = %v; want %v", got, want) |
| } |
| } |
| |
| type countExporter struct { |
| sync.Mutex |
| count int64 |
| totalCount int64 |
| } |
| |
| func (e *countExporter) ExportView(vd *Data) { |
| if len(vd.Rows) == 0 { |
| return |
| } |
| d := vd.Rows[0].Data.(*CountData) |
| |
| e.Lock() |
| defer e.Unlock() |
| e.count = d.Value |
| e.totalCount += d.Value |
| } |
| |
| type vdExporter struct { |
| sync.Mutex |
| vds []*Data |
| } |
| |
| func (e *vdExporter) ExportView(vd *Data) { |
| e.Lock() |
| defer e.Unlock() |
| |
| e.vds = append(e.vds, vd) |
| } |
| |
| // restart stops the current processors and creates a new one. |
| func restart() { |
| defaultWorker.stop() |
| defaultWorker = newWorker() |
| go defaultWorker.start() |
| } |