def test_search_release_offline_number_error(self):
name = inspect.currentframe().f_code.co_name
print("\n{} - {} - BEGIN".format(self.clname, name))
self.collection = Collection(False, self.db_path)
db_return = self.collection.search_release_offline('999999')
self.assertIsNone(db_return)
print("{} - {} - END".format(self.clname, name))
def __init__(self, _db_conn, _user_int, _userToken, _appIdentifier,
_db_file = False, _musicbrainz_user = False, _musicbrainz_pass = False):
self.user = _user_int # take an instance of the User_int class and set as attribute
self.cli = Collection_view_cli() # instantiate cli frontend class
self.collection = Collection(_db_conn, _db_file)
if self.collection.db_not_found == True:
self.cli.ask('Setting up DiscoBASE, press enter...')
super(Coll_ctrl_cli, self).setup_db(_db_file)
self.collection = Collection(_db_conn, _db_file)
if self.user.WANTS_ONLINE:
if not self.collection.discogs_connect(_userToken, _appIdentifier):
log.error("connecting to Discogs API, let's stay offline!\n")
else: # only try to initialize brainz if discogs is online already
self.brainz = Brainz(_musicbrainz_user, _musicbrainz_pass, _appIdentifier)
log.info("CTRL: Initial ONLINE status is %s", self.ONLINE)
self.first_track_on_release = ""
def test_track_report_snippet(self):
name = inspect.currentframe().f_code.co_name
print("\n{} - {} - BEGIN".format(self.clname, name))
self.collection = Collection(False, self.db_path)
db_return = self.collection.track_report_snippet(4, 133)
self.assertEqual(len(db_return), 3) # a snippet is always 3 tracks
# track 3
self.assertEqual(db_return[0]["track_pos"], 3)
self.assertEqual(db_return[0]["discogs_title"], "Material Love")
self.assertEqual(db_return[0]["d_track_no"], "B1")
self.assertEqual(db_return[0]["trans_rating"], "")
self.assertEqual(db_return[0]["key"], None)
self.assertEqual(db_return[0]["bpm"], 140)
self.assertEqual(db_return[0]["a_key"], None)
self.assertEqual(db_return[0]["a_bpm"], None)
# track 4
self.assertEqual(db_return[1]["track_pos"], 4)
self.assertEqual(db_return[1]["discogs_title"], "The Crane")
self.assertEqual(db_return[1]["d_track_no"], "AA")
self.assertEqual(db_return[1]["trans_rating"], "")
self.assertEqual(db_return[1]["key"], "Am")
self.assertEqual(db_return[1]["bpm"], 120)
self.assertEqual(db_return[1]["a_key"], None)
self.assertEqual(db_return[1]["a_bpm"], None)
# track 5
self.assertEqual(db_return[2]["track_pos"], 5)
print("{} - {} - END".format(self.clname, name))
def test_d_get_first_catno(self):
name = inspect.currentframe().f_code.co_name
print("\n{} - {} - BEGIN".format(self.clname, name))
self.collection = Collection(False, self.db_path)
if self.collection.discogs_connect(self.conf.discogs_token,
self.conf.discogs_appid):
print('We are ONLINE')
# we need to fetch a release by id first - let's also check it
d_return = self.collection.search_release_online('69092')
self.assertEqual(len(d_return),
1) # should be single release in a list!
self.assertEqual(int(d_return[0].id),
69092) # we get it as a string!
self.assertEqual(d_return[0].artists[0].name, 'Amon Tobin')
self.assertEqual(d_return[0].title, 'Out From Out Where')
#print(d_return[0].labels)
#for item in d_return[0].labels:
#print(dir(item.data))
#print(item.data.keys())
d_return_catno = self.collection.d_get_first_catno(
d_return[0].labels)
self.assertEqual(d_return_catno, 'ZEN 70')
else:
print('We are OFFLINE, testing if we properly fail!')
d_return = self.collection.search_release_online('69092')
self.assertFalse(d_return)
# if release can't be fetched online it does not make sense to ask
# d_get_first_catno() to retrieve it. This should be handled elsewhere
print("{} - {} - END".format(self.clname, name))
def test_search_release_online_text_multiple(self):
print("\nTestMix.search_release_online_text_multiple: BEGIN")
self.collection = Collection(False, self.db_path)
if self.collection.discogs_connect(self.conf.discogs_token,
self.conf.discogs_appid):
print('We are ONLINE')
d_return = self.collection.search_release_online(
'Amon Tobin') # artist or title
self.assertGreater(len(d_return),
770) # list with more than 770 Release objects
self.assertEqual(d_return.pages, 17) # _currently_ 17 pages
self.assertEqual(d_return.per_page, 50) # 50 per_page
self.assertEqual(d_return[0].id, 3618346)
self.assertEqual(d_return[0].artists[0].name, 'Amon Tobin')
self.assertEqual(d_return[0].title, 'Amon Tobin') # yes, really!
self.assertEqual(d_return[1].id, 3620565)
self.assertEqual(d_return[1].artists[0].name, 'Amon Tobin')
self.assertEqual(d_return[1].title, 'Amon Tobin') # yes, really!
self.assertEqual(d_return[1].tracklist[0].title, 'ISAM Live')
else:
print('We are OFFLINE, testing if we properly fail!')
db_return = self.collection.search_release_online(
'Amon Tobin') # artist or title
self.assertFalse(db_return)
print("TestMix.search_release_online_text_multiple: DONE\n")
def test_search_release_offline_text_error(self):
name = inspect.currentframe().f_code.co_name
print("\n{} - {} - BEGIN".format(self.clname, name))
self.collection = Collection(False, self.db_path)
db_return = self.collection.search_release_offline(
'XXX') # artist or title
self.assertIsNone(db_return) # returns None if nothing found
#self.assertEqual(db_return, []) # FIXME should this better be empty list?
print("{} - {} - END".format(self.clname, name))
def test_search_release_offline_number(self):
name = inspect.currentframe().f_code.co_name
print("\n{} - {} - BEGIN".format(self.clname, name))
self.collection = Collection(False, self.db_path)
db_return = self.collection.search_release_offline('123456')
self.assertIsNotNone(db_return)
self.assertEqual(len(db_return), 10) # should be 10 columns
self.assertEqual(db_return['discogs_id'], 123456)
self.assertEqual(db_return['d_artist'], 'Märtini Brös.')
self.assertEqual(db_return['discogs_title'], 'Material Love')
print("{} - {} - END".format(self.clname, name))
def test_stats_match_method_release(self):
name = inspect.currentframe().f_code.co_name
print("\n{} - {} - BEGIN".format(self.clname, name))
self.collection = Collection(False, self.db_path)
db_return = self.collection.stats_match_method_release()
#self.debug_db(db_return)
self.assertEqual(len(db_return), 3) # should be a list with 4 Rows
self.assertEqual(db_return[0]['m_match_method'], None)
self.assertEqual(db_return[1]['m_match_method'], 'CatNo (exact)')
self.assertEqual(db_return[2]['m_match_method'], "Discogs URL")
print("{} - {} - END".format(self.clname, name))
def test_search_release_track_offline_nothing(self):
name = inspect.currentframe().f_code.co_name
print("\n{} - {} - BEGIN".format(self.clname, name))
self.collection = Collection(False, self.db_path)
dbr = self.collection.search_release_track_offline(artist='',
release='',
track='')
#self.debug_db(dbr)
self.assertIsNotNone(dbr)
self.assertEqual(len(dbr), 0) # should be a list with 0 Rows
print("{} - {} - END".format(self.clname, name))
def test_search_release_track_offline_track(self):
name = inspect.currentframe().f_code.co_name
print("\n{} - {} - BEGIN".format(self.clname, name))
self.collection = Collection(False, self.db_path)
dbr = self.collection.search_release_track_offline(artist='',
release='',
track='Hedup!')
#self.debug_db(dbr)
self.assertIsNotNone(dbr)
self.assertEqual(len(dbr), 1) # should be a list with 1 Rows
self.assertEqual(dbr[0]['d_artist'], 'Märtini Brös.')
self.assertEqual(dbr[0]['discogs_title'], 'Material Love')
self.assertEqual(dbr[0]['d_track_name'], 'Hedup!')
print("{} - {} - END".format(self.clname, name))
def test_search_release_offline_text_multiple(self):
print("\nTestMix.search_release_offline_text_multiple: BEGIN")
self.collection = Collection(False, self.db_path)
db_return = self.collection.search_release_offline(
'Amon') # artist or title
self.assertIsNotNone(db_return)
self.assertEqual(len(db_return), 2) # should be a list with 2 Rows
self.assertEqual(db_return[0]['discogs_id'], 69092)
self.assertEqual(db_return[0]['d_artist'], 'Amon Tobin')
self.assertEqual(db_return[0]['discogs_title'], 'Out From Out Where')
self.assertEqual(db_return[1]['discogs_id'], 919698)
self.assertEqual(db_return[1]['d_artist'], 'Amon Tobin')
self.assertEqual(db_return[1]['discogs_title'], 'Foley Room')
print("TestMix.search_release_offline_text_multiple: DONE\n")
def test_search_release_track_offline_release(self):
name = inspect.currentframe().f_code.co_name
print("\n{} - {} - BEGIN".format(self.clname, name))
self.collection = Collection(False, self.db_path)
dbr = self.collection.search_release_track_offline(artist='',
release='material',
track='')
#self.debug_db(dbr)
self.assertIsNotNone(dbr)
self.assertEqual(len(dbr),
3) # it's one release but finds all track entries!
self.assertEqual(dbr[0]['d_artist'], 'Märtini Brös.')
self.assertEqual(dbr[1]['d_track_name'], 'Material Love')
self.assertEqual(dbr[2]['discogs_title'], 'Material Love')
print("{} - {} - END".format(self.clname, name))
def test_search_release_track_offline_artist_without_tracks(self):
name = inspect.currentframe().f_code.co_name
print("\n{} - {} - BEGIN".format(self.clname, name))
self.collection = Collection(False, self.db_path)
dbr = self.collection.search_release_track_offline(artist='Amon',
release='',
track='')
#self.debug_db(dbr)
self.assertIsNotNone(dbr)
self.assertEqual(len(dbr), 2) # should be a list with 2 Rows
self.assertEqual(dbr[0]['d_artist'], 'Amon Tobin')
self.assertEqual(dbr[0]['discogs_title'], 'Foley Room')
self.assertEqual(dbr[0]['d_artist'], 'Amon Tobin')
self.assertEqual(dbr[1]['discogs_title'], 'Out From Out Where')
print("{} - {} - END".format(self.clname, name))
def test_track_report_occurences(self):
name = inspect.currentframe().f_code.co_name
print("\n{} - {} - BEGIN".format(self.clname, name))
self.collection = Collection(False, self.db_path)
db_return = self.collection.track_report_occurences(123456, 'B2')
self.assertEqual(len(db_return), 11) # track was used 11 times
# check some occurences
self.assertEqual(db_return[0]["mix_id"], 125) # first occurence
self.assertEqual(db_return[0]["track_pos"], 2) # used at pos 2
self.assertEqual(db_return[0]["name"],
'test 125 last_tr, tr_of_one_mix, one_mix')
self.assertEqual(db_return[7]["mix_id"], 132) # 8th occurence
self.assertEqual(db_return[7]["track_pos"], 2) # used at pos 2
self.assertEqual(db_return[10]["mix_id"], 135) # 11th occurence
self.assertEqual(db_return[10]["track_pos"], 2) # used at pos 2
print("{} - {} - END".format(self.clname, name))
def test_get_tracks_by_bpm(self):
name = inspect.currentframe().f_code.co_name
print("\n{} - {} - BEGIN".format(self.clname, name))
self.collection = Collection(False, self.db_path)
db_return = self.collection.get_tracks_by_bpm(125, 6)
self.assertIsNotNone(db_return)
self.assertEqual(len(db_return), 3) # should be a list with 3 Rows
self.assertEqual(db_return[0]['d_artist'], 'Source Direct')
self.assertEqual(db_return[0]['d_track_no'], 'AA')
self.assertEqual(db_return[0]['chosen_bpm'], 120)
self.assertEqual(db_return[1]['d_artist'], 'Märtini Brös.')
self.assertEqual(db_return[1]['d_track_no'], 'A1')
self.assertEqual(db_return[1]['chosen_bpm'], 125)
self.assertEqual(db_return[2]['d_artist'], 'Märtini Brös.')
self.assertEqual(db_return[2]['d_track_no'], 'B2')
self.assertEqual(db_return[2]['chosen_bpm'], 130)
print("{} - {} - END".format(self.clname, name))
def test_get_tracks_by_key(self):
name = inspect.currentframe().f_code.co_name
print("\n{} - {} - BEGIN".format(self.clname, name))
self.collection = Collection(False, self.db_path)
db_return = self.collection.get_tracks_by_key("Am")
self.assertIsNotNone(db_return)
#for row in db_return:
# for field in row:
# print(field)
# print()
self.assertEqual(len(db_return), 2) # should be a list with 2 Rows
self.assertEqual(db_return[0]['d_artist'], 'Source Direct')
self.assertEqual(db_return[0]['d_track_no'], 'AA')
self.assertEqual(db_return[0]['chosen_bpm'], 120)
self.assertEqual(db_return[1]['d_artist'], 'Märtini Brös.')
self.assertEqual(db_return[1]['d_track_no'], 'A1')
self.assertEqual(db_return[1]['chosen_bpm'], 125)
print("{} - {} - END".format(self.clname, name))
def test_get_all_db_releases(self):
name = inspect.currentframe().f_code.co_name
print("\n{} - {} - BEGIN".format(self.clname, name))
# instantiate the Collection model class
self.collection = Collection(False, self.db_path)
db_return = self.collection.get_all_db_releases()
#self.debug_db(db_return)
self.assertIsNotNone(db_return)
self.assertEqual(len(db_return), 4)
self.assertEqual(db_return[2]['discogs_id'], 123456)
self.assertEqual(db_return[2]['discogs_title'], 'Material Love')
#self.assertEqual(db_return[2]['import_timestamp'], '2020-01-25 22:33:35')
self.assertEqual(db_return[2]['d_artist'], 'Märtini Brös.')
#self.assertEqual(db_return[2]['in_d_collection'], 1)
self.assertEqual(db_return[3]['discogs_id'], 8620643)
self.assertEqual(db_return[3]['discogs_title'], 'The Crane')
#self.assertEqual(db_return[3]['import_timestamp'], '2020-01-30 10:06:26')
self.assertEqual(db_return[3]['d_artist'], 'Source Direct')
#self.assertEqual(db_return[3]['in_d_collection'], 1)
print("{} - {} - END".format(self.clname, name))
def test_search_release_online_number(self):
print("\nTestMix.search_release_online_number: BEGIN")
self.collection = Collection(False, self.db_path)
if self.collection.discogs_connect(self.conf.discogs_token,
self.conf.discogs_appid):
print('We are ONLINE')
d_return = self.collection.search_release_online('69092')
#print(dir(d_return))
self.assertEqual(len(d_return),
1) # should be single release in a list!
self.assertEqual(int(d_return[0].id),
69092) # we get it as a string!
self.assertEqual(d_return[0].artists[0].name, 'Amon Tobin')
self.assertEqual(d_return[0].title, 'Out From Out Where')
else:
print('We are OFFLINE, testing if we properly fail!')
d_return = self.collection.search_release_online('69092')
self.assertFalse(
d_return
) # FIXME returns list because d.release not throwing error
print("TestMix.search_release_online_number: DONE\n")