#!/usr/bin/env python
# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from nose.tools import eq_
from pytube import utils


class TestUtils(object):
    blob = "Lorem ipsum dolor sit amet, consectetur adipiscing elit"

    def test_truncate(self):
        """Truncate string works as expected"""
        truncated = utils.truncate(self.blob, 11)
        eq_(truncated, 'Lorem ipsum')

    def test_safe_filename(self):
        """Unsafe characters get stripped from generated filename"""
        eq_(utils.safe_filename("abc1245$$"), "abc1245")
        eq_(utils.safe_filename("abc##"), "abc")
        eq_(utils.safe_filename("abc:foo"), "abc -foo")
        eq_(utils.safe_filename("abc_foo"), "abc foo")

    def test_sizeof(self):
        """Accurately converts the bytes to its humanized equivalent"""
        eq_(utils.sizeof(1), '1 byte')
        eq_(utils.sizeof(2), '2 bytes')
        eq_(utils.sizeof(2400), '2 KB')
        eq_(utils.sizeof(2400000), '2 MB')
        eq_(utils.sizeof(2400000000), '2 GB')
        eq_(utils.sizeof(2400000000000), '2 TB')
        eq_(utils.sizeof(2400000000000000), '2 PB')