test-driven-api-JingfengPan/test_database.py at main · MPCS-51042/test-driven-api-JingfengPan · GitHub

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
from database import Database

def test_get():
    # GIVEN
    db = Database()
    db._data = {"Jinx": "Zaun"}

    # WHEN
    result = db.get("Jinx")

    # THEN
    assert result == "Zaun"

def test_put():
    # GIVEN
    db = Database()

    # WHEN
    db.put("Ekko", "Zaun")

    # THEN
    assert db._data["Ekko"] == "Zaun"

def test_all():
    # GIVEN
    db = Database()
    dict = {"Jinx": "Zaun", "Ekko": "Zaun"}
    db._data = dict
    # WHEN
    result = db.all()

    # THEN
    assert result == dict

def test_delete():
    # GIVEN
    db = Database()
    db._data = {"Jinx": "Zaun", "Ekko": "Zaun"}

    # WHEN
    db.delete("Jinx")

    # THEN
    assert "Jinx" not in db._data
    assert db._data == {"Ekko": "Zaun"}