From a94b570dc6cbcea5fea5c3d1d22d2ec271344932 Mon Sep 17 00:00:00 2001 From: Hu Shuai Date: Tue, 30 Jun 2020 18:11:57 +0800 Subject: [PATCH] Add a unit test for newAzureResourceFromID in discovery/azure/azure.go. (#7484) This PR is about adding a unit test for newAzureResourceFromID in discovery/azure/azure.go. Signed-off-by: Hu Shuai --- discovery/azure/azure_test.go | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/discovery/azure/azure_test.go b/discovery/azure/azure_test.go index 31550f8b57..3e7f218ca9 100644 --- a/discovery/azure/azure_test.go +++ b/discovery/azure/azure_test.go @@ -201,3 +201,24 @@ func TestMapFromVMScaleSetVMWithTags(t *testing.T) { t.Errorf("Expected %v got %v", expectedVM, actualVM) } } + +func TestNewAzureResourceFromID(t *testing.T) { + for _, tc := range []struct { + id string + expected azureResource + }{ + { + id: "/a/b/c/group/d/e/f/name", + expected: azureResource{"name", "group"}, + }, + { + id: "/a/b/c/group/d/e/f/name/g/h", + expected: azureResource{"name", "group"}, + }, + } { + actual, _ := newAzureResourceFromID(tc.id, nil) + if !reflect.DeepEqual(tc.expected, actual) { + t.Errorf("Expected %v got %v", tc.expected, actual) + } + } +}