def test_regex_copy(mock_server):
imposter = Imposter(
Stub(responses=Response(
status_code="${code}",
headers={"X-Test": "${header}"},
body="Hello, ${name}!",
copy=[
Copy("path", "${code}", UsingRegex("\\d+")),
Copy({"headers": "X-Request"}, "${header}", UsingRegex(".+")),
Copy({"query": "name"}, "${name}",
UsingRegex("AL\\w+", ignore_case=True)),
],
)))
with mock_server(imposter):
response = requests.get(imposter.url / str(456),
params={"name": "Alice"},
headers={"X-REQUEST": "Header value"})
assert_that(
response,
is_response().with_status_code(456).with_body(
"Hello, Alice!").with_headers(
has_entry("X-Test", "Header value")),
)
def test_jsonpath_copy(mock_server):
imposter = Imposter(
Stub(responses=Response(body="Have you read BOOK?",
copy=Copy("body", "BOOK",
UsingJsonpath("$..title")))))
with mock_server(imposter):
response = requests.post(
imposter.url,
json={
"books": [
{
"book": {
"title": "Game of Thrones",
"summary": "Dragons and political intrigue",
}
},
{
"book": {
"title": "Harry Potter",
"summary": "Dragons and a boy wizard"
}
},
{
"book": {
"title": "The Hobbit",
"summary": "A dragon and short people"
}
},
]
},
)
assert_that(response,
is_response().with_body("Have you read Game of Thrones?"))
def test_xpath_copy(mock_server):
imposter = Imposter(
Stub(responses=Response(body="Have you read BOOK?",
copy=Copy("body", "BOOK",
UsingXpath("(//title)[2]")))))
with mock_server(imposter):
response = requests.post(imposter.url, data=BOOKS_XML)
assert_that(response,
is_response().with_body("Have you read Harry Potter?"))
def test_from_structure():
# Given
using = UsingRegexBuilder(selector="selector").build()
expected_copy = CopyBuilder(from_="from", into="into", using=using)
structure = expected_copy.as_structure()
# When
actual = Copy.from_structure(structure)
# Then
assert actual.from_ == "from"
assert actual.into == "into"
assert actual.using.selector == "selector"
def test_xpath_copy_namespaced(mock_server):
imposter = Imposter(
Stub(responses=Response(
body="Have you read BOOK?",
copy=Copy(
"body",
"BOOK",
UsingXpath(
"//isbn:title",
ns={"isbn": "http://schemas.isbn.org/ns/1999/basic.dtd"}),
),
)))
with mock_server(imposter):
response = requests.post(imposter.url, data=BOOKS_XML_NAMESPACED)
assert_that(response,
is_response().with_body("Have you read Game of Thrones?"))