blob: f6c77f479769b2a7663353ff6e177fd3252f4b58 [file] [log] [blame]
// 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)
}
}
}