#!/usr/bin/env python

"""
unit tests for AtomicWrite
"""

import atomic
import os
import tempfile
import unittest

class AtomicWriteUnitTest(unittest.TestCase):

    def test_atomicwrite(self):
        tf = tempfile.mktemp()
        self.assertFalse(os.path.exists(tf))
        atomic.write('foo', tf)
        self.assertTrue(os.path.exists(tf))
        self.assertEqual(open(tf, 'r').read(), 'foo')

if __name__ == '__main__':
    unittest.main()