1 files changed, 25 insertions, 0 deletions
diff --git a/tests/units/toolkit/packagekit.py b/tests/units/toolkit/packagekit.py
new file mode 100755
index 0000000..1378e2f
--- /dev/null
+++ b/tests/units/toolkit/packagekit.py
@@ -0,0 +1,25 @@
+#!/usr/bin/env python
+# sugar-lint: disable
+
+from __init__ import tests
+
+from sugar_network.toolkit.packagekit import cleanup_distro_version
+
+
+class Packagekit(tests.Test):
+
+    def test_cleanup_distro_version(self):
+		self.assertEqual('0.3.1-1', cleanup_distro_version('1:0.3.1-1'))
+		self.assertEqual('0.3.1-1', cleanup_distro_version('0.3.1-1ubuntu0'))
+		self.assertEqual('0.3-post1-rc2', cleanup_distro_version('0.3-post1-rc2'))
+		self.assertEqual('0.3.1-2', cleanup_distro_version('0.3.1-r2-r3'))
+		self.assertEqual('6.17', cleanup_distro_version('6b17'))
+		self.assertEqual('20-1', cleanup_distro_version('b20_1'))
+		self.assertEqual('17', cleanup_distro_version('p17'))
+		self.assertEqual('7-pre3-2.1.1-3', cleanup_distro_version('7~u3-2.1.1-3'))	# Debian snapshot
+		self.assertEqual('7-pre3-2.1.1-pre1-1', cleanup_distro_version('7~u3-2.1.1~pre1-1ubuntu2'))
+		self.assertEqual(None, cleanup_distro_version('cvs'))
+
+
+if __name__ == '__main__':
+    tests.main()