12826 fix tests

netbox/dcim/graphql/filters.py

@@ -38,6 +38,7 @@ __all__ = (
     'RackFilter',
     'RackReservationFilter',
     'RackRoleFilter',
+    'RackTypeFilter',
     'RearPortFilter',
     'RearPortTemplateFilter',
     'RegionFilter',
@@ -234,6 +235,12 @@ class PowerPortTemplateFilter(BaseFilterMixin):
     pass
 
 
+@strawberry_django.filter(models.RackType, lookups=True)
+@autotype_decorator(filtersets.RackTypeFilterSet)
+class RackTypeFilter(BaseFilterMixin):
+    pass
+
+
 @strawberry_django.filter(models.Rack, lookups=True)
 @autotype_decorator(filtersets.RackFilterSet)
 class RackFilter(BaseFilterMixin):

netbox/dcim/graphql/schema.py

@@ -159,6 +159,11 @@ class DCIMQuery:
         return models.PowerPortTemplate.objects.get(id=id)
     power_port_template_list: List[PowerPortTemplateType] = strawberry_django.field()
 
+    @strawberry.field
+    def rack_type(self, id: int) -> RackTypeType:
+        return models.RackType.objects.get(id=id)
+    rack_type_list: List[RackTypeType] = strawberry_django.field()
+
     @strawberry.field
     def rack(self, id: int) -> RackType:
         return models.Rack.objects.get(id=id)

netbox/dcim/graphql/types.py

@@ -50,6 +50,7 @@ __all__ = (
     'RackType',
     'RackReservationType',
     'RackRoleType',
+    'RackTypeType',
     'RearPortType',
     'RearPortTemplateType',
     'RegionType',
@@ -606,6 +607,15 @@ class PowerPortTemplateType(ModularComponentTemplateType):
     poweroutlet_templates: List[Annotated["PowerOutletTemplateType", strawberry.lazy('dcim.graphql.types')]]
 
 
+@strawberry_django.type(
+    models.RackType,
+    fields='__all__',
+    filters=RackTypeFilter
+)
+class RackTypeType(ImageAttachmentsMixin, NetBoxObjectType):
+    _name: str
+
+
 @strawberry_django.type(
     models.Rack,
     fields='__all__',

netbox/dcim/tests/test_views.py

@@ -370,7 +370,7 @@ class RackTypeTestCase(ViewTestCases.PrimaryObjectViewTestCase):
 
         cls.csv_data = (
             "name,width,u_height,weight,max_weight,weight_unit",
-            ",Rack 4,19,42,100,2000,kg",
+            "Rack 4,19,42,100,2000,kg",
             "Rack 5,19,42,100,2000,kg",
             "Rack 6,19,42,100,2000,kg",
         )