Differential D4508 Diff 16008 swh/scanner/tests/test_db.py

Changeset View

Standalone View

swh/scanner/tests/test_db.py

This file was added.

				# Copyright (C) 2020 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.scanner.db import Db

				from .data import present_swhids


				def test_db_create_from(tmp_path, live_server, test_sample_folder, test_swhids_sample):
				tmp_dbfile = tmp_path / "tmp_db.sqlite"
				api_url = live_server.url() + "/"
				config = {"web-api": {"url": api_url, "auth-token": None}}

				db = Db(tmp_dbfile)
				cur = db.conn.cursor()
				db.create_from(config, test_swhids_sample, cur)

				for swhid in present_swhids:
				cur = db.conn.cursor()
				assert db.check(swhid, cur)


				def test_db_create_from_one_not_present(
				tmp_path, live_server, test_sample_folder, test_swhids_sample
				):
				not_present_swhid = "swh:1:cnt:fa8eacf43d8646129ae8adfa1648f9307d999999"
				present_swhids.append(not_present_swhid)

				tmp_dbfile = tmp_path / "tmp_db.sqlite"
				api_url = live_server.url() + "/"
				config = {"web-api": {"url": api_url, "auth-token": None}}

				db = Db(tmp_dbfile)
				cur = db.conn.cursor()
				db.create_from(config, test_swhids_sample, cur)

				for swhid in present_swhids:
				vlorentzUnsubmitted Not Done Inline Actions Sorry I did not notice it earlier, but it looks like this line never runs. vlorentz: Sorry I did not notice it earlier, but it looks like this line never runs.
				cur = db.conn.cursor()
				if swhid != not_present_swhid:
				assert db.check(swhid, cur)
				else:
				assert not db.check(swhid, cur)


				def test_db_add(tmp_path):
				swhid = "swh:1:cnt:fa8eacf43d8646129ae8adfa1648f9307d999999"
				tmp_dbfile = tmp_path / "tmp_db.sqlite"
				db = Db(tmp_dbfile)
				cur = db.conn.cursor()

				db.create_table(cur)
				db.add(swhid, cur)
				assert db.check(swhid, cur)


				def test_db_add_non_present_swhid(tmp_path):
				swhid = "swh:1:cnt:fa8eacf43d8646129ae8adfa1648f9307d99999"
				tmp_dbfile = tmp_path / "tmp_db.sqlite"
				db = Db(tmp_dbfile)

				cur = db.conn.cursor()
				db.create_table(cur)
				assert not db.check(swhid, cur)