41 lines
1.6 KiB
Python
41 lines
1.6 KiB
Python
from qprocess import write_csv
|
|
import csv
|
|
import unittest
|
|
import os
|
|
# Torrent Items needed for testing
|
|
# canidate['state'] {canidate["ratio"]} {torrent["tags"]} torrent["seeding_time"] {torrent["category"]} {torrent["name"][0:20]} {canidate["time_active"]}
|
|
class TestWriteCSV(unittest.TestCase):
|
|
def test_write_csv_dragnet(self):
|
|
self.cv = csv
|
|
outfile = './test_dragnet_outfile.csv'
|
|
state = 'downloading'
|
|
ratio = 1.05
|
|
tags = 'ipt'
|
|
added = 1
|
|
age = 240000
|
|
time = 123456
|
|
thash = 'asfasdf23412adfqwer'
|
|
tname = 'thisismynamehahahah'
|
|
trname = 'https://localhost.stackoverflow.tech/317332f1c125bc9c1b9b14fb8e054908/announce'
|
|
header = ['state','ratio','tags','added','age','time','thash','tname','trname']
|
|
row = [state,ratio,tags,added,age,time,thash,tname,trname]
|
|
write_csv(self.cv,outfile,header,row)
|
|
self.assertTrue(os.path.exists(outfile))
|
|
|
|
def test_write_csv_telemetry(self):
|
|
self.cv = csv
|
|
outfile = './test_telemetry_outfile.csv'
|
|
state = 'downloading'
|
|
ratio = 1.05
|
|
tags = 'ipt'
|
|
added = 1
|
|
thash = 'asfasdf23412adfqwer'
|
|
tname = 'thisismynamehahahah'
|
|
trname = 'https://localhost.stackoverflow.tech/317332f1c125bc9c1b9b14fb8e054908/announce'
|
|
header = ['state','ratio','tags','added','hash','name','tracker']
|
|
row = [state,ratio,tags,added,thash,tname,trname]
|
|
write_csv(self.cv,outfile,header,row)
|
|
self.assertTrue(os.path.exists(outfile))
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |