From 6f5d3cf6e74905fc925628fb84b757a43542d80c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andr=C3=A9=20Sterba?= <andre.sterba@stud.h-da.de> Date: Wed, 5 May 2021 22:24:03 +0200 Subject: [PATCH] Add test for getDeviceByName --- nucleus/principalNetworkDomain_test.go | 60 ++++++++++++++++++++++++-- 1 file changed, 56 insertions(+), 4 deletions(-) diff --git a/nucleus/principalNetworkDomain_test.go b/nucleus/principalNetworkDomain_test.go index 6ff0f9327..24db61460 100644 --- a/nucleus/principalNetworkDomain_test.go +++ b/nucleus/principalNetworkDomain_test.go @@ -1,14 +1,16 @@ package nucleus import ( + "errors" + "reflect" + "testing" + "code.fbi.h-da.de/cocsn/gosdn/mocks" "code.fbi.h-da.de/cocsn/yang-models/generated/openconfig" - "errors" + "github.com/google/uuid" "github.com/openconfig/ygot/ygot" "github.com/stretchr/testify/mock" - "reflect" - "testing" ) func TestNewPND(t *testing.T) { @@ -657,7 +659,7 @@ func Test_pndImplementation_ChangeOND(t *testing.T) { func Test_pndImplementation_GetDevice(t *testing.T) { p := newPnd() - d, err := NewDevice(sbi, &GnmiTransportOptions{}) + d, err := NewDevice(sbi, &GnmiTransportOptions{}, "") if err != nil { t.Error(err) return @@ -702,6 +704,56 @@ func Test_pndImplementation_GetDevice(t *testing.T) { } } +func Test_pndImplementation_GetDeviceByName(t *testing.T) { + p := newPnd() + d, err := NewDevice(sbi, &GnmiTransportOptions{}, "my-device") + if err != nil { + t.Error(err) + return + } + if err = p.addDevice(d); err != nil { + t.Error(err) + return + } + type args struct { + name string + } + tests := []struct { + name string + args args + want ygot.GoStruct + wantErr bool + }{ + { + name: "default", + args: args{name: d.Name}, + want: sbi.Schema().Root, + wantErr: false, + }, + { + name: "device not found", + args: args{name: "test-device"}, + want: nil, + wantErr: true, + }, + } + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + foundDevice, err := p.getDeviceByName(tt.args.name) + if (err != nil) != tt.wantErr { + t.Errorf("GetDeviceByName() error = %v, wantErr %v", err, tt.wantErr) + return + } + if foundDevice != nil { + if !reflect.DeepEqual(foundDevice.GoStruct, tt.want) { + t.Errorf("GetDeviceByName() got = %v, want %v", foundDevice.GoStruct, tt.want) + } + } + + }) + } +} + func Test_pndImplementation_Confirm(t *testing.T) { tests := []struct { name string -- GitLab