def test_LastLineMock(self):
readCSVFile = ReadCSVFile()
propertydata = []
propertydata.append("emailAddress, firstName, lastName, password")
propertydata.append(
"test1:testemail, testfirst, testlast, testpassword")
propertydata.append(
"test2:testemail2, testfirst2, testlast2, testpassword2")
readCSVFile.getFileData = MagicMock(return_value=propertydata)
self.assertEqual(
"test2:testemail2, testfirst2, testlast2, testpassword2",
readCSVFile.getLastLines("customer.csv", 1))
class testReadCSV(unittest.TestCase):
readFile = ReadCSVFile()
def test_CSV(self):
self.assertEqual(['07711223344', 'Derek', 'Somerville', '1234'],
self.readFile)
def test_fakeData(self):
testPhone = ['07712321234356', '07789462734', '07654756475']
testPasswords = ['9999', '6666', '4444']
items = [{
"phone": random.choice(testPhone),
"password": random.choice(testPasswords)
} for _ in range(3)]
self.assertEqual(items, self.readFile)
return self
def test_readFileData(self):
self.readFile.getFileData = MagicMock(
return_value=['07711223344', 'Derek', 'Somerville', '1234'])
self.assertEqual(
self.readFile.getFileData("customer",
['07722334455', 'Matt', 'Barr', '4321']))
def test_FakeList(self):
logIn = LogIn()
customers = Customers()
main = Main()
testList = [
'07711122233', '077897654321', '07896789894', '07123421234'
]
displayFakeList = LogIn()
displayFakeList.getPassword = MagicMock(return_value=testList.pop())
self.assertEqual('Enter phone number', displayFakeList.display())
class Customers:
readCSVFile = ReadCSVFile()
def setReadCSVFile(self, reader):
self.readCSVFile = reader
def loadCustomers(self):
customerData = self.readCSVFile.getFileData("customer.csv")
return customerData
def formatCustomers(self):
display = ""
customerData = self.loadCustomers()
for counter in range(1,len(customerData)):
display += customerData[counter][0] + " \n"
return display
def loadCustomers(self):
readCSVFile = ReadCSVFile()
customerData = readCSVFile.getFileData("customer")
return customerData
def test_FirstItem(self):
readCSVFile = ReadCSVFile()
self.assertEqual("emailAddress",
readCSVFile.getFileData("customer.csv")[0][0])
def test_lastLine(self):
readCSVFile = ReadCSVFile()
self.assertEqual(
["*****@*****.**", "Ethan", "Kelly", "9876"],
readCSVFile.getLastLines("customer.csv", 1))
from src.ReadCSVFile import ReadCSVFile
readCSVFile = ReadCSVFile()
def loadCustomers():
customerData = readCSVFile.getFileData("customer.csv")
return customerData
def formatCustomers():
display = ""
customerData = loadCustomers()
for counter in range(1,len(customerData)):
display += customerData[counter][0] + "\n"
return display
def test_getFileData():
if "emailAddress" == ReadCSVFile.getFileData("customer.csv")[0][0]:
print("Pass","test_getFileData")
else:
print("fail", "test_getFileData")
def Load_Customers(self):
readCSVFile = ReadCSVFile()
customerData = readCSVFile.getFileData("customer.csv")
return customerData