#!/usr/bin/env python
#
# Mock FreeBSD zpool command
#

import os
import sys

TEST_ROOT = os.sep.join(os.path.realpath(__file__).split(os.sep)[:-5])
MOCK_DATA = os.path.join(TEST_ROOT, 'data', 'mock-output')

from argparse import ArgumentParser
parser = ArgumentParser()
dummy_args, args = parser.parse_known_args()

if args == [ 'list', '-Hp' ]:
    filename = os.path.join(MOCK_DATA, 'freebsd', 'zpool', 'pools')
elif args[:2] == [ 'iostat', '-v', ]:
    filename = os.path.join(MOCK_DATA, 'freebsd', 'zpool', 'iostat')
else:
    sys.exit(1)

try:
    sys.stdout.write(open(filename, 'r').read())
    sys.exit(0)
except Exception as e:
    print e
    sys.exit(1)
