python-p3-freebie-tracker/lib/test/testModels.py at main · earlearl98/python-p3-freebie-tracker · 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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

from conftest import SQLITE_URL
from models import Company, Freebie, Dev


# engine = create_engine(SQLITE_URL)
# Session = sessionmaker(bind=engine)
# session = Session()

class TestFreebie:
  def test_returns_dev_instance(self):


    engine = create_engine(SQLITE_URL)
    Session = sessionmaker(bind=engine)
    session = Session()

    freebie = Freebie(
      item_name="earl",
      value=150,
      company_id=3,
      dev_id=4,

    )


    session.add(freebie)
    session.commit()

    assert isinstance(freebie.dev, Dev)
    assert isinstance(freebie.company, Company)
    session.query(Freebie).delete()
    session.commit()


class TestCompany:
  def test_returns_dev_instance(self):


    engine = create_engine(SQLITE_URL)
    Session = sessionmaker(bind=engine)
    session = Session()

    freebie = Freebie(
      item_name="earl",
      value=150,
      company_id=3,
      dev_id=4,

    )

    session.add(freebie)
    session.commit()

    assert isinstance(freebie.dev, Dev)
    session.query(Freebie).delete()
    session.commit()