| // 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 org.apache.cloudstack.network.contrail.model; |
| |
| import static org.mockito.ArgumentMatchers.anyLong; |
| import static org.mockito.Mockito.mock; |
| import static org.mockito.Mockito.when; |
| |
| import java.io.IOException; |
| import java.util.UUID; |
| |
| import junit.framework.TestCase; |
| import net.juniper.contrail.api.ApiConnector; |
| import net.juniper.contrail.api.ApiConnectorMock; |
| |
| import org.apache.cloudstack.network.contrail.management.ContrailManagerImpl; |
| import org.apache.cloudstack.network.contrail.management.ModelDatabase; |
| import org.junit.Test; |
| |
| import com.cloud.network.Network; |
| import com.cloud.network.dao.NetworkVO; |
| import com.cloud.vm.VMInstanceVO; |
| import com.cloud.vm.VirtualMachine; |
| import com.cloud.vm.dao.UserVmDao; |
| |
| public class VirtualMachineModelTest extends TestCase { |
| |
| @Test |
| public void testVirtualMachineDBLookup() { |
| ModelDatabase db = new ModelDatabase(); |
| VMInstanceVO vm = mock(VMInstanceVO.class); |
| |
| // Create 3 dummy Virtual Machine model objects |
| // Add these models to database. |
| // Each VM is identified by unique UUId. |
| VirtualMachineModel vm0 = new VirtualMachineModel(vm, "fbc1f8fa-4b78-45ee-bba0-b551dbf72353"); |
| db.getVirtualMachines().add(vm0); |
| |
| VirtualMachineModel vm1 = new VirtualMachineModel(vm, "fbc1f8fa-4b78-45ee-bba0-b551dbf83464"); |
| db.getVirtualMachines().add(vm1); |
| |
| VirtualMachineModel vm2 = new VirtualMachineModel(vm, "fbc1f8fa-4b78-45ee-bba0-b551dbf94575"); |
| db.getVirtualMachines().add(vm2); |
| |
| |
| assertEquals(3, db.getVirtualMachines().size()); |
| |
| assertSame(vm0, db.lookupVirtualMachine("fbc1f8fa-4b78-45ee-bba0-b551dbf72353")); |
| assertSame(vm1, db.lookupVirtualMachine("fbc1f8fa-4b78-45ee-bba0-b551dbf83464")); |
| assertSame(vm2, db.lookupVirtualMachine("fbc1f8fa-4b78-45ee-bba0-b551dbf94575")); |
| } |
| |
| @Test |
| public void testCreateVirtualMachine() throws IOException { |
| |
| String uuid = UUID.randomUUID().toString(); |
| ContrailManagerImpl contrailMgr = mock(ContrailManagerImpl.class); |
| ModelController controller = mock(ModelController.class); |
| ApiConnector api = new ApiConnectorMock(null, 0); |
| when(controller.getManager()).thenReturn(contrailMgr); |
| when(controller.getApiAccessor()).thenReturn(api); |
| |
| // Create Virtual-Network (VN) |
| NetworkVO network = MockNetworkVO.getNetwork(Network.State.Allocated); |
| |
| when(contrailMgr.getCanonicalName(network)).thenReturn("testnetwork"); |
| when(contrailMgr.getProjectId(network.getDomainId(), network.getAccountId())).thenReturn("testProjectId"); |
| |
| // Create Virtual-Machine (VM) |
| VMInstanceVO vm = mock(VMInstanceVO.class); |
| when(vm.getInstanceName()).thenReturn("testVM1"); |
| when(vm.getState()).thenReturn(VirtualMachine.State.Starting); |
| when(vm.getDomainId()).thenReturn(10L); |
| when(vm.getAccountId()).thenReturn(42L); |
| |
| UserVmDao VmDao = mock(UserVmDao.class); |
| when(VmDao.findById(anyLong())).thenReturn(null); |
| when(controller.getVmDao()).thenReturn(VmDao); |
| |
| VirtualMachineModel vmModel = new VirtualMachineModel(vm, uuid); |
| |
| assertEquals(vmModel.getInstanceName(), "testVM1"); |
| assertEquals(vmModel.getUuid(), uuid); |
| |
| vmModel.build(controller, vm); |
| try { |
| vmModel.update(controller); |
| } catch (Exception ex) { |
| fail("virtual-network update failed "); |
| } |
| |
| //verify |
| assertTrue(vmModel.verify(controller)); |
| } |
| } |