Remove unix timestamp duplication

This commit is contained in:
hugovk 2014-02-26 20:26:08 +02:00
parent 4074c4fae9
commit 1c1f56a62d

View file

@ -10,6 +10,9 @@ import pylast
class TestSequenceFunctions(unittest.TestCase): class TestSequenceFunctions(unittest.TestCase):
def unix_timestamp(self):
return int(time.mktime(datetime.datetime.now().timetuple()))
def setUp(self): def setUp(self):
self.username = "TODO" self.username = "TODO"
password_hash = "TODO" password_hash = "TODO"
@ -25,7 +28,7 @@ class TestSequenceFunctions(unittest.TestCase):
# Arrange # Arrange
artist = "Test Artist" artist = "Test Artist"
title = "Test Title" title = "Test Title"
timestamp = int(time.mktime(datetime.datetime.now().timetuple())) timestamp = self.unix_timestamp()
lastfm_user = self.network.get_user(self.username) lastfm_user = self.network.get_user(self.username)
# Act # Act
@ -42,7 +45,7 @@ class TestSequenceFunctions(unittest.TestCase):
# Arrange # Arrange
artist = "Test Artist 2" artist = "Test Artist 2"
title = "Test Title 2" title = "Test Title 2"
timestamp = int(time.mktime(datetime.datetime.now().timetuple())) timestamp = self.unix_timestamp()
library = pylast.Library(user = self.username, network = self.network) library = pylast.Library(user = self.username, network = self.network)
self.network.scrobble(artist = artist, title = title, timestamp = timestamp) self.network.scrobble(artist = artist, title = title, timestamp = timestamp)
lastfm_user = self.network.get_user(self.username) lastfm_user = self.network.get_user(self.username)