exercism/python
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

rest-api: add test template (#1904)

Browse Source
This commit is contained in:
Corey McCandless
2019-08-16 11:44:51 -04:00
committed by GitHub
parent 2ae116500b
commit 6135b7c005
2 changed files with 95 additions and 288 deletions
Download Patch File Download Diff File Expand all files Collapse all files

359
exercises/rest-api/rest_api_test.py
View File

@@ -1,224 +1,105 @@
import unittest
import json
from rest_api import RestAPI
# Tests adapted from `problem-specifications//canonical-data.json` @ v1.1.1
import json
class RestAPITest(unittest.TestCase):
class RestApiTest(unittest.TestCase):
def test_no_users(self):
database = {"users": []}
api = RestAPI(database)
response = api.get('/users')
self.assertDictEqual(json.loads(response), database)
response = api.get("/users")
expected = {"users": []}
self.assertDictEqual(json.loads(response), expected)
def test_add_user(self):
database = {"users": []}
api = RestAPI(database)
payload = json.dumps({
'user': 'Adam'
})
response = api.post('/add', payload)
expected = {
'name': 'Adam',
'owes': {},
'owed_by': {},
'balance': 0
}
payload = json.dumps({"user": "Adam"})
response = api.post("/add", payload)
expected = {"name": "Adam", "owes": {}, "owed_by": {}, "balance": 0.0}
self.assertDictEqual(json.loads(response), expected)
def test_get_single_user(self):
database = {
'users': [
{
'name': 'Adam',
'owes': {},
'owed_by': {},
'balance': 0
},
{
'name': 'Bob',
'owes': {},
'owed_by': {},
'balance': 0
}
"users": [
{"name": "Adam", "owes": {}, "owed_by": {}, "balance": 0.0},
{"name": "Bob", "owes": {}, "owed_by": {}, "balance": 0.0},
]
}
api = RestAPI(database)
payload = json.dumps({
'users': ['Bob']
})
response = api.get('/users', payload)
payload = json.dumps({"users": ["Bob"]})
response = api.get("/users", payload)
expected = {
'users': [
{
'name': 'Bob',
'owes': {},
'owed_by': {},
'balance': 0
}
]
"users": [{"name": "Bob", "owes": {}, "owed_by": {}, "balance": 0.0}]
}
self.assertDictEqual(json.loads(response), expected)
def test_iou_both_users_have_0_balance(self):
def test_both_users_have_0_balance(self):
database = {
'users': [
{
'name': 'Adam',
'owes': {},
'owed_by': {},
'balance': 0
},
{
'name': 'Bob',
'owes': {},
'owed_by': {},
'balance': 0
}
"users": [
{"name": "Adam", "owes": {}, "owed_by": {}, "balance": 0.0},
{"name": "Bob", "owes": {}, "owed_by": {}, "balance": 0.0},
]
}
api = RestAPI(database)
payload = json.dumps({
'lender': 'Adam',
'borrower': 'Bob',
'amount': 3
})
response = api.post('/iou', payload)
payload = json.dumps({"lender": "Adam", "borrower": "Bob", "amount": 3.0})
response = api.post("/iou", payload)
expected = {
'users': [
{
'name': 'Adam',
'owes': {},
'owed_by': {
'Bob': 3
},
'balance': 3
},
{
'name': 'Bob',
'owes': {
'Adam': 3
},
'owed_by': {},
'balance': -3
}
"users": [
{"name": "Adam", "owes": {}, "owed_by": {"Bob": 3.0}, "balance": 3.0},
{"name": "Bob", "owes": {"Adam": 3.0}, "owed_by": {}, "balance": -3.0},
]
}
self.assertDictEqual(json.loads(response), expected)
def test_borrower_has_negative_balance(self):
database = {
'users': [
{
'name': 'Adam',
'owes': {},
'owed_by': {},
'balance': 0
},
{
'name': 'Bob',
'owes': {
'Chuck': 3
},
'owed_by': {},
'balance': -3
},
{
'name': 'Chuck',
'owes': {},
'owed_by': {
'Bob': 3
},
'balance': 3
}
"users": [
{"name": "Adam", "owes": {}, "owed_by": {}, "balance": 0.0},
{"name": "Bob", "owes": {"Chuck": 3.0}, "owed_by": {}, "balance": -3.0},
{"name": "Chuck", "owes": {}, "owed_by": {"Bob": 3.0}, "balance": 3.0},
]
}
api = RestAPI(database)
payload = json.dumps({
'lender': 'Adam',
'borrower': 'Bob',
'amount': 3
})
response = api.post('/iou', payload)
payload = json.dumps({"lender": "Adam", "borrower": "Bob", "amount": 3.0})
response = api.post("/iou", payload)
expected = {
'users': [
"users": [
{"name": "Adam", "owes": {}, "owed_by": {"Bob": 3.0}, "balance": 3.0},
{
'name': 'Adam',
'owes': {},
'owed_by': {
'Bob': 3
},
'balance': 3
"name": "Bob",
"owes": {"Adam": 3.0, "Chuck": 3.0},
"owed_by": {},
"balance": -6.0,
},
{
'name': 'Bob',
'owes': {
'Adam': 3,
'Chuck': 3
},
'owed_by': {},
'balance': -6
}
]
}
self.assertDictEqual(json.loads(response), expected)
def test_lender_has_negative_balance(self):
database = {
'users': [
{
'name': 'Adam',
'owes': {},
'owed_by': {},
'balance': 0
},
{
'name': 'Bob',
'owes': {
'Chuck': 3
},
'owed_by': {},
'balance': -3
},
{
'name': 'Chuck',
'owes': {},
'owed_by': {
'Bob': 3
},
'balance': 3
}
"users": [
{"name": "Adam", "owes": {}, "owed_by": {}, "balance": 0.0},
{"name": "Bob", "owes": {"Chuck": 3.0}, "owed_by": {}, "balance": -3.0},
{"name": "Chuck", "owes": {}, "owed_by": {"Bob": 3.0}, "balance": 3.0},
]
}
api = RestAPI(database)
payload = json.dumps({
'lender': 'Bob',
'borrower': 'Adam',
'amount': 3
})
response = api.post('/iou', payload)
payload = json.dumps({"lender": "Bob", "borrower": "Adam", "amount": 3.0})
response = api.post("/iou", payload)
expected = {
'users': [
"users": [
{"name": "Adam", "owes": {"Bob": 3.0}, "owed_by": {}, "balance": -3.0},
{
'name': 'Adam',
'owes': {
'Bob': 3
},
'owed_by': {},
'balance': -3
"name": "Bob",
"owes": {"Chuck": 3.0},
"owed_by": {"Adam": 3.0},
"balance": 0.0,
},
{
'name': 'Bob',
'owes': {
'Chuck': 3
},
'owed_by': {
'Adam': 3
},
'balance': 0
}
]
}
self.assertDictEqual(json.loads(response), expected)
@@ -226,49 +107,17 @@ class RestAPITest(unittest.TestCase):
def test_lender_owes_borrower(self):
database = {
"users": [
{
"name": "Adam",
"owes": {
"Bob": 3.0
},
"owed_by": {},
"balance": -3.0
},
{
"name": "Bob",
"owes": {},
"owed_by": {
"Adam": 3.0
},
"balance": 3.0
}
{"name": "Adam", "owes": {"Bob": 3.0}, "owed_by": {}, "balance": -3.0},
{"name": "Bob", "owes": {}, "owed_by": {"Adam": 3.0}, "balance": 3.0},
]
}
api = RestAPI(database)
payload = json.dumps({
'lender': 'Adam',
'borrower': 'Bob',
'amount': 2
})
response = api.post('/iou', payload)
payload = json.dumps({"lender": "Adam", "borrower": "Bob", "amount": 2.0})
response = api.post("/iou", payload)
expected = {
'users': [
{
"name": "Adam",
"owes": {
"Bob": 1.0
},
"owed_by": {},
"balance": -1.0
},
{
"name": "Bob",
"owes": {},
"owed_by": {
"Adam": 1.0
},
"balance": 1.0
}
"users": [
{"name": "Adam", "owes": {"Bob": 1.0}, "owed_by": {}, "balance": -1.0},
{"name": "Bob", "owes": {}, "owed_by": {"Adam": 1.0}, "balance": 1.0},
]
}
self.assertDictEqual(json.loads(response), expected)
@@ -276,51 +125,17 @@ class RestAPITest(unittest.TestCase):
def test_lender_owes_borrower_less_than_new_loan(self):
database = {
"users": [
{
"name": "Adam",
"owes": {
"Bob": 3.0
},
"owed_by": {},
"balance": -3.0
},
{
"name": "Bob",
"owes": {},
"owed_by": {
"Adam": 3.0
},
"balance": 3.0
}
{"name": "Adam", "owes": {"Bob": 3.0}, "owed_by": {}, "balance": -3.0},
{"name": "Bob", "owes": {}, "owed_by": {"Adam": 3.0}, "balance": 3.0},
]
}
api = RestAPI(database)
payload = json.dumps({
'lender': 'Adam',
'borrower': 'Bob',
'amount': 4.0
})
response = api.post('/iou', payload)
payload = json.dumps({"lender": "Adam", "borrower": "Bob", "amount": 4.0})
response = api.post("/iou", payload)
expected = {
'users': [
{
"name": "Adam",
"owes": {
},
"owed_by": {
"Bob": 1.0
},
"balance": 1.0
},
{
"name": "Bob",
"owes": {
"Adam": 1.0
},
"owed_by": {
},
"balance": -1.0
}
"users": [
{"name": "Adam", "owes": {}, "owed_by": {"Bob": 1.0}, "balance": 1.0},
{"name": "Bob", "owes": {"Adam": 1.0}, "owed_by": {}, "balance": -1.0},
]
}
self.assertDictEqual(json.loads(response), expected)
@@ -328,53 +143,21 @@ class RestAPITest(unittest.TestCase):
def test_lender_owes_borrower_same_as_new_loan(self):
database = {
"users": [
{
"name": "Adam",
"owes": {
"Bob": 3.0
},
"owed_by": {},
"balance": -3.0
},
{
"name": "Bob",
"owes": {},
"owed_by": {
"Adam": 3.0
},
"balance": 3.0
}
{"name": "Adam", "owes": {"Bob": 3.0}, "owed_by": {}, "balance": -3.0},
{"name": "Bob", "owes": {}, "owed_by": {"Adam": 3.0}, "balance": 3.0},
]
}
api = RestAPI(database)
payload = json.dumps({
'lender': 'Adam',
'borrower': 'Bob',
'amount': 3.0
})
response = api.post('/iou', payload)
payload = json.dumps({"lender": "Adam", "borrower": "Bob", "amount": 3.0})
response = api.post("/iou", payload)
expected = {
'users': [
{
"name": "Adam",
"owes": {
},
"owed_by": {
},
"balance": 0.0
},
{
"name": "Bob",
"owes": {
},
"owed_by": {
},
"balance": 0.0
}
"users": [
{"name": "Adam", "owes": {}, "owed_by": {}, "balance": 0.0},
{"name": "Bob", "owes": {}, "owed_by": {}, "balance": 0.0},
]
}
self.assertDictEqual(json.loads(response), expected)
if __name__ == '__main__':
if __name__ == "__main__":
unittest.main()