tests/test_group_assign.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import datetime as dt

import pytest

from TSGroupAssigner import GroupAssigner, DateException

# sample input
creds = {
    'host': 'localhost',
    'port': 10011,
    'user': 'serveradmin',
    'password': '5up3r_53cr37',
    'sid': 1,
    'gid': 24
}


class TestGroupAssigner:
    def test_missing_input(self):
        # the main class is missing arguments and should fail with a TypeError

        with pytest.raises(TypeError):
            GroupAssigner().start()

    def test_sleepstart_startdate(self):
        # startdate is too far in the future sleepstart should produce a DateException

        # start date 3 days in the future
        startdate = dt.date.today() + dt.timedelta(days=3)
        duration = dt.timedelta()

        with pytest.raises(DateException):
            GroupAssigner(date=startdate, delta=duration, **creds).start()

    def test_datecheck_enddate(self):
        # this should produce a exit code 0 SystemExit as the end date is in the past

        # start date 2 days in the past
        startdate = dt.date.today() + dt.timedelta(days=-2)
        duration = dt.timedelta()

        with pytest.raises(SystemExit):
            GroupAssigner(date=startdate, delta=duration, **creds).start()

    def test_connect_noconnection(self):
        # connect should fail with ConnectionRefusedError

        # start date is today
        startdate = dt.date.today()
        duration = dt.timedelta()

        with pytest.raises(ConnectionRefusedError):
            GroupAssigner(date=startdate, delta=duration, **creds).start()