def setUp(self):
descStoreTest = os.path.realpath(
os.path.dirname(__file__)) + '/descStoreTest'
if not os.path.isdir(descStoreTest):
os.makedirs(descStoreTest)
self.descDir = os.path.realpath(descStoreTest)
self.trackPointStore = TrackPointFileStore(self.descDir)
def setUp(self):
descStoreTest = os.path.realpath(os.path.dirname(__file__)) + '/descStoreTest'
if not os.path.isdir(descStoreTest):
os.makedirs(descStoreTest);
self.descDir = os.path.realpath(descStoreTest)
self.trackPointStore = TrackPointFileStore(self.descDir)
def setUp(self):
currentDir = os.path.dirname(__file__)
self.fileLog = fileLog = os.path.realpath(currentDir) + '/1msource.log'
self.descDir = descDir = os.path.realpath(
os.path.realpath(currentDir) + '/../test')
trackPointStoreStrategy = TrackPointFileStore(descDir)
trackPointLineParseStrategy = ParserLine()
self.heatClickLog = ClickHeatLog(fileLog, trackPointLineParseStrategy,
trackPointStoreStrategy)
class TestTrackPointFileStoreSuccess(unittest.TestCase):
def setUp(self):
descStoreTest = os.path.realpath(os.path.dirname(__file__)) + '/descStoreTest'
if not os.path.isdir(descStoreTest):
os.makedirs(descStoreTest);
self.descDir = os.path.realpath(descStoreTest)
self.trackPointStore = TrackPointFileStore(self.descDir)
def tearDown(self):
#remove all content in folder
shutil.rmtree(self.descDir)
# pass
def testSaveTrackPointMakeDescFolderBySiteNameAndGroupName(self):
trackPoint = TrackPoint(x='1',y='2',width='1024',browser='firefox',c='1',referUrl='http://www.dantri.com',day='2012-10-07',site='mimo',group='home')
self.assertFalse(os.path.exists(self.descDir+'/mimo,home'))
self.trackPointStore.saveTrackPoint(trackPoint)
self.assertTrue(os.path.isdir(self.descDir+'/mimo,home'))
def testSaveTrackPointMakeFileLogDatabaseByDay(self):
trackPoint = TrackPoint(x='1',y='2',width='1024',browser='firefox',c='1',referUrl='http://www.dantri.com',day='2012-10-07',site='mimo',group='home')
self.trackPointStore.saveTrackPoint(trackPoint)
self.assertTrue(os.path.isfile(self.descDir+'/mimo,home/2012-10-07.log'))
self.assertTrue(os.path.isfile(self.descDir+'/mimo,home/url.txt'))
def testSaveTrackPointToSameDescDataIntegrity(self):
trackPoint = TrackPoint(x='1',y='2',width='1024',browser='firefox',c='1',referUrl='http://www.dantri.com',day='2012-10-07',site='mimo',group='home')
trackPoint2 = TrackPoint(x='2',y='2',width='1024',browser='firefox',c='1',referUrl='http://www.dantri.com',day='2012-10-07',site='mimo',group='home')
self.trackPointStore.saveTrackPoint(trackPoint)
self.trackPointStore.saveTrackPoint(trackPoint2)
fileUrlOpen = io.open(self.descDir+'/mimo,home/url.txt', 'r')
self.assertEqual('http://www.dantri.com>0>0>0', fileUrlOpen.readline())
fileUrlOpen.close()
fileDayLogOpen = io.open(self.descDir+'/mimo,home/2012-10-07.log', 'r')
self.assertEqual('1|2|1024|firefox|1\n', fileDayLogOpen.readline())
self.assertEqual('2|2|1024|firefox|1\n', fileDayLogOpen.readline())
fileDayLogOpen.close()
def testSaveTrackPointDataIntegrityWithMultiTrackPointMultiFolder(self):
trackPoint = TrackPoint(x='1',y='2',width='1024',browser='firefox',c='1',referUrl='http://www.dantri.com',day='2012-10-07',site='mimo',group='home')
trackPoint2 = TrackPoint(x='2',y='2',width='1024',browser='chrome',c='1',referUrl='http://www.vnexpress.com',day='2012-10-07',site='mimo',group='profile')
self.trackPointStore.saveTrackPoint(trackPoint)
self.trackPointStore.saveTrackPoint(trackPoint2)
fileUrlOpen = io.open(self.descDir+'/mimo,home/url.txt', 'r')
self.assertEqual('http://www.dantri.com>0>0>0', fileUrlOpen.readline())
fileUrlOpen.close()
fileUrlOpen = io.open(self.descDir+'/mimo,profile/url.txt', 'r')
self.assertEqual('http://www.vnexpress.com>0>0>0', fileUrlOpen.readline())
fileUrlOpen.close()
fileDayLogOpen = io.open(self.descDir+'/mimo,home/2012-10-07.log', 'r')
self.assertEqual('1|2|1024|firefox|1\n', fileDayLogOpen.readline())
fileDayLogOpen.close()
fileDayLogOpen = io.open(self.descDir+'/mimo,profile/2012-10-07.log', 'r')
self.assertEqual('2|2|1024|chrome|1\n', fileDayLogOpen.readline())
fileDayLogOpen.close()
def testSaveTrackPointsDataIntegrityWithMultiTrackPointMultiFolder(self):
trackPoint = TrackPoint(x='1',y='2',width='1024',browser='firefox',c='1',referUrl='http://www.dantri.com',day='2012-10-07',site='mimo',group='home')
trackPoint2 = TrackPoint(x='2',y='2',width='1024',browser='chrome',c='1',referUrl='http://www.vnexpress.com',day='2012-10-07',site='mimo',group='profile')
self.trackPointStore.saveTrackPoints([trackPoint,trackPoint2])
fileUrlOpen = io.open(self.descDir+'/mimo,home/url.txt', 'r')
self.assertEqual('http://www.dantri.com>0>0>0', fileUrlOpen.readline())
fileUrlOpen.close()
fileUrlOpen = io.open(self.descDir+'/mimo,profile/url.txt', 'r')
self.assertEqual('http://www.vnexpress.com>0>0>0', fileUrlOpen.readline())
fileUrlOpen.close()
fileDayLogOpen = io.open(self.descDir+'/mimo,home/2012-10-07.log', 'r')
self.assertEqual('1|2|1024|firefox|1\n', fileDayLogOpen.readline())
fileDayLogOpen.close()
fileDayLogOpen = io.open(self.descDir+'/mimo,profile/2012-10-07.log', 'r')
self.assertEqual('2|2|1024|chrome|1\n', fileDayLogOpen.readline())
fileDayLogOpen.close()
class TestTrackPointFileStoreSuccess(unittest.TestCase):
def setUp(self):
descStoreTest = os.path.realpath(
os.path.dirname(__file__)) + '/descStoreTest'
if not os.path.isdir(descStoreTest):
os.makedirs(descStoreTest)
self.descDir = os.path.realpath(descStoreTest)
self.trackPointStore = TrackPointFileStore(self.descDir)
def tearDown(self):
#remove all content in folder
shutil.rmtree(self.descDir)
# pass
def testSaveTrackPointMakeDescFolderBySiteNameAndGroupName(self):
trackPoint = TrackPoint(x='1',
y='2',
width='1024',
browser='firefox',
c='1',
referUrl='http://www.dantri.com',
day='2012-10-07',
site='mimo',
group='home')
self.assertFalse(os.path.exists(self.descDir + '/mimo,home'))
self.trackPointStore.saveTrackPoint(trackPoint)
self.assertTrue(os.path.isdir(self.descDir + '/mimo,home'))
def testSaveTrackPointMakeFileLogDatabaseByDay(self):
trackPoint = TrackPoint(x='1',
y='2',
width='1024',
browser='firefox',
c='1',
referUrl='http://www.dantri.com',
day='2012-10-07',
site='mimo',
group='home')
self.trackPointStore.saveTrackPoint(trackPoint)
self.assertTrue(
os.path.isfile(self.descDir + '/mimo,home/2012-10-07.log'))
self.assertTrue(os.path.isfile(self.descDir + '/mimo,home/url.txt'))
def testSaveTrackPointToSameDescDataIntegrity(self):
trackPoint = TrackPoint(x='1',
y='2',
width='1024',
browser='firefox',
c='1',
referUrl='http://www.dantri.com',
day='2012-10-07',
site='mimo',
group='home')
trackPoint2 = TrackPoint(x='2',
y='2',
width='1024',
browser='firefox',
c='1',
referUrl='http://www.dantri.com',
day='2012-10-07',
site='mimo',
group='home')
self.trackPointStore.saveTrackPoint(trackPoint)
self.trackPointStore.saveTrackPoint(trackPoint2)
fileUrlOpen = io.open(self.descDir + '/mimo,home/url.txt', 'r')
self.assertEqual('http://www.dantri.com>0>0>0', fileUrlOpen.readline())
fileUrlOpen.close()
fileDayLogOpen = io.open(self.descDir + '/mimo,home/2012-10-07.log',
'r')
self.assertEqual('1|2|1024|firefox|1\n', fileDayLogOpen.readline())
self.assertEqual('2|2|1024|firefox|1\n', fileDayLogOpen.readline())
fileDayLogOpen.close()
def testSaveTrackPointDataIntegrityWithMultiTrackPointMultiFolder(self):
trackPoint = TrackPoint(x='1',
y='2',
width='1024',
browser='firefox',
c='1',
referUrl='http://www.dantri.com',
day='2012-10-07',
site='mimo',
group='home')
trackPoint2 = TrackPoint(x='2',
y='2',
width='1024',
browser='chrome',
c='1',
referUrl='http://www.vnexpress.com',
day='2012-10-07',
site='mimo',
group='profile')
self.trackPointStore.saveTrackPoint(trackPoint)
self.trackPointStore.saveTrackPoint(trackPoint2)
fileUrlOpen = io.open(self.descDir + '/mimo,home/url.txt', 'r')
self.assertEqual('http://www.dantri.com>0>0>0', fileUrlOpen.readline())
fileUrlOpen.close()
fileUrlOpen = io.open(self.descDir + '/mimo,profile/url.txt', 'r')
self.assertEqual('http://www.vnexpress.com>0>0>0',
fileUrlOpen.readline())
fileUrlOpen.close()
fileDayLogOpen = io.open(self.descDir + '/mimo,home/2012-10-07.log',
'r')
self.assertEqual('1|2|1024|firefox|1\n', fileDayLogOpen.readline())
fileDayLogOpen.close()
fileDayLogOpen = io.open(self.descDir + '/mimo,profile/2012-10-07.log',
'r')
self.assertEqual('2|2|1024|chrome|1\n', fileDayLogOpen.readline())
fileDayLogOpen.close()
def testSaveTrackPointsDataIntegrityWithMultiTrackPointMultiFolder(self):
trackPoint = TrackPoint(x='1',
y='2',
width='1024',
browser='firefox',
c='1',
referUrl='http://www.dantri.com',
day='2012-10-07',
site='mimo',
group='home')
trackPoint2 = TrackPoint(x='2',
y='2',
width='1024',
browser='chrome',
c='1',
referUrl='http://www.vnexpress.com',
day='2012-10-07',
site='mimo',
group='profile')
self.trackPointStore.saveTrackPoints([trackPoint, trackPoint2])
fileUrlOpen = io.open(self.descDir + '/mimo,home/url.txt', 'r')
self.assertEqual('http://www.dantri.com>0>0>0', fileUrlOpen.readline())
fileUrlOpen.close()
fileUrlOpen = io.open(self.descDir + '/mimo,profile/url.txt', 'r')
self.assertEqual('http://www.vnexpress.com>0>0>0',
fileUrlOpen.readline())
fileUrlOpen.close()
fileDayLogOpen = io.open(self.descDir + '/mimo,home/2012-10-07.log',
'r')
self.assertEqual('1|2|1024|firefox|1\n', fileDayLogOpen.readline())
fileDayLogOpen.close()
fileDayLogOpen = io.open(self.descDir + '/mimo,profile/2012-10-07.log',
'r')
self.assertEqual('2|2|1024|chrome|1\n', fileDayLogOpen.readline())
fileDayLogOpen.close()
