diff --git a/test_db.py b/test_db.py
index 1333abb..95aad06 100644
--- a/test_db.py
+++ b/test_db.py
@@ -1,30 +1,40 @@
 # Copyright (C) 2015  The Software Heritage developers
 # See the AUTHORS file at the top-level directory of this distribution
 # License: GNU General Public License version 3, or any later version
 # See top-level LICENSE file for more information
 
 import unittest
 
+from io import StringIO
 from nose.tools import istest
 from nose.plugins.attrib import attr
 
 from .db_testing import DbTestFixture
+from swh.storage.db import Db
 
 
 @attr('db')
 class TestDb(DbTestFixture, unittest.TestCase):
 
-    @istest
-    def connect(self):
-        self.assertFalse(self.conn.closed)
+    def setUp(self):
+        super().setUp()
+        self.db = Db(self.conn)
+
+    def tearDown(self):
+        self.db.conn.close()
+        super().tearDown()
 
     @istest
-    def insert_content(self):
-        self.cursor.execute(
-            'INSERT INTO content (sha1, sha1_git, sha256, length, status) ' +
-            'VALUES (%s, %s, %s, %s, %s)',
-            ('34973274ccef6ab4dfaaf86599792fa9c3fe4689',
-             'd81cc0710eb6cf9efd5b920a8453e1e07157b6cd',
-             '673650f936cb3b0a2f93ce09d81be10748b1b203'
-             'c19e8176b4eefc1964a0cf3a',
-             3, 'visible'))
+    def add_content(self):
+        cur = self.cursor
+        sha1 = '34973274ccef6ab4dfaaf86599792fa9c3fe4689'
+        self.db.content_mktemp(cur)
+        self.db.content_copy_to_temp(StringIO(
+            sha1 + '\t'
+            'd81cc0710eb6cf9efd5b920a8453e1e07157b6cd\t'
+            '673650f936cb3b0a2f93ce09d81be10748b1b203'
+            'c19e8176b4eefc1964a0cf3a\t' '3\n'), cur)
+        self.db.content_add_from_temp(cur)
+        self.cursor.execute('SELECT sha1 FROM content WHERE sha1 = %s',
+                            (sha1,))
+        self.assertEqual(self.cursor.fetchone(), (sha1,))