device_test.go

package nucleus

import (
	"reflect"
	"testing"

	"code.fbi.h-da.de/cocsn/gosdn/forks/goarista/gnmi"
	"code.fbi.h-da.de/cocsn/yang-models/generated/openconfig"
	"github.com/google/uuid"
	"github.com/openconfig/ygot/ygot"
)

func TestDevice_Id(t *testing.T) {
	type fields struct {
		GoStruct  ygot.GoStruct
		SBI       SouthboundInterface
		Transport Transport
		UUID      uuid.UUID
		Name      string
	}
	tests := []struct {
		name   string
		fields fields
		want   uuid.UUID
	}{
		{
			name: "default",
			fields: fields{
				UUID: did,
			},
			want: did,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			d := &Device{
				GoStruct:  tt.fields.GoStruct,
				SBI:       tt.fields.SBI,
				Transport: tt.fields.Transport,
				UUID:      tt.fields.UUID,
				Name:      tt.fields.Name,
			}
			if got := d.ID(); !reflect.DeepEqual(got, tt.want) {
				t.Errorf("ID() = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestNewDevice(t *testing.T) {
	sbi := &OpenConfig{}
	type args struct {
		sbi  SouthboundInterface
		opts TransportOptions
		name string
	}
	tests := []struct {
		name    string
		args    args
		want    *Device
		wantErr bool
	}{
		{
			name: "default",
			args: args{
				sbi: sbi,
				opts: &GnmiTransportOptions{
					Config: gnmi.Config{
						Addr:     "test:///",
						Username: "test",
						Password: "test",
					},
				},
				name: "MyDevice",
			},
			want: &Device{
				GoStruct: &openconfig.Device{},
				SBI:      sbi,
				UUID:     uuid.New(),
				Transport: &Gnmi{
					Options: &GnmiTransportOptions{
						Config: gnmi.Config{
							Addr:     "test:///",
							Username: "test",
							Password: "test",
						},
					},
				},
				Name: "MyDevice",
			},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, err := NewDevice(tt.args.sbi, tt.args.opts, tt.args.name)
			if err != nil {
				t.Error(err)
			}
			tt.want.Transport.(*Gnmi).client = got.Transport.(*Gnmi).client
			tt.want.UUID = got.ID()
			if (err != nil) != tt.wantErr {
				t.Errorf("NewDevice() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if !reflect.DeepEqual(got, tt.want) {
				t.Errorf("NewDevice() got = %v, want %v", got, tt.want)
			}
		})
	}
}