| // Copyright 2018 Google LLC |
| // |
| // 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 metadata |
| |
| import ( |
| "context" |
| "encoding/json" |
| "reflect" |
| "testing" |
| "time" |
| |
| "github.com/google/go-cmp/cmp" |
| ) |
| |
| func TestMetadata(t *testing.T) { |
| md := &Metadata{EventID: "test event ID"} |
| ctx := NewContext(context.Background(), md) |
| got, err := FromContext(ctx) |
| if err != nil { |
| t.Fatalf("FromContext error: %v", err) |
| } |
| if !reflect.DeepEqual(got, md) { |
| t.Fatalf("FromContext\nGot %v\nWant %v", got, md) |
| } |
| } |
| |
| func TestMetadataError(t *testing.T) { |
| if _, err := FromContext(nil); err == nil { |
| t.Errorf("FromContext got no error, wanted an error") |
| } |
| if _, err := FromContext(context.Background()); err == nil { |
| t.Errorf("FromContext got no error, wanted an error") |
| } |
| if _, err := FromContext(NewContext(context.Background(), nil)); err == nil { |
| t.Errorf("FromContext got no error, wanted an error") |
| } |
| } |
| |
| func TestUnmarshalJSON(t *testing.T) { |
| ts, err := time.Parse("2006-01-02T15:04:05Z07:00", "2019-11-04T23:01:10.112Z") |
| if err != nil { |
| t.Fatalf("Error parsing time: %v.", err) |
| } |
| var tests = []struct { |
| name string |
| data []byte |
| want Metadata |
| }{ |
| { |
| name: "MetadataWithResource", |
| data: []byte(`{ |
| "eventId": "1234567", |
| "timestamp": "2019-11-04T23:01:10.112Z", |
| "eventType": "google.pubsub.topic.publish", |
| "resource": { |
| "service": "pubsub.googleapis.com", |
| "name": "mytopic", |
| "type": "type.googleapis.com/google.pubsub.v1.PubsubMessage" |
| }, |
| "data": { |
| "@type": "type.googleapis.com/google.pubsub.v1.PubsubMessage", |
| "attributes": null, |
| "data": "test data" |
| } |
| }`), |
| want: Metadata{ |
| EventID: "1234567", |
| Timestamp: ts, |
| EventType: "google.pubsub.topic.publish", |
| Resource: &Resource{ |
| Service: "pubsub.googleapis.com", |
| Name: "mytopic", |
| Type: "type.googleapis.com/google.pubsub.v1.PubsubMessage", |
| }, |
| }, |
| }, |
| { |
| name: "MetadataWithString", |
| data: []byte(`{ |
| "eventId": "1234567", |
| "timestamp": "2019-11-04T23:01:10.112Z", |
| "eventType": "google.pubsub.topic.publish", |
| "resource": "projects/myproject/mytopic", |
| "data": { |
| "@type": "type.googleapis.com/google.pubsub.v1.PubsubMessage", |
| "attributes": null, |
| "data": "test data" |
| } |
| }`), |
| want: Metadata{ |
| EventID: "1234567", |
| Timestamp: ts, |
| EventType: "google.pubsub.topic.publish", |
| Resource: &Resource{ |
| RawPath: "projects/myproject/mytopic", |
| }, |
| }, |
| }, |
| } |
| |
| for _, tc := range tests { |
| var m Metadata |
| if err := json.Unmarshal(tc.data, &m); err != nil { |
| t.Errorf("UnmarshalJSON(%s) error: %v", tc.name, err) |
| } |
| if !cmp.Equal(m, tc.want) { |
| t.Errorf("UnmarshalJSON(%s) error: got %v, want %v", tc.name, m, tc.want) |
| } |
| } |
| } |