class TestViewFunctions(unittest.TestCase):
def setUp(self):
# Create our controller
self.controller = Controller()
#create our view
self.view = View(self.controller.get_model())
# Create an example machine
self.machine1 = Machine(1, MachineType.TREADMILL, [1, 1, 1])
def testGetMachineType(self):
# Add our example machine
self.controller.add_machine(self.machine1)
# self.view.print_machine_type(1)
self.assertEqual(self.view.get_machine_type(self.machine1.id),self.machine1.type)
def testDisplayStatus(self):
# Add our example machine
self.controller.add_machine(self.machine1)
final_string = self.view.format_line(["ID", "Machine Type"]) + \
self.view.format_line([self.machine1.id, self.machine1.get_type()])
self.assertEqual(self.view.display_status(), final_string)
def setUp(self):
# Create our controller
self.controller = Controller()
#create our view
self.view = View(self.controller.get_model())
# Create an example machine
self.machine1 = Machine(1, MachineType.TREADMILL, [1, 1, 1])
def welcome(controller, view):
"""
The welcome and setup function for the test
:controller: The controller created in the main loop
:view: The view created in the main loop
:returns: TODO
"""
# Add our first machine
machine1 = Machine(1, MachineType.BICYCLE, [1, 1, 1], True)
machine1.set_status(Status.OPEN)
controller.add_machine(machine1)
# Add our second machine
machine2 = Machine(2, MachineType.TREADMILL, [2, 3, 5], True)
controller.add_machine(machine2)
machine2.set_status(Status.BUSY)
# Add our third machine
machine3 = Machine(17, MachineType.BICYCLE, [3, 13, 10], True)
controller.add_machine(machine3)
# Add our fourth machine
machine4 = Machine(12, MachineType.BICYCLE, [1, 2, 3], True)
controller.add_machine(machine4)
# Add our fifth machine
machine5 = Machine(7, MachineType.BICYCLE, [6, 6, 1], True)
controller.add_machine(machine5)
print("This is a text based view of the locations of the machines")
print(view.display_locations())
print("Now for a GUI view of the locations of the machines")
view.start_gui()
print("Now let's add a custom machine")
id_num = int(input("Please input an ID number: "))
location = []
location.append(int(input("Please input the x location of the machine: ")))
location.append(int(input("Please input the y location of the machine: ")))
location.append(1)
custom_machine = Machine(id_num, MachineType.TREADMILL, location, True)
controller.add_machine(custom_machine)
view2 = GTKView(controller.get_model())
print("How we will display the new GUI")
view2.start_gui()