1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
# Copyright 2018 Phidica Veia
# 2021 Nick Boultbee
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
import sqlite3
from senf import fsn2uri
from quodlibet.formats import AudioFile
from tests.helper import temp_filename
from quodlibet.library import SongFileLibrary
from . import PluginTestCase
def get_example_db(song_path, rating, playcount, skipcount, lastplayed, dateadded):
# create a temporary database in memory
db = sqlite3.connect(":memory:")
# create a simplified version of a banshee track table
csr = db.cursor()
csr.execute("""CREATE TABLE CoreTracks(
ArtistID INTEGER,
AlbumID INTEGER,
Uri TEXT,
Title TEXT,
Rating INTEGER,
PlayCount INTEGER,
SkipCount INTEGER,
LastPlayedStamp INTEGER,
DateAddedStamp INTEGER
)
""")
# insert song and save
song_uri = fsn2uri(song_path)
csr.execute(
"INSERT INTO CoreTracks VALUES (?,?,?,?,?,?,?,?,?)",
(1, 1, song_uri, "Music", rating, playcount, skipcount, lastplayed, dateadded),
)
db.commit()
# give the user the in-memory database
return db
class TBansheeImport(PluginTestCase):
def setUp(self):
self.mod = self.modules["bansheeimport"]
def test(self):
lib = SongFileLibrary()
with temp_filename() as song_fn:
song = AudioFile({"~filename": song_fn})
song.sanitize()
lib.add([song])
# test recovery of basic song
data = {
"path": song("~filename"),
"rating": 1,
"playcount": 1,
"skipcount": 2,
"lastplayed": 1371802107,
"added": 1260691996,
}
db = get_example_db(
data["path"],
data["rating"],
data["playcount"],
data["skipcount"],
data["lastplayed"],
data["added"],
)
importer = self.mod.BansheeDBImporter(lib)
importer.read(db)
count = importer.finish()
db.close()
self.assertEqual(song("~#rating"), data["rating"] / 5.0)
self.assertEqual(song("~#playcount"), data["playcount"])
self.assertEqual(song("~#skipcount"), data["skipcount"])
self.assertEqual(song("~#lastplayed"), data["lastplayed"])
self.assertEqual(song("~#added"), data["added"])
self.assertEqual(count, 1)
# test recovery of different version of same song
data_mod = {
"path": song("~filename"),
"rating": 2,
"playcount": 4,
"skipcount": 1,
"lastplayed": data["lastplayed"] - 1,
"added": data["added"] + 1,
}
db = get_example_db(
data_mod["path"],
data_mod["rating"],
data_mod["playcount"],
data_mod["skipcount"],
data_mod["lastplayed"],
data_mod["added"],
)
importer = self.mod.BansheeDBImporter(lib)
importer.read(db)
count = importer.finish()
db.close()
self.assertEqual(song("~#rating"), data_mod["rating"] / 5.0)
self.assertEqual(song("~#playcount"), data_mod["playcount"])
self.assertEqual(song("~#skipcount"), data_mod["skipcount"])
self.assertEqual(song("~#lastplayed"), data["lastplayed"])
self.assertEqual(song("~#added"), data["added"])
self.assertEqual(count, 1)
# test that no recovery is performed when data is identical
db = get_example_db(
data_mod["path"],
data_mod["rating"],
data_mod["playcount"],
data_mod["skipcount"],
data_mod["lastplayed"],
data_mod["added"],
)
importer = self.mod.BansheeDBImporter(lib)
importer.read(db)
count = importer.finish()
db.close()
self.assertEqual(count, 0)
|
