Create test framework

1 files changed, 53 insertions, 0 deletions

diff --git a/tests/db/db_manager_test.py b/tests/db/db_manager_test.py
new file mode 100644
index 0000000..911c62d
--- /dev/null
+++ b/tests/db/db_manager_test.py
@@ -0,0 +1,53 @@
+from db.db_manager import DbManager
+
+
+class TestDbManager:
+
+    def test_create_tables(self):
+        assert DbManager.create_album_table == "CREATE TABLE IF NOT EXISTS "\
+            "album (id INTEGER PRIMARY KEY, name TEXT, date TEXT, "\
+            "musicbrainz_albumid TEXT)"
+
+        assert DbManager.create_album_artist_table == "CREATE TABLE IF NOT "\
+            "EXISTS album_artist (album_id INTEGER, artist_id INTEGER, "\
+            "CONSTRAINT ALBUM_ARTIST_PK PRIMARY KEY (album_id, artist_id), "\
+            "CONSTRAINT ALBUM_ARTIST_FK_ALBUM FOREIGN KEY (album_id) "\
+            "REFERENCES album(id), CONSTRAINT ALBUM_ARTIST_FK_ARTIST "\
+            "FOREIGN KEY(artist_id) REFERENCES artist(id))"
+
+        assert DbManager.create_album_track_table == "CREATE TABLE IF NOT "\
+            "EXISTS album_track (album_id INTEGER, track_id INTEGER, "\
+            "CONSTRAINT ALBUM_TRACK_PK PRIMARY KEY (album_id, track_id), "\
+            "FOREIGN KEY(album_id) REFERENCES album(id), FOREIGN "\
+            "KEY(track_id) REFERENCES track(id))"
+
+        assert DbManager.create_artist_table == "CREATE TABLE IF NOT EXISTS "\
+            "artist (id INTEGER PRIMARY KEY, name TEXT(2000000000), sortname "\
+            "TEXT(2000000000), musicbrainz_artistid TEXT(2000000000))"
+
+        assert DbManager.create_artist_track_table == "CREATE TABLE IF NOT "\
+            "EXISTS artist_track (artist_id INTEGER, track_id INTEGER, "\
+            "CONSTRAINT ARTIST_TRACK_PK PRIMARY KEY (artist_id, track_id), "\
+            "FOREIGN KEY(artist_id) REFERENCES artist(id), FOREIGN "\
+            "KEY(track_id) REFERENCES track(id))"
+
+        assert DbManager.create_track_table == "CREATE TABLE IF NOT EXISTS "\
+            "track (id INTEGER, tracknumber INTEGER, name TEXT(2000000000), "\
+            "grouping TEXT(2000000000), filename TEXT(2000000000), "\
+            "CONSTRAINT TRACK_PK PRIMARY KEY (id))"
+
+        assert DbManager.create_musicbrainz_artist_index == "CREATE UNIQUE "\
+            "INDEX IF NOT EXISTS artist_musicbrainz_artistid ON "\
+            "artist(musicbrainz_artistid ASC)"
+
+        assert DbManager.create_track_filename_index == "CREATE INDEX IF NOT "\
+            "EXISTS track_filename_IDX ON track(filename)"
+
+        assert DbManager.create_track_grouping_index == "CREATE INDEX IF NOT "\
+            "EXISTS track_grouping_IDX ON track(grouping)"
+
+        assert DbManager.create_track_name_index == "CREATE INDEX IF NOT "\
+            "EXISTS track_name_IDX ON track(name)"
+
+        assert DbManager.create_track_number_index == "CREATE INDEX IF NOT "\
+            "EXISTS track_tracknumber_IDX ON track(tracknumber)"