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