Source code for intranet.apps.bus.tests

from django.urls import reverse

from ...test.ion_test import IonTestCase
from .models import Route


[docs]class BusTest(IonTestCase):
    """Test for bus module"""

[docs]    def test_bus(self):
        self.login()

        self.assertEqual(self.client.get(reverse("bus")).status_code, 200)

[docs]    def test_routes(self):
        route = Route.objects.get_or_create(route_name="JT-101", bus_number="JT-101")[0]
        route.status = "a"
        route.space = "_1"
        route.save()

        route.reset_status()

        self.assertEqual(route.status, "o")
        self.assertEqual(route.space, "")

[docs]    def test_route_representation(self):
        route = Route.objects.get_or_create(route_name="JT-101", bus_number="JT-101")[0]
        route_str = str(route)

        self.assertEqual(route.route_name, route_str)