def test_number1(self, mockconnection, mockinput):
     mocksql = mock.Mock()
     mocksql.cursor().commit.return_value = True
     mocksql.cursor().fetchone.return_value = (1, 'shrey', 'shrey', 'Shreya', 'Bangalore', 560034, 27, None, None, None)
     g = Admin(mocksql, 1)
     m = g.number_1()
     assert m is True
 def test_employeedetails(self, mockconnection):
     mocksql = mock.Mock()
     mocksql.cursor().fetchall.return_value = [(1, 'shrey', 'shrey', 'Shreya', 'Bangalore', 560034, 27, None, None,
                                                None, 'shreya', 'shreya', 'shreya', 'shreya')]
     g = Admin(mocksql, 1)
     m = g.employee_details()
     assert m is True
 def test_number4(self, mockconnection, mockinput):
     mocksql = mock.Mock()
     mocksql.cursor().fetchall.return_value = [(1, 'shrey', 'shrey', 'Shreya', 'Bangalore', 560034, 27, None, None,
                                                None, 'shreya', 'shreya', 'shreya', 'shreya')]
     g = Admin(mocksql, 1)
     m = g.number_4()
     assert m is True
 def test_addnewroute(self, mockconnection, mockinput):
     mocksql = mock.Mock()
     mocksql.cursor().commit.return_value = True
     mocksql.cursor().fetchall.return_value = [(1, 'shrey', 'shrey', 'Shreya', 'Bangalore', 560034, 27, None, None,
                                                None, 'shreya', 'shreya', 'shreya', 'shreya')]
     g = Admin(mocksql, 1)
     m = g.add_new_route()
     assert m is True
Beispiel #5
0
    def admin_menu(self, con, id):
        """
            function to view admin menu
        :param con: database connection object
        :param id: admin id
        :return:
        """
        entry = ''
        while entry != 'exit':
            print('Following are the functions you can perform.')
            print('1 - Check total bookings')
            print('2 - Check bookings of an Employee')
            print('3 - Add new cab')
            print('4 - Update cab route and timings')
            print('5 - Add new Employee')
            print('6 - Update employee details')
            print('7 - Delete an employee')

            entry = builtins.input(
                'Please enter the code associated with the function you want to perform:'
            )
            if entry != 'exit':
                s = Admin(con, id)
                s.indirect(entry)
 def test_number4_exception(self, mockconnection, mockinput):
     mocksql = mock.Mock()
     mocksql.cursor().execute.side_effect = sqlite3.Error
     g = Admin(mocksql, 1)
     m = g.number_4()
     assert m is False
 def test_number7(self, mockconnection, mockinput, mockfunc):
     mocksql = mock.Mock()
     mocksql.cursor().commit.return_value = True
     g = Admin(mocksql, 1)
     m = g.number_7()
     assert m is True
 def test_addnewroute_exception(self, mockconnection, mockinput):
     mocksql = mock.Mock()
     mocksql.cursor().execute.side_effect = sqlite3.Error
     g = Admin(mocksql, 1)
     m = g.add_new_route()
     assert m is False
 def test_changeroute(self, mockconnection, mockinput):
     mocksql = mock.Mock()
     mocksql.cursor().commit.return_value = True
     g = Admin(mocksql, 1)
     m = g.change_route()
     assert m is True
 def test_addnewtime(self, mockconnection, mockinput):
     mocksql = mock.Mock()
     mocksql.cursor().commit.return_value = True
     g = Admin(mocksql, 1)
     m = g.add_new_time()
     assert m is True
 def test_employeedetails_exception(self, mockconnection, mockinput):
     mocksql = mock.Mock()
     mocksql.cursor().execute.side_effect = sqlite3.Error
     g = Admin(mocksql, 1)
     m = g.change_cab_time()
     assert m is False
 def test_indirect(self, mockfunction, mockconnection):
     mocksql = mock.Mock()
     g = Admin(mocksql, 1)
     m = g.indirect(1)
     assert m is True
 def test_employeedetails_exception(self, mockconnection):
     mocksql = mock.Mock()
     mocksql.cursor().execute.side_effect = sqlite3.Error
     g = Admin(mocksql, 1)
     m = g.employee_details()
     assert m is False