from django.test import TestCase

from org.models import Organization
from ..models import Bus


class BusModelTest(TestCase):

    def setUp(self):
        self.org = Organization.objects.create(
            name='Some Org',
            logo='/media/logos/some-org-logo.jpg',
            description='We are a familiar condominium',
            rules='Please check our conduct code page at https://some-url.foo'
        )
        self.bus = Bus.objects.create(
            name='Bus 1',
            organization=self.org
        )

    def test_str(self):
        self.assertEqual('Bus 1', str(self.bus))

    def test_can_create(self):
        self.assertTrue(Bus.objects.exists())

from django.test import TestCase

from org.models import Organization
from ..models import Bus, Route


class BusModelTest(TestCase):

    def setUp(self):
        self.org = Organization.objects.create(
            name='Some Org',
            logo='/media/logos/some-org-logo.jpg',
            description='We are a familiar condominium',
            rules='Please check our conduct code page at https://some-url.foo'
        )
        self.bus = Bus.objects.create(
            name='Bus 1',
            organization=self.org
        )

    def test_str(self):
        self.assertEqual('Bus 1', str(self.bus))

    def test_can_create(self):
        self.assertTrue(Bus.objects.exists())


class RouteModelTest(TestCase):

    def setUp(self):
        self.org = Organization.objects.create(
            name='Some Org',
            logo='/media/logos/some-org-logo.jpg',
            description='We are a familiar condominium',
            rules='Please check our conduct code page at https://some-url.foo'
        )
        self.route = Route.objects.create(
            name='Route 1',
            organization=self.org
        )

    def test_str(self):
        self.assertEqual('Route 1', str(self.route))

    def test_can_create(self):
        self.assertTrue(Route.objects.exists())

Add some Route model tests
Add some Route model tests

  from django.test import TestCase
  
  from org.models import Organization
- from ..models import Bus
+ from ..models import Bus, Route
  
  
  class BusModelTest(TestCase):
  
      def setUp(self):
          self.org = Organization.objects.create(
              name='Some Org',
              logo='/media/logos/some-org-logo.jpg',
              description='We are a familiar condominium',
              rules='Please check our conduct code page at https://some-url.foo'
          )
          self.bus = Bus.objects.create(
              name='Bus 1',
              organization=self.org
          )
  
      def test_str(self):
          self.assertEqual('Bus 1', str(self.bus))
  
      def test_can_create(self):
          self.assertTrue(Bus.objects.exists())
  
+ 
+ class RouteModelTest(TestCase):
+ 
+     def setUp(self):
+         self.org = Organization.objects.create(
+             name='Some Org',
+             logo='/media/logos/some-org-logo.jpg',
+             description='We are a familiar condominium',
+             rules='Please check our conduct code page at https://some-url.foo'
+         )
+         self.route = Route.objects.create(
+             name='Route 1',
+             organization=self.org
+         )
+ 
+     def test_str(self):
+         self.assertEqual('Route 1', str(self.route))
+ 
+     def test_can_create(self):
+         self.assertTrue(Route.objects.exists())
+ 
Add some Route model tests
## Code Before:
from django.test import TestCase

from org.models import Organization
from ..models import Bus


class BusModelTest(TestCase):

    def setUp(self):
        self.org = Organization.objects.create(
            name='Some Org',
            logo='/media/logos/some-org-logo.jpg',
            description='We are a familiar condominium',
            rules='Please check our conduct code page at https://some-url.foo'
        )
        self.bus = Bus.objects.create(
            name='Bus 1',
            organization=self.org
        )

    def test_str(self):
        self.assertEqual('Bus 1', str(self.bus))

    def test_can_create(self):
        self.assertTrue(Bus.objects.exists())

## Instruction:
Add some Route model tests
## Code After:
from django.test import TestCase

from org.models import Organization
from ..models import Bus, Route


class BusModelTest(TestCase):

    def setUp(self):
        self.org = Organization.objects.create(
            name='Some Org',
            logo='/media/logos/some-org-logo.jpg',
            description='We are a familiar condominium',
            rules='Please check our conduct code page at https://some-url.foo'
        )
        self.bus = Bus.objects.create(
            name='Bus 1',
            organization=self.org
        )

    def test_str(self):
        self.assertEqual('Bus 1', str(self.bus))

    def test_can_create(self):
        self.assertTrue(Bus.objects.exists())


class RouteModelTest(TestCase):

    def setUp(self):
        self.org = Organization.objects.create(
            name='Some Org',
            logo='/media/logos/some-org-logo.jpg',
            description='We are a familiar condominium',
            rules='Please check our conduct code page at https://some-url.foo'
        )
        self.route = Route.objects.create(
            name='Route 1',
            organization=self.org
        )

    def test_str(self):
        self.assertEqual('Route 1', str(self.route))

    def test_can_create(self):
        self.assertTrue(Route.objects.exists())
