Initial import of OSS OctoDNS

tests/helpers.py

@@ -0,0 +1,69 @@
+#
+#
+#
+
+from __future__ import absolute_import, division, print_function, \
+    unicode_literals
+
+from shutil import rmtree
+from tempfile import mkdtemp
+
+
+class SimpleSource(object):
+
+    def __init__(self, id='test'):
+        pass
+
+
+class SimpleProvider(object):
+    SUPPORTS_GEO = False
+
+    def __init__(self, id='test'):
+        pass
+
+    def populate(self, zone, source=True):
+        pass
+
+    def supports(self, record):
+        return True
+
+    def __repr__(self):
+        return self.__class__.__name__
+
+
+class GeoProvider(object):
+    SUPPORTS_GEO = True
+
+    def __init__(self, id='test'):
+        pass
+
+    def populate(self, zone, source=True):
+        pass
+
+    def supports(self, record):
+        return True
+
+    def __repr__(self):
+        return self.__class__.__name__
+
+
+class NoSshFpProvider(SimpleProvider):
+
+    def supports(self, record):
+        return record._type != 'SSHFP'
+
+
+class TemporaryDirectory(object):
+
+    def __init__(self, delete_on_exit=True):
+        self.delete_on_exit = delete_on_exit
+
+    def __enter__(self):
+        self.dirname = mkdtemp()
+        return self
+
+    def __exit__(self, *args, **kwargs):
+        if self.delete_on_exit:
+            rmtree(self.dirname)
+        else:
+            raise Exception(self.dirname)