1/* 2Copyright (c) 2015 VMware, Inc. All Rights Reserved. 3 4Licensed under the Apache License, Version 2.0 (the "License"); 5you may not use this file except in compliance with the License. 6You may obtain a copy of the License at 7 8 http://www.apache.org/licenses/LICENSE-2.0 9 10Unless required by applicable law or agreed to in writing, software 11distributed under the License is distributed on an "AS IS" BASIS, 12WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13See the License for the specific language governing permissions and 14limitations under the License. 15*/ 16 17package object 18 19import ( 20 "context" 21 "fmt" 22 "reflect" 23 "testing" 24 25 "github.com/vmware/govmomi/test" 26 "github.com/vmware/govmomi/vim25/mo" 27) 28 29func TestSearch(t *testing.T) { 30 c := test.NewAuthenticatedClient(t) 31 s := NewSearchIndex(c) 32 33 ref, err := s.FindChild(context.Background(), NewRootFolder(c), "ha-datacenter") 34 if err != nil { 35 t.Fatal(err) 36 } 37 38 dc, ok := ref.(*Datacenter) 39 if !ok { 40 t.Errorf("Expected Datacenter: %#v", ref) 41 } 42 43 folders, err := dc.Folders(context.Background()) 44 if err != nil { 45 t.Fatal(err) 46 } 47 48 ref, err = s.FindChild(context.Background(), folders.DatastoreFolder, "datastore1") 49 if err != nil { 50 t.Fatal(err) 51 } 52 53 _, ok = ref.(*Datastore) 54 if !ok { 55 t.Errorf("Expected Datastore: %#v", ref) 56 } 57 58 ref, err = s.FindByInventoryPath(context.Background(), "/ha-datacenter/network/VM Network") 59 if err != nil { 60 t.Fatal(err) 61 } 62 63 _, ok = ref.(*Network) 64 if !ok { 65 t.Errorf("Expected Network: %#v", ref) 66 } 67 68 crs, err := folders.HostFolder.Children(context.Background()) 69 if err != nil { 70 if err != nil { 71 t.Fatal(err) 72 } 73 } 74 if len(crs) != 0 { 75 var cr mo.ComputeResource 76 ref = crs[0] 77 err = s.Properties(context.Background(), ref.Reference(), []string{"host"}, &cr) 78 if err != nil { 79 t.Fatal(err) 80 } 81 82 var host mo.HostSystem 83 ref = NewHostSystem(c, cr.Host[0]) 84 err = s.Properties(context.Background(), ref.Reference(), []string{"name", "hardware", "config"}, &host) 85 if err != nil { 86 t.Fatal(err) 87 } 88 89 dnsConfig := host.Config.Network.DnsConfig.GetHostDnsConfig() 90 dnsName := fmt.Sprintf("%s.%s", dnsConfig.HostName, dnsConfig.DomainName) 91 shost, err := s.FindByDnsName(context.Background(), dc, dnsName, false) 92 if err != nil { 93 t.Fatal(err) 94 } 95 if !reflect.DeepEqual(ref, shost) { 96 t.Errorf("%#v != %#v\n", ref, shost) 97 } 98 99 shost, err = s.FindByUuid(context.Background(), dc, host.Hardware.SystemInfo.Uuid, false, nil) 100 if err != nil { 101 t.Fatal(err) 102 } 103 if !reflect.DeepEqual(ref, shost) { 104 t.Errorf("%#v != %#v\n", ref, shost) 105 } 106 } 107 108 vms, err := folders.VmFolder.Children(context.Background()) 109 if err != nil { 110 t.Fatal(err) 111 } 112 if len(vms) != 0 { 113 var vm mo.VirtualMachine 114 ref = vms[0] 115 err = s.Properties(context.Background(), ref.Reference(), []string{"config", "guest"}, &vm) 116 if err != nil { 117 t.Fatal(err) 118 } 119 svm, err := s.FindByDatastorePath(context.Background(), dc, vm.Config.Files.VmPathName) 120 if err != nil { 121 t.Fatal(err) 122 } 123 if !reflect.DeepEqual(ref, svm) { 124 t.Errorf("%#v != %#v\n", ref, svm) 125 } 126 127 svm, err = s.FindByUuid(context.Background(), dc, vm.Config.Uuid, true, nil) 128 if err != nil { 129 t.Fatal(err) 130 } 131 if !reflect.DeepEqual(ref, svm) { 132 t.Errorf("%#v != %#v\n", ref, svm) 133 } 134 135 if vm.Guest.HostName != "" { 136 svm, err := s.FindByDnsName(context.Background(), dc, vm.Guest.HostName, true) 137 if err != nil { 138 t.Fatal(err) 139 } 140 if !reflect.DeepEqual(ref, svm) { 141 t.Errorf("%#v != %#v\n", ref, svm) 142 } 143 } 144 145 if vm.Guest.IpAddress != "" { 146 svm, err := s.FindByIp(context.Background(), dc, vm.Guest.IpAddress, true) 147 if err != nil { 148 t.Fatal(err) 149 } 150 if !reflect.DeepEqual(ref, svm) { 151 t.Errorf("%#v != %#v\n", ref, svm) 152 } 153 } 154 } 155} 156