import zfs_autobackup.compressors
from basetest import *
import time

class TestSendRecvPipes(unittest2.TestCase):
    """test input/output pipes for zfs send and recv"""

    def setUp(self):
        prepare_zpools()
        self.longMessage=True



    def test_send_basics(self):
        """send basics (remote/local send pipe)"""


        with self.subTest("local local pipe"):
            with patch('time.strftime', return_value="20101111000000"):
                self.assertFalse(ZfsAutobackup(["test", "test_target1", "--exclude-received", "--no-holds", "--no-progress", "--send-pipe=dd bs=1M",  "--recv-pipe=dd bs=2M"]).run())

            shelltest("zfs destroy -r test_target1/test_source1/fs1/sub")

        with self.subTest("remote local pipe"):
            with patch('time.strftime', return_value="20101111000000"):
                self.assertFalse(ZfsAutobackup(["test", "test_target1", "--exclude-received", "--no-holds", "--no-progress", "--ssh-source=localhost", "--send-pipe=dd bs=1M",  "--recv-pipe=dd bs=2M"]).run())

            shelltest("zfs destroy -r test_target1/test_source1/fs1/sub")

        with self.subTest("local remote pipe"):
            with patch('time.strftime', return_value="20101111000000"):
                self.assertFalse(ZfsAutobackup(["test", "test_target1",  "--exclude-received", "--no-holds", "--no-progress", "--ssh-target=localhost", "--send-pipe=dd bs=1M",  "--recv-pipe=dd bs=2M"]).run())

            shelltest("zfs destroy -r test_target1/test_source1/fs1/sub")

        with self.subTest("remote remote pipe"):
            with patch('time.strftime', return_value="20101111000000"):
                self.assertFalse(ZfsAutobackup(["test", "test_target1",  "--exclude-received", "--no-holds", "--no-progress", "--ssh-source=localhost", "--ssh-target=localhost", "--send-pipe=dd bs=1M",  "--recv-pipe=dd bs=2M"]).run())

    def test_compress(self):
        """send basics (remote/local send pipe)"""

        for compress in zfs_autobackup.compressors.COMPRESS_CMDS.keys():

            with self.subTest("compress "+compress):
                with patch('time.strftime', return_value="20101111000000"):
                    self.assertFalse(ZfsAutobackup(["test", "test_target1", "--exclude-received", "--no-holds", "--no-progress", "--compress="+compress]).run())

                shelltest("zfs destroy -r test_target1/test_source1/fs1/sub")