ActorUtilsTest

Complexity

Total Complexity

4

Size/Duplication

Total Lines	15
Duplicated Lines	0 %

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	tearDown()	0	2	1
A	test_actor_uri_na()	0	3	1
A	setUp()	0	3	1
A	test_actor_uri()	0	3	1

# -*- coding: utf-8 -*-
import unittest
import os
from datetime import date
from shutil import copyfile

from oe_utils.utils.actor_utils import actor_uri
from oe_utils.utils.file_utils import (
    get_file_size,
    get_last_modified_date,
    convert_size
)


class ActorUtilsTest(unittest.TestCase):
    def setUp(self):
        self.user = {"userid": "jalapjo", "actor": {"uri": "https://actor.test.org.test/a/123", "id": 123,
                                                    "omschrijving": "The Jalapeno, Jose"}}

    def tearDown(self):
        pass

    def test_actor_uri(self):
        uri = actor_uri(self.user)
        self.assertEqual("https://actor.test.org.test/a/123", uri)

    def test_actor_uri_na(self):
        uri = actor_uri({})
        self.assertEqual("NA", uri)


class FileUtilsTest(unittest.TestCase):
    def setUp(self):
        self.fixtures_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'fixtures')
        self.test_file = os.path.join(self.fixtures_path, 'Test_document.pdf')
        self.non_existing_file = os.path.join(self.fixtures_path, 'I_do_not_exist.pdf')

    def tearDown(self):
        pass

    def test_get_file_size(self):
        filesize = get_file_size(self.test_file)
        self.assertEqual('7.94 KB', filesize)

    def test_get_file_size_non_existing_file(self):
        self.assertIsNone(get_file_size(self.non_existing_file))

    def test_get_last_modified_date(self):
        today = date.today().strftime('%d/%m/%Y')
        # create new file to guarantee the last modified date is today
        copyfile(self.test_file,
                 os.path.join(self.fixtures_path, 'test_utils_temp_document.pdf'))
        self.new_file = os.path.join(self.fixtures_path, 'test_utils_temp_document.pdf')
        lm_date = get_last_modified_date(self.new_file)
        self.assertEqual(today, lm_date)
        os.remove(self.new_file)

    def test_get_last_modified_date_no_file(self):
        self.assertIsNone(get_last_modified_date('testststs'))

    def test_get_last_modified_date_non_existing_file(self):
        self.assertIsNone(get_file_size(self.non_existing_file))

    def test_convert_size(self):
        self.assertEqual('2.0 KB', convert_size(2048))
        self.assertEqual('20.0 MB', convert_size(20971520))
        self.assertEqual('1.0 GB', convert_size(1073741824))
        self.assertEqual('0B', convert_size(0))