# -*- coding: utf-8 -*-

import os

import errno

import stat

import tempfile

from unittest import TestCase

from mock import patch, mock_open, call, DEFAULT

from rattail import daemon

class DaemonTests(TestCase):

    pidfile = '/tmp/test_rattail_daemon.pid'

    def test_init(self):

        d = daemon.Daemon(self.pidfile)

        self.assertEqual(d.pidfile, self.pidfile)

        self.assertEqual(d.stdin, os.devnull)

        self.assertEqual(d.stdout, os.devnull)

        self.assertEqual(d.stderr, os.devnull)

    def test_delpid(self):

        # normal delete

        with open(self.pidfile, 'w') as f:

            f.write('foo')

        self.assertTrue(os.path.exists(self.pidfile))

        d = daemon.Daemon(self.pidfile)

        d.delpid()

        self.assertFalse(os.path.exists(self.pidfile))

        # file doesn't exist (no error)

        d.delpid()

    @patch('sys.exit')

    @patch('sys.stderr')

    def test_start(self, stderr, exit_):

        d = daemon.Daemon(self.pidfile)

        with patch.object(d, 'daemonize') as daemonize:

            with patch.object(d, 'run') as run:

                # pidfile exists

                with open(self.pidfile, 'w') as f:

                    f.write('42\n')

                self.assertTrue(os.path.exists(self.pidfile))

                exit_.side_effect = RuntimeError

                self.assertRaises(RuntimeError, d.start)

                stderr.write.assert_called_once_with("pidfile /tmp/test_rattail_daemon.pid already exist. Daemon already running?\n")

                exit_.assert_called_once_with(1)

                self.assertFalse(daemonize.called)

                self.assertFalse(run.called)

                stderr.reset_mock()

                exit_.reset_mock()

                # no pidfile, no daemonize

                os.remove(self.pidfile)

                self.assertFalse(os.path.exists(self.pidfile))

                d.start(daemonize=False)

                self.assertFalse(stderr.write.called)

                self.assertFalse(exit_.called)

                self.assertFalse(d.daemonize.called)

                d.run.assert_called_once_with()

                d.run.reset_mock()

                # no pidfile, with daemonize

                self.assertFalse(os.path.exists(self.pidfile))

                d.start(daemonize=True)

                self.assertFalse(stderr.write.called)

                self.assertFalse(exit_.called)

                d.daemonize.assert_called_once_with()

                d.run.assert_called_once_with()

@patch('os.chdir')

@patch('os.chmod')

@patch('os.dup2')

@patch('os.fork')

@patch('os.getpid')

@patch('os.setsid')

@patch('os.umask')

@patch('sys.exit')

@patch('sys.stderr')

@patch('sys.stdin')

@patch('sys.stdout')

@patch('atexit.register')

class DaemonDaemonizeTests(TestCase):

    pidfile = '/tmp/test_rattail_daemon.pid'

    def tearDown(self):

        if os.path.exists(self.pidfile):

            os.remove(self.pidfile)