from __future__ import absolute_import

import unittest

import silenttestrunner

from mercurial import (

    error,

    scmutil,

)

class mockfile(object):

    def __init__(self, name, fs):

        self.name = name

        self.fs = fs

    def __enter__(self):

        return self

    def __exit__(self, *args, **kwargs):

        pass

    def write(self, text):

        self.fs.contents[self.name] = text

    def read(self):

        return self.fs.contents[self.name]

class mockvfs(object):

    def __init__(self):

        self.contents = {}

    def read(self, path):

        return mockfile(path, self).read()

    def readlines(self, path):

    def __call__(self, path, mode, atomictemp):

        return mockfile(path, self)

class testsimplekeyvaluefile(unittest.TestCase):

    def setUp(self):

        self.vfs = mockvfs()

        self.assertEqual(sorted(self.vfs.read('kvfile').split('\n')),

                         ['', 'Key2=value2', 'key1=value1'])

    def testinvalidkeys(self):

        d = {'0key1': 'value1', 'Key2': 'value2'}

        self.assertRaises(error.ProgrammingError,

                          scmutil.simplekeyvaluefile(self.vfs, 'kvfile').write,

                          d)

        d = {'key1@': 'value1', 'Key2': 'value2'}

        self.assertRaises(error.ProgrammingError,

                          scmutil.simplekeyvaluefile(self.vfs, 'kvfile').write,

                          d)

    def testinvalidvalues(self):

        d = {'key1': 'value1', 'Key2': 'value2\n'}

        self.assertRaises(error.ProgrammingError,

                          scmutil.simplekeyvaluefile(self.vfs, 'kvfile').write,

                          d)

    def testcorruptedfile(self):

        self.vfs.contents['badfile'] = 'ababagalamaga\n'

        self.assertRaises(error.CorruptedState,

                          scmutil.simplekeyvaluefile(self.vfs, 'badfile').read)

if __name__ == "__main__":

    silenttestrunner.main(__name__)