test_db.py
# Copyright (C) 2015-2017 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
from swh.model.hashutil import hash_to_bytes
def test_add_content(swh_storage):
with swh_storage.db() as db:
cur = db.cursor()
sha1 = hash_to_bytes('34973274ccef6ab4dfaaf86599792fa9c3fe4689')
db.mktemp('content', cur)
db.copy_to([{
'sha1': sha1,
'sha1_git': hash_to_bytes(
'd81cc0710eb6cf9efd5b920a8453e1e07157b6cd'),
'sha256': hash_to_bytes(
'673650f936cb3b0a2f93ce09d81be107'
'48b1b203c19e8176b4eefc1964a0cf3a'),
'blake2s256': hash_to_bytes('69217a3079908094e11121d042354a7c'
'1f55b6482ca1a51e1b250dfd1ed0eef9'),
'length': 3}],
'tmp_content',
['sha1', 'sha1_git', 'sha256', 'blake2s256', 'length'],
cur)
db.content_add_from_temp(cur)
cur.execute('SELECT sha1 FROM content WHERE sha1 = %s',
(sha1,))
assert cur.fetchone()[0] == sha1