from django.db import models
from django.contrib.auth.models import User

class Author(models.Model):
    name = models.TextField(max_length=50)

class Category(models.Model):
    name = models.CharField(max_length=100)

class Article(models.Model):
    author = models.ForeignKey(Author, on_delete=models.CASCADE)
    title = models.CharField(max_length=200)
    body = models.TextField()
    category = models.ForeignKey(Category, on_delete=models.CASCADE)

from django.test import TestCase
from blog.models import Article, Author, Category


class TestContactModel(TestCase):
    def setUp(self):
        self.article = Article(author='jhon', title='how to test', body='this is body', category='djangooo')
        self.article.save()

    def test_contact_creation(self):
        self.assertEqual(article.objects.count(), 1)

    def test_contact_representation(self):
        self.assertEqual(self.article.title, str(self.article))