2019-03-11

pytest-mock を使ってみた #python

Python

概要

Python のテストを Pytest で書いてみた
Mock を使おうと思ったら pytest-mock という標準の薄いラッパーを使うのがいいらしい
そもそも標準を知らないので少し苦労した
なのでログとして残しておく

テスト対象

HTTPリクエストして、レスポンスを返すような処理を書いてみる。

レスポンスをモックしてテストしてみる。

import requests

def sample():
    try:
        # ここを Mock したい
        res = requests.get("http://checkip.amazonaws.com/")

    except requests.RequestException as e:
        print(e)
        raise e

    return {
        "statusCode": res.status_code,
        "ip": res.text,
    }

テストコード

本来なら 200 のレスポンスが返ってくるはずだが、 404 にしてみる

from hello_mock import app

def test_sample(mocker):
    # 嘘のレスポンスを作成
    responseMock = mocker.Mock()
    responseMock.status_code = 404
    responseMock.text = '127.0.0.1'

    # requests.get の戻り値を patch する
    mocker.patch('requests.get').return_value = responseMock

    actual = app.sample()
    assert actual['statusCode'] == 404
    assert actual['ip'] == '127.0.0.1'

補足

pytest-mock を使うと、 mocker 経由で Mock を扱えるので便利。

side-effect とかを使えば例外のテストなんかもできそう。

参考

2019-02-17

DockerHub から image のタグ一覧を取得する

shell

Docker Hub の image から tag を探そうと思ったときに、数が多いと中々目当てのものが見つからなかったりする。

API で取得するスクリプトを組んでる人がいたが、API のバージョンが古かったり、タグがたくさんあるとページングされたりするので、自分で書き直してみる。

#!/usr/bin/env bash

function docker-taglist() {
  local image=${1}

  local response=$(curl --silent --show-error "https://registry.hub.docker.com/v2/repositories/${image}/tags/")
  local names=$(echo ${response} | jq -r .results[].name)

  echo -e "${names}"

  # 最大 5 ページまで取得する
  local page=1
  local next=$(echo ${response} | jq -r .next)
  while [[ ${page} -lt 5 && ${next} != "null" ]]
  do
    response=$(curl --silent --show-error "${next}")
    names="${names}\n$(echo ${response} | jq -r .results[].name)"

    echo -e "${names}"

    page=$((${page}+1))
    next=$(echo ${response} | jq -r .next)
  done
}

エイリアスに登録しておけば、こんな感じで使える。

$ docker-taglist library/mysql
5.5
5.5.62
5.6
5.6.43
...

ページ数は引数で受け取れるようにするといいかもしれない。

参考

DockerHubのイメージのタグ一覧をコマンドで取得する | Mazn.net

2019-02-11

「サーバ/インフラを支える技術」を読んだ

読書メモ

[24時間365日] サーバ/インフラを支える技術 ?スケーラビリティ、ハイパフォーマンス、省力運用 (WEB+DB PRESS plusシリーズ)

作者: 安井真伸,横川和哉,ひろせまさあき,伊藤直也,田中慎司,勝見祐己
出版社/メーカー: 技術評論社
発売日: 2008/08/07
メディア: 単行本（ソフトカバー）
購入: 133人クリック: 2,270回
この商品を含むブログ (288件) を見る

10年以上前の本なのだが、古本屋でたまたま手にとって、よさそうだなと思って読んでみた。

冗長化とか構成管理とか、今でも通じる話の基本的な考え方が書いてあって、今読んでもいい本だと思う。

ツールは多分今だと違うやつだろうなとか、これはオンプレ時代だと大切だったんだろうな、みたいな話は流し読み。

個人的に一番よかったのは個別のインスタンスの負荷をどう見ていくかというところ。

ロードアベレージ見て、CPU負荷かI/O負荷か切り分けて、そしてそれがどういう仕組みなのか、みたいなところがとてもわかり易かった。

この辺の考え方はどこかで役に立ちそう。

コンテナベースの世界になるとまた違ってきそうだけど。

前よりはマシになっている気はするが、どうしてもインフラの苦手意識は消えない。

モダンな話は入門監視 ―モダンなモニタリングのためのデザインパターンあたりを読むと書いてあるのかなぁ。

それにしても10年前の本だと思うと、はてなってやっぱりすごい会社だったんだなと感じる。

2019-01-28

Go で JSON を扱うユーティリティを作ってみた #golang

golang

概要

Go で JSON を扱うときは構造体を用意して扱うのが一般的と思われる。

ただ外部の API 経由で取得する JSON なんかだと、フィールドが大量だったり、階層が深かったりして扱うのが大変だったりする。

一部の値だけ使いたいときにもっと手軽に扱えないかなー、と思ってユーティリティを作ってみた。

github.com

使い方

README とか Example も書いたんだけどこんな感じ。

jsonVal := `
  {
      "team": "FC Barcelona",
      "captain": {
          "name":"Messi", 
          "position":"Forward"
      }
  }
  `
// JSON 文字列からコンテナを生成
container, err := chazuke.FromJSON(jsonVal)
if err != nil {
    panic(err.Error())
}

// Get で ほしい情報にアクセスして Value で取り出す
team, err := container.Get("team").Value()
if err != nil {
    panic(err.Error())
}
fmt.Println(team) // -> FC Barcelona

// 階層化してる場合はこんな感じ
captain, err := container.Get("captain").Get("name").Value()
if err != nil {
    panic(err.Error())
}
fmt.Println(captain) // -> Messi