test_model.py

import unittest
from unittest.mock import patch

import model as m
from Transaction import Transaction


def test_current_user():
    with patch("model.sqlite3") as mock_sql:
        mock_sql.connect().cursor().fetchone.return_value = ["paulina"]

        assert m.current_user() == "paulina"


class TestLogIn:
    def test_log_in_success(self):
        with patch("model.sqlite3") as mock_sql:
            mock_sql.connect().cursor().fetchone.return_value = (
                "307e1fb4b8594b49b8eb119a4a38cc5020fd9eb18afa9a38b8c75abb4ac8ae6e",
                "86f712b2c0e419af5f9cfc53f0bd9f0b3cb0c81e4d9299125f2e9e99e504f3a7f"
                "2b534894ffbdca10ce0a5507142c91a4d66f859f6df5771ba04e5fa477f28e0",
            )
            assert m.log_in("asdf", "asdf")
            mock_sql.connect().cursor().fetchone.return_value = ["asdf"]
            assert m.current_user() == "asdf"

    def test_log_in_failure(self):
        with patch("model.sqlite3") as mock_sql:
            mock_sql.connect().cursor().fetchone.return_value = ("asdf", "asdf")
            assert not m.log_in("asdf", "asdf")


class TestCreate:
    def test_create_success(self):
        with patch("model.sqlite3"):
            assert m.create("asdf", "asdf", 124532523525)

    def test_create_fail_no_username(self):
        with patch("model.sqlite3"):
            assert not m.create("", "asdf", 124532523525)

    def test_create_fail_no_password(self):
        with patch("model.sqlite3"):
            assert not m.create("asdf", "", 124532523525)

    def test_create_fail_negative_value(self):
        with patch("model.sqlite3"):
            assert not m.create("asdf", "asdf", -124532523525)


def test_update_holdings():
    with patch("model.sqlite3"):
        m.update_holdings()


def test_calculate_transaction_revenue():
    assert m.calculate_transaction_revenue(1, 50, 7) == 43


def test_calculate_transaction_cost():
    assert m.calculate_transaction_cost(1, 50, 7) == 57


class TestLookupTickerSymbol(unittest.TestCase):
    def test_lookup_ticker_symbol_success(self):
        """Make the external API call to test status of API key"""
        assert m.lookup_ticker_symbol("Apple") == "AAPL"

    def test_lookup_ticker_symbol_fail(self):
        assert not m.lookup_ticker_symbol("asdf")


class TestQuoteLastPrice(unittest.TestCase):
    def test_quote_last_price_success(self):
        assert m.quote_last_price("AAPL")


def test_transaction_class():
    t1 = Transaction(
        last_price=124.5,
        brokerage_fee=10.50,
        current_balance=100000,
        trade_volume=5.0,
        new_balance=m.calculate_transaction_cost(5, 124.5, 10.50),
        ticker_symbol="AAPL",
        current_number_shares=500.0,
    )

    t2 = Transaction(
        last_price=124.5,
        brokerage_fee=10.50,
        current_balance=100000,
        trade_volume=5.0,
        new_balance=m.calculate_transaction_cost(5, 124.5, 10.50),
        ticker_symbol="AAPL",
        current_number_shares=500.0,
    )

    assert t1 == t2


def test_display_user_transactions():
    with patch("model.sqlite3") as mock_sql:
        mock_sql.connect().cursor().fetchall.return_value = [
            "AAPL",
            1.0,
            282.9,
            "2020-04-25 10:38 PM",
        ]

        assert m.display_user_transactions() == [
            "AAPL",
            1.0,
            282.9,
            "2020-04-25 10:38 PM",
        ]


def test_log_out():
    with patch("model.sqlite3"):
        m.log_out()