add nes tests

Author: Ckk3

src/strawberry_sqlalchemy_mapper/loader.py

@@ -81,10 +81,12 @@ async def load_fn(keys: List[Tuple]) -> List[Any]:
                     related_model_key_label = relationship.local_remote_pairs[1][1].key
 
                     self_model_key = relationship.local_remote_pairs[0][0].key
+                    related_model_key = relationship.local_remote_pairs[1][0].key
 
                     remote_to_use = relationship.local_remote_pairs[0][1]
                     query_keys = tuple([item[0] for item in keys])
 
+                    # This query returns every row equal (self_model.key, related_model)
                     query = (
                         select(
                             label(self_model_key_label, getattr(
@@ -94,12 +96,12 @@ async def load_fn(keys: List[Tuple]) -> List[Any]:
                         .join(
                             relationship.secondary,
                             getattr(relationship.secondary.c,
-                                    related_model_key_label) == related_model.id
+                                    related_model_key_label) == getattr(related_model, related_model_key)
                         )
                         .join(
                             self_model,
                             getattr(relationship.secondary.c,
-                                    self_model_key_label) == self_model.id
+                                    self_model_key_label) == getattr(self_model, self_model_key)
                         )
                         .filter(
                             remote_to_use.in_(query_keys)

src/strawberry_sqlalchemy_mapper/mapper.py

@@ -517,24 +517,16 @@ async def resolve(self, info: Info):
                         ]
                     )
                 else:
-                    # If has a secondary table, gets only the first id since the other id cannot be get without a query
-                    # breakpoint()
-                    local_remote_pairs_secondary_table_local = relationship.local_remote_pairs[0][0]
+                    # If has a secondary table, gets only the first ID as additional IDs require a separate query
+                    local_remote_pairs_secondary_table_local = relationship.local_remote_pairs[
+                        0][0]
                     relationship_key = tuple(
                         [
-                            getattr(self, local_remote_pairs_secondary_table_local.key),
+                            getattr(
+                                self, local_remote_pairs_secondary_table_local.key),
                         ]
                     )
 
-                    # relationship_key = tuple(
-                    #     [
-                    #         getattr(self, local.key)
-                    #         for local, _ in relationship.local_remote_pairs or []
-                    #         if local.key
-                    #     ]
-                    # )
-                    # breakpoint()
-
                 if any(item is None for item in relationship_key):
                     if relationship.uselist:
                         return []
@@ -544,7 +536,6 @@ async def resolve(self, info: Info):
                     loader = info.context["sqlalchemy_loader"]
                 else:
                     loader = info.context.sqlalchemy_loader
-                # breakpoint()
                 related_objects = await loader.loader_for(relationship).load(
                     relationship_key
                 )