github.com/openebs/node-disk-manager@v1.9.1-0.20230225014141-4531f06ffa1e/api-service/node/services/nodeName_test.go (about)

     1  /*
     2  Copyright 2020 The OpenEBS Authors
     3  Licensed under the Apache License, Version 2.0 (the "License");
     4  you may not use this file except in compliance with the License.
     5  You may obtain a copy of the License at
     6      http://www.apache.org/licenses/LICENSE-2.0
     7  Unless required by applicable law or agreed to in writing, software
     8  distributed under the License is distributed on an "AS IS" BASIS,
     9  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    10  See the License for the specific language governing permissions and
    11  limitations under the License.
    12  */
    13  
    14  package services
    15  
    16  import (
    17  	"context"
    18  	"os"
    19  	"testing"
    20  
    21  	protos "github.com/openebs/node-disk-manager/spec/ndm"
    22  )
    23  
    24  // TestName tests Name service
    25  func TestName(t *testing.T) {
    26  	os.Setenv("NODE_NAME", "TEST_NODE")
    27  
    28  	var ctx context.Context
    29  	var null *protos.Null
    30  
    31  	n := NewNode()
    32  	res, err := n.Name(ctx, null)
    33  	if err != nil {
    34  		t.Error("Error in finding node name")
    35  	}
    36  	if res.NodeName != "TEST_NODE" {
    37  		t.Errorf("Expected node name was %v and got %v", "TEST_NODE", res.NodeName)
    38  	}
    39  
    40  }