モデルに登録するコードのテストコードを書きたい

Question

モデルに登録するコードのテストコードを書きたいです。

from app import UserData
from models.userregist import UserRegist

def user_regist():
    user_li = UserData.get_data()
    for i in range(len(user_li)):
        data = {
            'id': i,
            'age': user_li[i][0],
            'adress': user_li[i][1]
        }
        UserRegist(data).save()

というコードがあります。

user_li = UserData.get_data()

のuser_liには

[[25,"America"],[19,"UK"]]

のような配列が入っています。
今まで他のコードではnoseを使って、eq_メソッドなどで値が等しいか調べていましたが、モデルに正しく値が登録されているかをチェックするにはどうしたらいいのでしょうか？

Comments

[t_obara]

正しくを一番理解されているのは貴殿だと思うのですが、「正しい」を提示されないと、何とも回答できないかと思います。何をテストするのかをよくお考えになってみてはいかがでしょうか。

[k0908]

UserRegistモデルに意図したものが入っているか、という点をチェックしたいです

Answer 1

下記の前提で正しければ。

• UserDataとUserRegistは質問者さんが作成された
• user_regist()が想定通りに動くかを確かめたい

UserDataとUserRegistのmockを作って単体テストをされるのはどうでしょうか。
https://docs.python.org/ja/3/library/unittest.mock-examples.html