| // Licensed to the Apache Software Foundation (ASF) under one or more |
| // contributor license agreements. See the NOTICE file distributed with |
| // this work for additional information regarding copyright ownership. |
| // The ASF licenses this file to You 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 graphx |
| |
| import ( |
| "reflect" |
| "strings" |
| "testing" |
| |
| "github.com/apache/beam/sdks/go/pkg/beam/core/runtime" |
| v1 "github.com/apache/beam/sdks/go/pkg/beam/core/runtime/graphx/v1" |
| ) |
| |
| func TestEncodeType(t *testing.T) { |
| t.Run("NoUnexportedFields", func(t *testing.T) { |
| type MyAwesomeType struct { |
| ExportedField string |
| } |
| rt := reflect.TypeOf((*MyAwesomeType)(nil)).Elem() |
| |
| pbT, err := encodeType(rt) |
| if err != nil { |
| t.Fatalf("got error = %v, want nil", err) |
| } |
| if got, want := pbT.Kind, v1.Type_STRUCT; got != want { |
| t.Fatalf("got pbT.Kind == %v, want %v", got, want) |
| } |
| }) |
| t.Run("UnregisteredWithUnexportedField", func(t *testing.T) { |
| type MyProblematicType struct { |
| unexportedField string |
| } |
| rt := reflect.TypeOf((*MyProblematicType)(nil)).Elem() |
| pbT, err := encodeType(rt) |
| if err == nil { |
| t.Fatalf("got type = %v, nil, want unexported field error", pbT) |
| } |
| if !strings.Contains(err.Error(), "type has unexported field: unexportedField") { |
| t.Errorf("expected error about unexported field, got %q", err.Error()) |
| } |
| }) |
| t.Run("RegisteredWithUnexportedField", func(t *testing.T) { |
| type MyRegisteredType struct { |
| unexportedField string |
| } |
| rt := reflect.TypeOf((*MyRegisteredType)(nil)).Elem() |
| runtime.RegisterType(rt) |
| pbT, err := encodeType(rt) |
| if err != nil { |
| t.Fatalf("got error = %v, want nil", err) |
| } |
| if got, want := pbT.Kind, v1.Type_EXTERNAL; got != want { |
| t.Fatalf("got pbT.Kind == %v, want %v", got, want) |
| } |
| }) |
| } |