import pytest
from assets import models
import datetime


@pytest.fixture
def category(db):
    category = models.AssetCategory.objects.create(name="Sound")
    yield category
    category.delete()


@pytest.fixture
def status(db):
    status = models.AssetStatus.objects.create(name="Broken", should_show=True)
    yield status
    status.delete()


@pytest.fixture
def cable_type(db):
    connector = models.Connector.objects.create(description="16A IEC", current_rating=16, voltage_rating=240, num_pins=3)
    cable_type = models.CableType.objects.create(circuits=11, cores=3, plug=connector, socket=connector)
    yield cable_type
    connector.delete()
    cable_type.delete()


@pytest.fixture
def test_cable(db, category, status, cable_type):
    cable = models.Asset.objects.create(asset_id="9666", description="125A -> Jack", comments="The cable from Hell...", status=status, category=category, date_acquired=datetime.date(2006, 6, 6), is_cable=True, cable_type=cable_type, length=10, csa="1.5", salvage_value=50)
    yield cable
    cable.delete()


@pytest.fixture
def test_asset(db, category, status):
    asset, created = models.Asset.objects.get_or_create(asset_id="91991", description="Spaceflower", status=status, category=category, date_acquired=datetime.date(1991, 12, 26), salvage_value=100)
    yield asset
    asset.delete()