コード例 #1
0
ファイル: test_base.py プロジェクト: ErinCall/catsnap
    def test_get_table_creates_table_conection(self, Client):
        client = Mock()
        Client.return_value = client
        mock_table = Mock()
        client.table.return_value = mock_table

        document = Document()
        document._table_name = 'grouchy'
        table = document._table()
        eq_(document._stored_table, mock_table)
        eq_(table, mock_table)
        client.table.assert_called_with('grouchy')