content: //
  // Licensed to the Apache Software Foundation (ASF) under one
  // or more contributor license agreements.  See the NOTICE file
  // distributed with this work for additional information
  // regarding copyright ownership.  The ASF licenses this file
  // to you under the Apache License, Version 2.0 (the
  // "License"); you may not use this file except in compliance
  // with the License.  You may obtain a copy of the License at
  //
  //   http://www.apache.org/licenses/LICENSE-2.0
  //
  // Unless required by applicable law or agreed to in writing,
  // software distributed under the License is distributed on an
  // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  // KIND, either express or implied.  See the License for the
  // specific language governing permissions and limitations
  // under the License.
  //
  package com.cloud.hypervisor.kvm.resource.wrapper;
  
  import static org.junit.Assert.assertEquals;
  import static org.mockito.ArgumentMatchers.nullable;
  import static org.mockito.Mockito.doReturn;
  import static org.mockito.Mockito.mock;
  import static org.mockito.Mockito.spy;
  import static org.mockito.Mockito.when;
  
  
  import org.junit.Before;
  import org.junit.Test;
  import org.junit.runner.RunWith;
  import org.libvirt.Connect;
  import org.libvirt.Domain;
  import org.libvirt.LibvirtException;
  
  import com.cloud.agent.api.routing.IpAssocVpcCommand;
  import com.cloud.agent.api.routing.NetworkElementCommand;
  import com.cloud.agent.api.to.IpAddressTO;
  import com.cloud.hypervisor.kvm.resource.LibvirtComputingResource;
  import com.cloud.network.Networks;
  import com.cloud.utils.ExecutionResult;
  import org.mockito.junit.MockitoJUnitRunner;
  
  @RunWith(MockitoJUnitRunner.class)
  public class LibvirtNetworkElementCommandWrapperTest {
      private static final String fullfile = "<domain type='kvm' id='143'>\n"
              + "  <name>r-3-VM</name>\n"
              + "  <uuid>8825b180-468f-4227-beb7-6b06fd342116</uuid>\n"
              + "  <description>CentOS 5.5 (64-bit)</description>\n"
              + "  <memory unit='KiB'>262144</memory>\n"
              + "  <currentMemory unit='KiB'>262144</currentMemory>\n"
              + "  <vcpu placement='static'>1</vcpu>\n"
              + "  <cputune>\n"
              + "    <shares>256</shares>\n"
              + "  </cputune>\n"
              + "  <sysinfo type='smbios'>\n"
              + "    <system>\n"
              + "      <entry name='manufacturer'>Apache Software Foundation</entry>\n"
              + "      <entry name='product'>CloudStack KVM Hypervisor</entry>\n"
              + "      <entry name='uuid'>8825b180-468f-4227-beb7-6b06fd342116</entry>\n"
              + "    </system>\n"
              + "  </sysinfo>\n"
              + "  <os>\n"
              + "    <type arch='x86_64' machine='pc-i440fx-rhel7.0.0'>hvm</type>\n"
              + "    <boot dev='cdrom'/>\n"
              + "    <boot dev='hd'/>\n"
              + "    <smbios mode='sysinfo'/>\n"
              + "  </os>\n"
              + "  <features>\n"
              + "    <acpi/>\n"
              + "    <apic/>\n"
              + "    <pae/>\n"
              + "  </features>\n"
              + "  <clock offset='utc'>\n"
              + "    <timer name='kvmclock'/>\n"
              + "  </clock>\n"
              + "  <on_poweroff>destroy</on_poweroff>\n"
              + "  <on_reboot>restart</on_reboot>\n"
              + "  <on_crash>destroy</on_crash>\n"
              + "  <devices>\n"
              + "    <emulator>/usr/libexec/qemu-kvm</emulator>\n"
              + "    <disk type='file' device='disk'>\n"
              + "      <driver name='qemu' type='qcow2' cache='none'/>\n"
              + "      <source file='/mnt/4436eeec-abec-3ef8-b733-c9541df20361/0c4aae69-2652-4a04-b460-1abb5a1a695c'/>\n"
              + "      <backingStore type='file' index='1'>\n"
              + "        <format type='raw'/>\n"
              + "        <source file='/mnt/4436eeec-abec-3ef8-b733-c9541df20361/d9ce07e5-9e13-11e7-816b-faac09070700'/>\n"
              + "        <backingStore/>\n"
              + "      </backingStore>\n"
              + "      <target dev='vda' bus='virtio'/>\n"
              + "      <serial>0c4aae6926524a04b460</serial>\n"
              + "      <alias name='virtio-disk0'/>\n"
              + "      <address type='pci' domain='0x0000' bus='0x00' slot='0x05' function='0x0'/>\n"
              + "    </disk>\n"
              + "    <disk type='file' device='cdrom'>\n"
              + "      <driver name='qemu' type='raw' cache='none'/>\n"
              + "      <backingStore/>\n"
              + "      <target dev='hdc' bus='ide'/>\n"
              + "      <readonly/>\n"
              + "      <alias name='ide0-1-0'/>\n"
              + "      <address type='drive' controller='0' bus='1' target='0' unit='0'/>\n"
              + "    </disk>\n"
              + "    <controller type='usb' index='0'>\n"
              + "      <alias name='usb'/>\n"
              + "      <address type='pci' domain='0x0000' bus='0x00' slot='0x01' function='0x2'/>\n"
              + "    </controller>\n"
              + "    <controller type='pci' index='0' model='pci-root'>\n"
              + "      <alias name='pci.0'/>\n"
              + "    </controller>\n"
              + "    <controller type='ide' index='0'>\n"
              + "      <alias name='ide'/>\n"
              + "      <address type='pci' domain='0x0000' bus='0x00' slot='0x01' function='0x1'/>\n"
              + "    </controller>\n"
              + "    <controller type='virtio-serial' index='0'>\n"
              + "      <alias name='virtio-serial0'/>\n"
              + "      <address type='pci' domain='0x0000' bus='0x00' slot='0x04' function='0x0'/>\n"
              + "    </controller>\n"
              + "    <interface type='bridge'>\n"
              + "      <mac address='02:00:7c:98:00:01'/>\n"
              + "      <source bridge='cloud0'/>\n"
              + "      <bandwidth>\n"
              + "        <inbound average='25600' peak='25600'/>\n"
              + "        <outbound average='25600' peak='25600'/>\n"
              + "      </bandwidth>\n"
              + "      <target dev='vnet1'/>\n"
              + "      <model type='virtio'/>\n"
              + "      <virtualport type='openvswitch'>\n"
              + "      </virtualport>\n"
              + "      <link state='up'/>\n"
              + "      <alias name='net0'/>\n"
              + "      <address type='pci' domain='0x0000' bus='0x00' slot='0x03' function='0x0'/>\n"
              + "    </interface>\n"
              + "    <interface type='bridge'>\n"
              + "      <mac address='02:00:7c:98:00:02'/>\n"
              + "      <source bridge='publicbr'/>\n"
              + "      <bandwidth>\n"
              + "        <inbound average='25600' peak='25600'/>\n"
              + "        <outbound average='25600' peak='25600'/>\n"
              + "      </bandwidth>\n"
              + "      <target dev='vnet2'/>\n"
              + "      <model type='virtio'/>\n"
              + "      <virtualport type='openvswitch'>\n"
              + "      </virtualport>\n"
              + "      <link state='up'/>\n"
              + "      <alias name='net1'/>\n"
              + "      <address type='pci' domain='0x0000' bus='0x00' slot='0x04' function='0x0'/>\n"
              + "    </interface>\n"
              + "    <interface type='bridge'>\n"
              + "      <mac address='02:00:7c:98:00:03'/>\n"
              + "      <source bridge='guestbr-100'/>\n"
              + "      <bandwidth>\n"
              + "        <inbound average='25600' peak='25600'/>\n"
              + "        <outbound average='25600' peak='25600'/>\n"
              + "      </bandwidth>\n"
              + "      <target dev='vnet3'/>\n"
              + "      <model type='virtio'/>\n"
              + "      <virtualport type='openvswitch'>\n"
              + "      </virtualport>\n"
              + "      <link state='up'/>\n"
              + "      <alias name='net2'/>\n"
              + "      <address type='pci' domain='0x0000' bus='0x00' slot='0x05' function='0x0'/>\n"
              + "    </interface>\n"
              + "    <interface type='bridge'>\n"
              + "      <mac address='02:00:7c:98:00:04'/>\n"
              + "      <source bridge='guestbr-101'/>\n"
              + "      <bandwidth>\n"
              + "        <inbound average='25600' peak='25600'/>\n"
              + "        <outbound average='25600' peak='25600'/>\n"
              + "      </bandwidth>\n"
              + "      <target dev='vnet4'/>\n"
              + "      <model type='virtio'/>\n"
              + "      <virtualport type='openvswitch'>\n"
              + "      </virtualport>\n"
              + "      <link state='up'/>\n"
              + "      <alias name='net3'/>\n"
              + "      <address type='pci' domain='0x0000' bus='0x00' slot='0x06' function='0x0'/>\n"
              + "    </interface>\n"
              + "    <serial type='pty'>\n"
              + "      <source path='/dev/pts/4'/>\n"
              + "      <target port='0'/>\n"
              + "      <alias name='serial0'/>\n"
              + "    </serial>\n"
              + "    <console type='pty' tty='/dev/pts/4'>\n"
              + "      <source path='/dev/pts/4'/>\n"
              + "      <target type='serial' port='0'/>\n"
              + "      <alias name='serial0'/>\n"
              + "    </console>\n"
              + "    <channel type='unix'>\n"
              + "      <source mode='bind' path='/var/lib/libvirt/qemu/i-85-285-VM.org.qemu.guest_agent.0'/>\n"
              + "      <target type='virtio' name='org.qemu.guest_agent.0' state='disconnected'/>\n"
              + "      <alias name='channel0'/>\n"
              + "      <address type='virtio-serial' controller='0' bus='0' port='1'/>\n"
              + "    </channel>\n"
              + "    <input type='tablet' bus='usb'>\n"
              + "      <alias name='input0'/>\n"
              + "    </input>\n"
              + "    <input type='mouse' bus='ps2'/>\n"
              + "    <input type='keyboard' bus='ps2'/>\n"
              + "    <graphics type='vnc' port='5903' autoport='yes' listen='10.100.100.11'>\n"
              + "      <listen type='address' address='10.100.100.11'/>\n"
              + "    </graphics>\n"
              + "    <video>\n"
              + "      <model type='cirrus' vram='16384' heads='1'/>\n"
              + "      <alias name='video0'/>\n"
              + "      <address type='pci' domain='0x0000' bus='0x00' slot='0x02' function='0x0'/>\n"
              + "    </video>\n"
              + "    <memballoon model='none'>\n"
              + "      <alias name='balloon0'/>\n"
              + "    </memballoon>\n"
              + "  </devices>\n"
              + "</domain>\n";
  
      private LibvirtComputingResource res;
      private final Domain domain = mock(Domain.class);
  
      final String memInfo = "MemTotal:        5830236 kB\n" +
              "MemFree:          156752 kB\n" +
              "Buffers:          326836 kB\n" +
              "Cached:          2606764 kB\n" +
              "SwapCached:            0 kB\n" +
              "Active:          4260808 kB\n" +
              "Inactive:         949392 kB\n";
      @Before
      public void setUp() throws Exception {
          // Use a spy because we only want to override getVifDriverClass
          LibvirtComputingResource resReal = new LibvirtComputingResource() {
              {
                  linkLocalBridgeName = "cloud0";
                  guestBridgeName = "guestbr";
                  publicBridgeName = "publicbr";
                  privBridgeName = "mgmtbr";
              }
          };
  
          res = spy(resReal);
  
          Connect conn = mock(Connect.class);
          LibvirtUtilitiesHelper helper = mock(LibvirtUtilitiesHelper.class);
  
          when(domain.getXMLDesc(0)).thenReturn(fullfile);
          when(conn.domainLookupByName(nullable(String.class))).thenReturn(domain);
          when(helper.getConnectionByVmName(nullable(String.class))).thenReturn(conn);
  
          doReturn(helper).when(res).getLibvirtUtilitiesHelper();
      }
  
      @Test
      public void testPrepareIpAssocVpcCommand() throws LibvirtException {
          IpAddressTO ip = new IpAddressTO(1, "171.31.1.3",
                  true, false, false,
                  "vlan://untagged",
                  "172.31.1.1",
                  "255.255.0.0",
                  "02:00:7c:98:00:02",
                  0,
                  true);
          ip.setTrafficType(Networks.TrafficType.Public);
          IpAddressTO[] ips = new IpAddressTO[] {
                  ip
          };
          final IpAssocVpcCommand command = new IpAssocVpcCommand(ips);
          command.setAccessDetail(NetworkElementCommand.ROUTER_IP, "127.0.0.1");
          ExecutionResult result = res.prepareCommand(command);
  
          assertEquals(1, ips[0].getNicDevId().intValue());
      }
  
      @Test
      public void testVpcPrivateGateway() throws LibvirtException {
          IpAddressTO ip = new IpAddressTO(1, "171.31.1.3",
                                           true, false, false,
                                           "vlan://untagged",
                                           "172.31.1.1",
                                           "255.255.0.0",
                                           "02:00:7c:98:00:03",
                                           0,
                                           false);
          ip.setTrafficType(Networks.TrafficType.Guest);
          IpAddressTO[] ips = new IpAddressTO[] {
                  ip
          };
          final IpAssocVpcCommand command = new IpAssocVpcCommand(ips);
          command.setAccessDetail(NetworkElementCommand.ROUTER_IP, "127.0.0.1");
          ExecutionResult result = res.prepareCommand(command);
  
          assertEquals(2, ips[0].getNicDevId().intValue());
      }
  
  }
  