def test_asset_class_has_attributes_attr():
from .factories import AssetClassFactory
a = AssetClassFactory(name="automobile")
assert a.attributes.all().count() == 0
assert a.attributes.first() is None
def setUp(self):
self.admin = AdminFactory()
self.client.credentials(HTTP_AUTHORIZATION="Token " +
self.admin.auth_token.key)
self.asset_class = AssetClassFactory(name="Asset Class")
self.data = {
"name": "Asset Class Name",
"alt_name": "Asset Class Alt Name",
"is_active": True,
}
def test_asset_partial_update_asset_class(self):
"""Assets can asset class updated by admins."""
new_asset_class = AssetClassFactory()
data = {"asset_class": new_asset_class.name}
response = self.client.patch(
reverse("v1:assets-detail", args=[self.asset.pk]), data)
self.asset.refresh_from_db()
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(self.asset.asset_class.name, data["asset_class"])
def build_automotive_domain():
from assets.tests.factories import (
DomainFactory,
AssetClassFactory,
AttributeFactory,
)
domain = DomainFactory(name="automotive")
car_class = AssetClassFactory(name="car")
domain.asset_classes.add(car_class)
vin_attribute = AttributeFactory(name="vin")
car_class.attributes.add(vin_attribute)
return domain
def setUp(self):
self.admin = AdminFactory()
self.client.credentials(HTTP_AUTHORIZATION="Token " +
self.admin.auth_token.key)
self.exchange = ExchangeFactory()
self.asset_class = AssetClassFactory()
self.asset = AssetFactory(name="Asset",
exchange=self.exchange,
asset_class=self.asset_class)
self.data = {
"id": str(uuid.uuid4()),
"name": "Asset Name",
"asset_class": self.asset_class.name,
"easy_to_borrow": True,
"exchange": self.exchange.name,
"marginable": True,
"shortable": True,
"status": Asset.ACTIVE,
"symbol": "SYMBOL",
"tradable": True,
}
def test_check_asset_class_factory_exists():
from .factories import AssetClassFactory
a = AssetClassFactory()
assert a.name