Skip to content
Snippets Groups Projects
Select Git revision
  • main default protected
1 result
Blame Permalink
test_generate.py 2.40 KiB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57 






"""
Check setup page at /generate URL.

MemeGenerator

Ruizhe Deng rxd@umich.edu
"""
import re
import bs4

def test_generate_empty(client):
    url = '/generate?top=&url=https%3A%2F%2Flive.staticflickr.com%2F3606%2F3369014165_7441c25d45_q.jpg&bot='
    response = client.get(url)
    soup = bs4.BeautifulSoup(response.data, "html.parser")
    assert response.status_code == 200

    src = [x.get("src") for x in soup.find_all('img')]
    # check if there is only one image link
    assert len(src) == 1
    # check if that link to the image is correct
    first_part = src[0].split('?')[0]
    first_part_reg = 'https:\/\/api\.memegen\.link\/images\/custom\/[a-zA-Z0-9_]*\/[a-zA-Z0-9_]*\.png'
    assert re.match(first_part_reg, first_part) is not None

def test_generate_one_empty(client):
    bot_url = '/generate?top=&url=https%3A%2F%2Flive.staticflickr.com%2F3606%2F3369014165_7441c25d45_q.jpg&bot=bot'
    top_url = '/generate?top=top&url=https%3A%2F%2Flive.staticflickr.com%2F3606%2F3369014165_7441c25d45_q.jpg&bot='
    # bot first
    response = client.get(bot_url)
    soup = bs4.BeautifulSoup(response.data, "html.parser")
    assert response.status_code == 200
    # check for link correctness
    src = [x.get("src") for x in soup.find_all('img')]
    first_part = src[0].split('?')[0]
    first_part_reg = 'https:\/\/api\.memegen\.link\/images\/custom\/[a-zA-Z0-9_]*\/[a-zA-Z0-9_]*\.png'
    assert re.match(first_part_reg, first_part) is not None

    # top check
    response = client.get(top_url)
    soup = bs4.BeautifulSoup(response.data, "html.parser")
    assert response.status_code == 200
    # check for link correctness
    src = [x.get("src") for x in soup.find_all('img')]
    first_part = src[0].split('?')[0]
    first_part_reg = 'https:\/\/api\.memegen\.link\/images\/custom\/[a-zA-Z0-9_]*\/[a-zA-Z0-9_]*\.png'
    assert re.match(first_part_reg, first_part) is not None

def check_generate_full(client):
    url = '/generate?top=a&url=https%3A%2F%2Flive.staticflickr.com%2F3606%2F3369014165_7441c25d45_q.jpg&bot=c'
    response = client.get(url)
    soup = bs4.BeautifulSoup(response.data, "html.parser")
    assert response.status_code == 200
    # check for link correctness
    src = [x.get("src") for x in soup.find_all('img')]
    first_part = src[0].split('?')[0]
    first_part_reg = 'https:\/\/api\.memegen\.link\/images\/custom\/[a-zA-Z0-9_]*\/[a-zA-Z0-9_]*\.png'
    assert re.match(first_part_reg, first_part) is not None