model: fix searching for Assoc. Set without NS

Adapted commit of FedorSelitsky <[email protected]>
from PR #14

---

Tests by Jakub Filak <[email protected]>

Author: filak-sap

pyodata/v2/model.py

@@ -729,7 +729,7 @@ def association_set_by_association(self, association_name, namespace=None):
                 association_name, namespace))
         for decl in list(self._decls.values()):
             for association_set in list(decl.association_sets.values()):
-                if association_set.association_type == association_name:
+                if association_set.association_type.name == association_name:
                     return association_set
         raise KeyError('Association set with association type {} does not exist'.format(association_name))
 

tests/test_model_v2.py

@@ -196,6 +196,14 @@ def test_edmx_associations(schema):
     assert association_set.association_type.name == 'toDataEntity'
     assert association_set.end_roles == {'DataValueHelp': 'ToRole_toDataEntity', 'MasterEntities': 'FromRole_toDataEntity'}
 
+    # with namespace
+    association_set = schema.association_set_by_association('CustomerOrders', namespace='EXAMPLE_SRV_SETS')
+    assert str(association_set) == 'AssociationSet(CustomerOrder_AssocSet)'
+
+    # without namespace
+    association_set = schema.association_set_by_association('CustomerOrders')
+    assert str(association_set) == 'AssociationSet(CustomerOrder_AssocSet)'
+
 
 def test_edmx_navigation_properties(schema):
     """Test parsing of navigation properties"""