From da9203fcf2162eccdcf8309aa98dcd879c494162 Mon Sep 17 00:00:00 2001
From: "Antoine R. Dumont (@ardumont)" <ardumont@softwareheritage.org>
Date: Tue, 7 Jun 2022 14:47:48 +0200
Subject: [PATCH] tests: Fix current_version attribute change

This fixes test build [1]

[1] https://jenkins.softwareheritage.org/view/swh%20master%20(draft)/job/DVAU/job/tests/1575/console

Related to T4305
---
 swh/vault/tests/test_git_bare_cooker.py | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/swh/vault/tests/test_git_bare_cooker.py b/swh/vault/tests/test_git_bare_cooker.py
index 6510f7d..923f77a 100644
--- a/swh/vault/tests/test_git_bare_cooker.py
+++ b/swh/vault/tests/test_git_bare_cooker.py
@@ -42,16 +42,12 @@ from swh.model.model import (
     TimestampWithTimezone,
 )
 from swh.storage import get_storage
-from swh.storage.postgresql.db import Db as StorageBackend
+from swh.storage.postgresql.storage import Storage
 from swh.vault.cookers.git_bare import GitBareCooker
 from swh.vault.in_memory_backend import InMemoryVaultBackend
 
 storage_postgresql_proc = factories.postgresql_proc(
-    load=[
-        partial(
-            initialize_database_for_module, "storage", StorageBackend.current_version
-        )
-    ],
+    load=[partial(initialize_database_for_module, "storage", Storage.current_version)],
 )
 
 storage_postgresql = factories.postgresql("storage_postgresql_proc")
-- 
GitLab