
from django.test import TestCase
from .models import Address


class ModelsTest(TestCase):
    def test_create_address(self):
        a = Address()
        a.street = '23, Test Street'
        a.town = 'London'
        a.postcode = 'xxxxx'
        a.country = 'UK'
        a.save()
        self.assertTrue(a.id >= 0)


from django.test import TestCase
from .models import Address, Venue


class ModelsTest(TestCase):
    def test_create_address(self):
        a = Address()
        a.street = '23, Test Street'
        a.town = 'London'
        a.postcode = 'xxxxx'
        a.country = 'UK'
        a.save()
        self.assertTrue(a.id >= 0)

    def test_create_venue(self):
        # Create the address first
        a = Address()
        a.street = '23, Test Street'
        a.town = 'London'
        a.postcode = 'xxxxx'
        a.country = 'UK'
        a.save()

        # Create the venue
        v = Venue()
        v.name = 'Venue Test'
        v.website = 'www.myvenue.com'
        v.address = a
        v.save()
        self.assertEqual(v.address.town, 'London')
        self.assertTrue(v.id >= 0)

Add Venue model creation test case
Add Venue model creation test case

  
  from django.test import TestCase
- from .models import Address
+ from .models import Address, Venue
  
  
  class ModelsTest(TestCase):
      def test_create_address(self):
          a = Address()
          a.street = '23, Test Street'
          a.town = 'London'
          a.postcode = 'xxxxx'
          a.country = 'UK'
          a.save()
          self.assertTrue(a.id >= 0)
  
+     def test_create_venue(self):
+         # Create the address first
+         a = Address()
+         a.street = '23, Test Street'
+         a.town = 'London'
+         a.postcode = 'xxxxx'
+         a.country = 'UK'
+         a.save()
+ 
+         # Create the venue
+         v = Venue()
+         v.name = 'Venue Test'
+         v.website = 'www.myvenue.com'
+         v.address = a
+         v.save()
+         self.assertEqual(v.address.town, 'London')
+         self.assertTrue(v.id >= 0)
+ 
Add Venue model creation test case
## Code Before:

from django.test import TestCase
from .models import Address


class ModelsTest(TestCase):
    def test_create_address(self):
        a = Address()
        a.street = '23, Test Street'
        a.town = 'London'
        a.postcode = 'xxxxx'
        a.country = 'UK'
        a.save()
        self.assertTrue(a.id >= 0)

## Instruction:
Add Venue model creation test case
## Code After:

from django.test import TestCase
from .models import Address, Venue


class ModelsTest(TestCase):
    def test_create_address(self):
        a = Address()
        a.street = '23, Test Street'
        a.town = 'London'
        a.postcode = 'xxxxx'
        a.country = 'UK'
        a.save()
        self.assertTrue(a.id >= 0)

    def test_create_venue(self):
        # Create the address first
        a = Address()
        a.street = '23, Test Street'
        a.town = 'London'
        a.postcode = 'xxxxx'
        a.country = 'UK'
        a.save()

        # Create the venue
        v = Venue()
        v.name = 'Venue Test'
        v.website = 'www.myvenue.com'
        v.address = a
        v.save()
        self.assertEqual(v.address.town, 'London')
        self.assertTrue(v.id >= 0)
