34 lines
845 B
Python
34 lines
845 B
Python
|
import unittest
|
||
|
|
import os.path
|
||
|
|
from bs4 import BeautifulSoup
|
||
|
|
|
||
|
|
|
||
|
|
class TestClient:
|
||
|
|
output_path = os.path.realpath('./output')
|
||
|
|
|
||
|
|
def get(self, path):
|
||
|
|
file_path = self.build_path(path)
|
||
|
|
content = "".join(open(file_path).readlines())
|
||
|
|
if path.endswith('html'):
|
||
|
|
content = BeautifulSoup(content, 'html.parser')
|
||
|
|
return content
|
||
|
|
|
||
|
|
def build_path(self, path):
|
||
|
|
if path.startswith('/'):
|
||
|
|
path = path[1:]
|
||
|
|
return os.path.join(self.output_path, path)
|
||
|
|
|
||
|
|
def exists(self, path):
|
||
|
|
try:
|
||
|
|
open(self.build_path(path)).close()
|
||
|
|
return True
|
||
|
|
except FileNotFoundError:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
class TestCase(unittest.TestCase):
|
||
|
|
client = TestClient()
|
||
|
|
|
||
|
|
def assertTitle(self, content, title):
|
||
|
|
self.assertIn(title, content.title.string)
|