"""
Test N2VC's ssh key generation
"""
import n2vc
import pytest
from . import base
import uuid


@pytest.mark.asyncio
async def test_model_create():
    """Test the creation of a new model."""
    client = base.get_n2vc()

    model_name = "test-{}".format(
        uuid.uuid4().hex[-4:],
    )

    pytest.assume(await client.CreateNetworkService(model_name))
    pytest.assume(await client.DestroyNetworkService(model_name))
    pytest.assume(await client.logout())


@pytest.mark.asyncio
async def test_destroy_non_existing_network_service():
    """Destroy a model that doesn't exist."""

    client = base.get_n2vc()

    model_name = "test-{}".format(
        uuid.uuid4().hex[-4:],
    )

    with pytest.raises(n2vc.vnf.NetworkServiceDoesNotExist):
        pytest.assume(await client.DestroyNetworkService(model_name))

    pytest.assume(await client.logout())


@pytest.mark.asyncio
async def test_model_create_duplicate():
    """Create a new model, and try to create the same model."""
    client = base.get_n2vc()

    model_name = "test-{}".format(
        uuid.uuid4().hex[-4:],
    )

    # Try to recreate bug 628
    for x in range(0, 1000):
        model = await client.get_model(model_name)
        pytest.assume(model)

    pytest.assume(await client.DestroyNetworkService(model_name))
    pytest.assume(await client.logout())