more asserts

Author: tobixen

caldav/compatibility_hints.py

@@ -224,6 +224,10 @@ def _dotted_feature_set_list(self, feature_path, feature_node):
                 ret.update(self._dotted_feature_set_list(full, subnode))
         return ret
 
+    def __eq__(self, other):
+        ## TODO: this won't work, but it's a start
+        return self.dotted_feature_set_list() == other.dotted_feature_set_list()
+
 #### OLD STYLE
 
 ## The lists below are specifying what tests should be skipped or

caldav/davclient.py

@@ -26,6 +26,7 @@
 from caldav.collection import Calendar
 from caldav.collection import CalendarSet
 from caldav.collection import Principal
+from caldav.compatibility_hints import FeatureSet
 from caldav.elements import cdav
 from caldav.elements import dav
 from caldav.lib import error
@@ -458,6 +459,7 @@ def __init__(
         ssl_cert: Union[str, Tuple[str, str], None] = None,
         headers: Mapping[str, str] = None,
         huge_tree: bool = False,
+        features: FeatureSet = None
     ) -> None:
         """
         Sets up a HTTPConnection object towards the server in the url.
@@ -491,6 +493,7 @@ def __init__(
         log.debug("url: " + str(url))
         self.url = URL.objectify(url)
         self.huge_tree = huge_tree
+        self.features = features
         # Prepare proxy info
         if proxy is not None:
             _proxy = proxy

tests/test_caldav.py

@@ -846,12 +846,12 @@ def _fixCalendar(self, **kwargs):
                 ret.objects = lambda load_objects: ret.events()
             if self.cleanup_regime == "post":
                 self.calendars_used.append(ret)
-
             return ret
 
     def testCheckCompatibility(self):
         checker = ServerQuirkChecker(self.caldav)
         checker.check_all()
+        assert(checker.features_checked == self.caldav.feature_set)
 
     def testSupport(self):
         """