lundi 11 février 2019

How to build a test in python

import unittest
import requests
from bs4 import BeautifulSoup
from unittest.mock import MagicMock


class Test(unittest.TestCase):

    def getURL(url):
        url = "https://en.wikipedia.org/w/api.php?format=json&action=query&titles=Albert%20Einstein&prop=revisions&rvprop=content"
        resp = requests.get(url)
        soup = BeautifulSoup(url.text)
        # Extrait les liens de la page
        list_as = soup.find_all( "p")
        return list_as

    def testy(monkeypatch):
        url = "https://en.wikipedia.org/w/api.php?format=json&action=query&titles=Albert%20Einstein&prop=revisions&rvprop=content"

        mock_result= MagicMock() 
        mock_result.text = BeautifulSoup(url.text)      
        mock_get = MagicMock(return_value=mock_result)
        monkeypatch.setattr("requests.get",mock_get)
        result = requests.get(url)
        mock_get.assert_called_once(url)
        assert result.text == "Humboldt University of Berlin"


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


Aucun commentaire:

Enregistrer un commentaire