salt/tests/unit/modules/test_swift.py

# -*- coding: utf-8 -*-
"""
    :codeauthor: Jayesh Kariya <jayeshk@saltstack.com>
"""

# Import Python Libs
from __future__ import absolute_import, print_function, unicode_literals

# Import Salt Libs
import salt.modules.swift as swift
from tests.support.mock import MagicMock, patch

# Import Salt Testing Libs
from tests.support.unit import TestCase


class SwiftTestCase(TestCase):
    """
    Test cases for salt.modules.swift
    """

    # 'delete' function tests: 1

    def test_delete(self):
        """
        Test for delete a container, or delete an object from a container.
        """
        with patch.object(swift, "_auth", MagicMock()):
            self.assertTrue(swift.delete("mycontainer"))

            self.assertTrue(swift.delete("mycontainer", path="myfile.png"))

    # 'get' function tests: 1

    def test_get(self):
        """
        Test for list the contents of a container,
        or return an object from a container.
        """
        with patch.object(swift, "_auth", MagicMock()):
            self.assertTrue(swift.get())

            self.assertTrue(swift.get("mycontainer"))

            self.assertTrue(
                swift.get("mycontainer", path="myfile.png", return_bin=True)
            )

            self.assertTrue(
                swift.get(
                    "mycontainer", path="myfile.png", local_file="/tmp/myfile.png"
                )
            )

            self.assertFalse(swift.get("mycontainer", path="myfile.png"))

    # 'put' function tests: 1

    def test_put(self):
        """
        Test for create a new container, or upload an object to a container.
        """
        with patch.object(swift, "_auth", MagicMock()):
            self.assertTrue(swift.put("mycontainer"))

            self.assertTrue(
                swift.put(
                    "mycontainer", path="myfile.png", local_file="/tmp/myfile.png"
                )
            )

            self.assertFalse(swift.put("mycontainer", path="myfile.png"))