TestArticles.py

import os

from django.test import TestCase

from app import create_app

GET_ALL_ARTICLES = '/api/v1/articles'
config = os.getenv('APP_SETTINGS')


class ArticleTestCase(TestCase):
  def setUp(self):
    """Define setup configurations"""
    self.app = create_app(config)
    self.client = self.app.test_client()
    
  def test_get_null_articles(self):
    """TEST whether the API can get null articles(GET)"""
    res = self.client.get(GET_ALL_ARTICLES,
                          headers=dict(Authorization="Bearer " + self.login()),
                          content_type='application/json')
    resp_data = json.loads(res.data.decode())
    self.assertEqual(res.status_code, 404)
    self.assertEqual(resp_data['message'], "No articles yet")
      
  def test_get_all_articles(self):
    """TEST whether the API can get all articles(POST)"""
    res = self.client.get(GET_ALL_ARTICLES,
                          headers=dict(Authorization="Bearer " + self.login()),
                          content_type='application/json')
    resp_data = json.loads(res.data.decode())
    self.assertEqual(res.status_code, 200)
    self.assertEqual(resp_data['message'], "Articles retrieved")


  

Thank you SharkDevs and Teatoller. I will implement that

You haven't imported the os module, kindly correct that.

Thank you John Wayodi. I have imported it.