1 files changed, 39 insertions, 0 deletions
diff --git a/tests/test_eclass.py b/tests/test_eclass.py
new file mode 100644
index 0000000..f336fcb
--- /dev/null
+++ b/tests/test_eclass.py
@@ -0,0 +1,39 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+    test_eclass.py
+    ~~~~~~~~~~~~~~
+    
+    eclass test suite
+    
+    :copyright: (c) 2013 by Jauhien Piatlicki
+    :license: GPL-2, see LICENSE for more details.
+"""
+
+import os
+import unittest
+
+from g_sorcery.eclass import EclassGenerator
+
+from tests.base import BaseTest
+
+
+class TestEclassGenerator(BaseTest):
+
+    def test_eclass_generator(self):
+        eclasses = ["test1", "test2"]
+        for eclass in eclasses:
+            os.system("echo 'eclass " + eclass + "' > " + os.path.join(self.tempdir.name, eclass + ".eclass"))
+
+        eclass_g = EclassGenerator(self.tempdir.name)
+        self.assertEqual(set(eclass_g.list()), set(eclasses) | set(["g-sorcery"]))
+
+        for eclass in eclasses:
+            self.assertEqual(eclass_g.generate(eclass), ["eclass " + eclass])
+
+
+def suite():
+    suite = unittest.TestSuite()
+    suite.addTest(TestEclassGenerator('test_eclass_generator'))
+    return suite