// 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.vpc;

import java.util.ArrayList;

import java.util.List;

import javax.inject.Inject;

import junit.framework.TestCase;

import org.junit.Before;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import com.cloud.exception.InvalidParameterValueException;

import com.cloud.network.Network.Service;

import com.cloud.network.vpc.Vpc;

import com.cloud.network.vpc.VpcManagerImpl;

import com.cloud.network.vpc.VpcVO;

import com.cloud.user.AccountVO;

import com.cloud.utils.component.ComponentContext;

@RunWith(SpringJUnit4ClassRunner.class)

//@ContextConfiguration(locations = "classpath:/VpcTestContext.xml")

public class VpcApiUnitTest extends TestCase {

@Inject

VpcManagerImpl _vpcService = null;

@Override

@Before

public void setUp() throws Exception {

ComponentContext.initComponentsLifeCycle();

}

@Test

public void getActiveVpc() {

//test for active vpc

boolean result = false;

Vpc vpc = null;

try {

List<String> svcs = new ArrayList<String>();

svcs.add(Service.SourceNat.getName());

vpc = _vpcService.getActiveVpc(1);

if (vpc != null) {

result = true;

}

} catch (Exception ex) {

} finally {

assertTrue("Get active Vpc: TEST FAILED, active vpc is not returned", result);

}

//test for inactive vpc

result = false;

vpc = null;

try {

List<String> svcs = new ArrayList<String>();

svcs.add(Service.SourceNat.getName());

vpc = _vpcService.getActiveVpc(2);

if (vpc != null) {

result = true;

}

} catch (Exception ex) {

} finally {

assertFalse("Get active Vpc: TEST FAILED, non active vpc is returned", result);

}

}

@Test

public void validateNtwkOffForVpc() {

//validate network offering

//1) correct network offering

VpcVO vo = new VpcVO(1, "new vpc", "new vpc", 1, 1, 1, "0.0.0.0/0", "vpc domain");

boolean result = false;

try {

_vpcService.validateNtwkOffForNtwkInVpc(2L, 1, "0.0.0.0", "111-", vo, "10.1.1.1", new AccountVO(), null);

result = true;

} catch (Exception ex) {

} finally {

assertTrue("Validate network offering: Test passed: the offering is valid for vpc creation", result);

}

//2) invalid offering - source nat is not included

result = false;

try {

_vpcService.validateNtwkOffForNtwkInVpc(2L, 2, "0.0.0.0", "111-", vo, "10.1.1.1", new AccountVO(), null);

result = true;

} catch (InvalidParameterValueException ex) {

} finally {

assertFalse("Validate network offering: TEST FAILED, can't use network offering without SourceNat service", result);

}

//3) invalid offering - conserve mode is off

result = false;

try {

_vpcService.validateNtwkOffForNtwkInVpc(2L, 3, "0.0.0.0", "111-", vo, "10.1.1.1", new AccountVO(), null);

result = true;

} catch (InvalidParameterValueException ex) {

} finally {

assertFalse("Validate network offering: TEST FAILED, can't use network offering without conserve mode = true", result);

}

//4) invalid offering - guest type shared

result = false;

try {

_vpcService.validateNtwkOffForNtwkInVpc(2L, 4, "0.0.0.0", "111-", vo, "10.1.1.1", new AccountVO(), null);

result = true;

} catch (InvalidParameterValueException ex) {

} finally {

assertFalse("Validate network offering: TEST FAILED, can't use network offering with guest type = Shared", result);

}

//5) Invalid offering - no redundant router support

result = false;

try {

_vpcService.validateNtwkOffForNtwkInVpc(2L, 5, "0.0.0.0", "111-", vo, "10.1.1.1", new AccountVO(), null);

result = true;

} catch (InvalidParameterValueException ex) {

} finally {

assertFalse("TEST FAILED, can't use network offering with guest type = Shared", result);

}

}

// public void destroyVpc() {

// boolean result = false;

// try {

// result = _vpcService.destroyVpc(vo, new AccountVO(), 1L);

// } catch (Exception ex) {

// s_logger.debug(ex);

// } finally {

// assertTrue("Failed to destroy VPC", result);

// }

// }

//

// public void deleteVpc() {

// //delete existing offering

// boolean result = false;

// try {

// List<String> svcs = new ArrayList<String>();

// svcs.add(Service.SourceNat.getName());

// result = _vpcService.deleteVpc(1);

// } catch (Exception ex) {

// } finally {

// assertTrue("Delete vpc: TEST FAILED, vpc failed to delete" + result, result);

// }

//

// //delete non-existing offering

// result = false;

// try {

// List<String> svcs = new ArrayList<String>();

// svcs.add(Service.SourceNat.getName());

// result = _vpcService.deleteVpc(100);

// } catch (Exception ex) {

// } finally {

// assertFalse("Delete vpc: TEST FAILED, true is returned when try to delete non existing vpc" + result, result);

// }

// }

}