【Python】mockの中身をテストするmock.assert | Udemyのセールを活用してほしかったあのスキルを手に入れよう！【非公式】

Python学習【365日チャレンジ！】196日目のマスターU（@Udemy11）です。

知り合いの方が「上司が馬鹿すぎてやってられない」なんてことを言ってたので、ちょっと愚痴を聴いてあげました。

ドラマ半沢直樹でもありましたが、組織では稟議を回して上役の承認をもらいますが、その稟議書に難癖をつけてなかなか承認しない上司がいるらしいです。

承認しない理由というのが、稟議書の内容がきちんとしていないからではなく、文章の書き方が気に入らないとか書式がおかしいというどうでもいいような話で修正しろと言ってくるらしいです。

例えば、「この案について〜中略〜お伺いします。」という文章があったとすると、「この案について〜中略〜伺います。」としなければならないだとか、案が2つあったときは、「2案」ではなく、「案の2」にしなければならないだとか、ほんとど〜でもいいようなところに注文をつけてくるらしいです。

基本、稟議書ってどのようなことをどのようにするかがわかればいいので、多少の物言いは問題ないとは思いますが、その人の中では、答えが1つしかないんでしょうね。

日本教育で育った典型的な上司のようですね。

このようなことは日本中で起こっているんでしょうけど、会社員ならうまくストレスにならない方法を身に着けておいたほうが良さそうです。

それでは、今日もPython学習をはじめましょう。

昨日の復習

昨日は外部サービスの処理をして取得する値を仮に代入できるmockを学習しました。

外部サービスがまだできていないけど、どんな返り値が返ってくるかわかっている場合は、外部の処理を待つまでもなく、mockを使えば想定値を入れてテストをすることができました。

詳細については、こちらの記事をごらんください。

Udemyのセールを活用してほしかったあのスキルを手に入れよう！【非公式】

2020.08.24

【Python】外部サービスから取得する値を擬似的に取得可能なmock

https://udemyfun.com/python-unittest-mock

Python学習【365日チャレンジ！】195日目のマスターU（@Udemy11）です。近所の方からスイカを一玉いただきました。普段なら、普通に切って、皮が付いたままかぶりつくのですが、今回は趣向を変えて、フルーツミックスとしてスーパーで売っているような感じで一口サイズにカットしていただきました。思った以上に食べやすくて、甘かったので、ついつい食べすぎて、お腹がパンパンになってしまいました。スイカはほぼ水分なので、体重が増えてしまいましたが、明日になればもとに戻っているような気がします。ほんとこの時期のスイカって...

今日は、mockの中で使うmock.assertを学習します。

mock.assert

昨日は、mockを使って外部サービスで取得する値を擬似的に取得しました。

昨日のテストだけだと、擬似的に値を代入しているだけで、メインのコードで外部サービスの値を取得する関数がきちんと呼ばれているかが確認できていないので、関数が呼ばれているかどうかを確認するためのものがmock.assertです。

mock.assertを使うことで、きちんと関数が呼ばれたか、1回だけ呼ばれているか、引数を他の関数に渡せているかなどを確認することができます。

これらを確認するためのコードを確認します。

salary.py

メインのファイルは昨日のsalary.pyと同じです。

import requests

class BonusApi(object):
    def bonus_price(self, year):
        req = requests.get('http://localhost/bonus', params={'year': year})
        return req.json()['price']
    
class Salary(object):
    def __init__(self, base=50, year=2020):
        self.bonus_api = BonusApi()
        self.base = base
        self.year = year
        
    def cal_salary(self):
        bonus = self.bonus_api.bonus_price(year=self.year)
        return self.base + bonus

test_salary.py

続いてテストファイル（test_salary.py）に、何行かコードを追加します。

import unittest

from unittest.mock import MagicMock

import salary


class TestSalary(unittest.TestCase):
    def test_cal_salary(self):
        s = salary.Salary()
        s.bonus_api.bonus_price = MagicMock(return_value=100)
        self.assertEqual(s.cal_salary(), 150)
        s.bonus_api.bonus_price.assert_called()
        s.bonus_api.bonus_price.assert_called_once()
        s.bonus_api.bonus_price.assert_called_with(year=2020)
        s.bonus_api.bonus_price.assert_called_once_with(year=2020)
        self.assertEqual(s.bonus_api.bonus_price.call_count, 1)

ハイライトしている13行目から17行目が付け加えたコードです。

それぞれの次のテストをおこないます。