RDBC-790 quickfixes

Author: poissoncorp

ravendb_embedded/embedded_server.py

@@ -33,7 +33,7 @@ def __new__(cls):
         return cls._INSTANCE
 
     def __init__(self):
-        if hasattr(self, 'initialized'):
+        if hasattr(self, "initialized"):
             return
         self.initialized = True
         self.server_task: Optional[Lazy[Tuple[str, subprocess.Popen]]] = None
@@ -44,6 +44,12 @@ def __init__(self):
         self.INSTANCE = self
         self.logger = logging.Logger(self.__class__.__name__, logging.DEBUG)
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.close()
+
     def _log_debug(self, message: str) -> None:
         if not self.logger.disabled and self.logger.isEnabledFor(logging.DEBUG):
             self.logger.log(logging.DEBUG, message)

ravendb_embedded/provide.py

@@ -21,7 +21,7 @@ def provide(self, target_directory: str) -> None:  # todo: test
         shutil.copytree(self.server_files, target_directory)
 
 
-class CopyServerFromNugetProvider(ProvideRavenDBServer):
+class CopyServerFromNugetProvider(CopyServerProvider):
     SERVER_FILES = "target/nuget/contentFiles/any/any/RavenDBServer"
 
     def __init__(self):