test_relink.py

from extract import Extract
from relink import *
#from test_extract import ANCHOR

from pathlib import Path
from pytest import fixture

TEST_MHT='sample.mhtml'
TEST_ANCHOR='h4'

@fixture
def soup():
	ex = Extract(TEST_MHT)
	rel = relink(ex, TEST_ANCHOR)
	return ex.soup

def test_rel(soup):
	assert soup

def test_toc(soup):
	print(RID_TOC)
	toc = soup.find(id=RID_TOC)
	assert toc
	assert 'List' in str(toc)
	assert RID in toc['id']
	assert len(toc.contents) > 1
	olist = toc.contents[1]
	item = olist.contents[0]
	anchor = item.contents[0]
	assert anchor
	assert RID in anchor['href']

def untest_div(soup):
	sib = soup.find(class_=RID_PARENT)
	print(str(soup.body.div.div.div)[500:1000])
	assert sib
	assert sib['class']
	assert RID_PARENT in sib['class']

def untest_find_anchor(soup):
	#print(soup.body.div.div.prettify())
	# ONLY works if `sibling.insert_after(tadd(toc, links))`
	result = soup.findAll('h4')
	assert result
	assert len(result) == 40