2 files changed, 57 insertions, 2 deletions
diff --git a/gluster/swift/__init__.py b/gluster/swift/__init__.py
index fa360b6..f2c2403 100644
--- a/gluster/swift/__init__.py
+++ b/gluster/swift/__init__.py
@@ -8,8 +8,10 @@ class PkgInfo(object):
         self.final = final
 
     def save_config(self, filename):
-        """Crates a file with the package configuration
-        which can be sourced by a bash script"""
+        """
+        Creates a file with the package configuration which can be sourced by
+        a bash script.
+        """
         with open(filename, 'w') as fd:
             fd.write("PKG_NAME=%s\n" % self.name)
             fd.write("PKG_VERSION=%s\n" % self.canonical_version)
diff --git a/test/unit/test_swift.py b/test/unit/test_swift.py
new file mode 100644
index 0000000..5c78a8b
--- /dev/null
+++ b/test/unit/test_swift.py
@@ -0,0 +1,53 @@
+# Copyright (c) 2013 Red Hat, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+# implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+""" Tests for gluster.swift """
+
+import os
+import unittest
+import shutil
+import tempfile
+
+import gluster.swift as gs
+
+
+class TestPkgInfo(unittest.TestCase):
+    """
+    Tests for gluster.swift PkgInfo class.
+    """
+
+    def test_constructor(self):
+        pi = gs.PkgInfo('a', 'b', 'c')
+        assert pi.canonical_version == 'a'
+        assert pi.name == 'b'
+        assert pi.final == 'c'
+
+    def test_pretty_version(self):
+        pi = gs.PkgInfo('a', 'b', False)
+        assert pi.pretty_version == 'a-dev'
+        pi = gs.PkgInfo('a', 'b', True)
+        assert pi.pretty_version == 'a'
+
+    def test_save_config(self):
+        pi = gs.PkgInfo('a', 'b', 'c')
+        td = tempfile.mkdtemp()
+        try:
+            sc = os.path.join(td, 'saved_config.txt')
+            pi.save_config(sc)
+            exp = 'PKG_NAME=b\nPKG_VERSION=a\n'
+            contents = file(sc, 'r').read()
+            assert contents == exp
+        finally:
+            shutil.rmtree(td)