import re

from lib import BaseTest


class ListSnapshot1Test(BaseTest):
    """
    list snapshots: regular list
    """
    fixtureDB = True
    fixtureCmds = [
        "aptly snapshot create snap1 from mirror wheezy-main",
        "aptly snapshot create snap2 from mirror wheezy-contrib",
        "aptly snapshot merge snap3 snap1 snap2",
        "aptly snapshot pull snap1 snap2 snap4 mame unrar",
        "aptly repo create local-repo",
        "aptly repo add local-repo ${files}",
        "aptly snapshot create snap5 from repo local-repo",
    ]
    runCmd = "aptly snapshot list"


class ListSnapshot2Test(BaseTest):
    """
    list snapshots: empty list
    """
    runCmd = "aptly snapshot list"


class ListSnapshot3Test(BaseTest):
    """
    list snapshots: raw regular list
    """
    fixtureDB = True
    fixtureCmds = [
        "aptly snapshot create snap1 from mirror wheezy-main",
        "aptly snapshot create snap2 from mirror wheezy-contrib",
        "aptly snapshot merge snap3 snap1 snap2",
        "aptly snapshot pull snap1 snap2 snap4 mame unrar",
        "aptly repo create local-repo",
        "aptly repo add local-repo ${files}",
        "aptly snapshot create snap5 from repo local-repo",
    ]
    runCmd = "aptly -raw snapshot list"


class ListSnapshot4Test(BaseTest):
    """
    list snapshots: raw empty list
    """
    runCmd = "aptly snapshot -raw list"


class ListSnapshot5Test(BaseTest):
    """
    list snapshots: raw regular list sorted by time
    """
    fixtureDB = True
    fixtureCmds = [
        "aptly snapshot create snap2 from mirror wheezy-main",
        "aptly snapshot create snap1 from mirror wheezy-contrib",
        "aptly snapshot merge snap3 snap1 snap2",
        "aptly snapshot pull snap1 snap2 snap4 mame unrar",
        "aptly repo create local-repo",
        "aptly repo add local-repo ${files}",
        "aptly snapshot create snap5 from repo local-repo",
    ]
    runCmd = "aptly -raw -sort=time snapshot list"


class ListSnapshot6Test(BaseTest):
    """
    list snapshots: regular list sorted by time
    """
    fixtureDB = True
    fixtureCmds = [
        "aptly snapshot create snap2 from mirror wheezy-contrib",
        "aptly snapshot create snap1 from mirror wheezy-main",
        "aptly snapshot merge snap3 snap1 snap2",
        "aptly snapshot pull snap1 snap2 snap4 mame unrar",
        "aptly repo create local-repo",
        "aptly repo add local-repo ${files}",
        "aptly snapshot create snap5 from repo local-repo",
    ]
    runCmd = "aptly -sort=time snapshot list"


class ListSnapshot7Test(BaseTest):
    """
    list snapshots: wrong parameter sort
    """
    fixtureCmds = [
        "aptly snapshot create empty empty"
    ]
    runCmd = "aptly -sort=planet snapshot list"
    expectedCode = 1


class ListSnapshot8Test(BaseTest):
    """
    list snapshots: json regular list
    """
    fixtureDB = True
    fixtureCmds = [
        "aptly snapshot create snap1 from mirror wheezy-main",
        "aptly snapshot create snap2 from mirror wheezy-contrib",
        "aptly snapshot merge snap3 snap1 snap2",
        "aptly snapshot pull snap1 snap2 snap4 mame unrar",
        "aptly repo create local-repo",
        "aptly repo add local-repo ${files}",
        "aptly snapshot create snap5 from repo local-repo",
    ]
    runCmd = "aptly -json snapshot list"

    def outputMatchPrepare(self, s):
        return re.sub(r'[ ]*"CreatedAt": "[^"]+",?\n', '', s)


class ListSnapshot9Test(BaseTest):
    """
    list snapshots: json empty list
    """
    runCmd = "aptly snapshot -json list"


class ListSnapshot10Test(BaseTest):
    """
    list snapshots: json regular list sorted by time
    """
    fixtureDB = True
    fixtureCmds = [
        "aptly snapshot create snap2 from mirror wheezy-main",
        "aptly snapshot create snap1 from mirror wheezy-contrib",
        "aptly snapshot merge snap3 snap1 snap2",
        "aptly snapshot pull snap1 snap2 snap4 mame unrar",
        "aptly repo create local-repo",
        "aptly repo add local-repo ${files}",
        "aptly snapshot create snap5 from repo local-repo",
    ]
    runCmd = "aptly -json -sort=time snapshot list"

    def outputMatchPrepare(self, s):
        return re.sub(r'[ ]*"CreatedAt": "[^"]+",?\n', '', s)