import unittest

from django.test.client import Client
from PersonasBackend.personas import models

class JsonTests( unittest.TestCase ):
    def setUp( self ):
        for persona in models.Persona.objects.all():
            persona.delete()
        self.cat = models.Category( name = "Random Things" )
        self.cat.save()
        self.persona = models.Persona(
            name = "Test Persona",
            category = self.cat,
            header_img_url = "http://www.blarg.com/myheader.png",
            footer_img_file = "footers/myfooter.png",
            status = "published"
            )
        self.persona.save()

    def tearDown( self ):
        self.persona.delete()
        self.cat.delete()

    def testCategoriesWorks( self ):
        # Just a smoke test to make sure nothing crashes...
        client = Client()
        response = client.get( "/en-US/personas_categories.dat" )
        assert response.status_code == 200

    def testAllPersonasWorks( self ):
        # Just a smoke test to make sure nothing crashes...
        client = Client()
        response = client.get( "/en-US/personas_all.dat" )
        assert response.status_code == 200

if __name__ == "__main__":
    unittest.main()