CC-5990, CC-5991 - Python cleanup, removed need for /usr/lib/airtime

python_apps/media-monitor/mm2/tests/test_pure.py

@@ -0,0 +1,75 @@
+# -*- coding: utf-8 -*-
+import unittest
+import os
+import media.monitor.pure as mmp
+
+class TestMMP(unittest.TestCase):
+    def setUp(self):
+        self.md1 = {'MDATA_KEY_MD5': '71185323c2ab0179460546a9d0690107',
+                    'MDATA_KEY_FTYPE': 'audioclip',
+                    'MDATA_KEY_MIME': 'audio/vorbis',
+                    'MDATA_KEY_DURATION': '0:0:25.000687',
+                    'MDATA_KEY_SAMPLERATE': 48000,
+                    'MDATA_KEY_BITRATE': 64000,
+                    'MDATA_KEY_REPLAYGAIN': 0,
+                    'MDATA_KEY_TITLE': u'ACDC_-_Back_In_Black-sample-64kbps'}
+
+    def test_apply_rules(self):
+        sample_dict = {
+            'key' : 'val',
+            'test' : 'IT',
+        }
+        rules = {
+            'key' : lambda x : x.upper(),
+            'test' : lambda y : y.lower()
+        }
+        sample_dict = mmp.apply_rules_dict(sample_dict, rules)
+        self.assertEqual(sample_dict['key'], 'VAL')
+        self.assertEqual(sample_dict['test'], 'it')
+
+    def test_default_to(self):
+        sd = { }
+        def_keys = ['one','two','three']
+        sd = mmp.default_to(dictionary=sd, keys=def_keys, default='DEF')
+        for k in def_keys: self.assertEqual( sd[k], 'DEF' )
+
+    def test_file_md5(self):
+        p = os.path.realpath(__file__)
+        m1 = mmp.file_md5(p)
+        m2 = mmp.file_md5(p,10)
+        self.assertTrue( m1 != m2 )
+        self.assertRaises( ValueError, lambda : mmp.file_md5('/file/path') )
+        self.assertTrue( m1 == mmp.file_md5(p) )
+
+    def test_sub_path(self):
+        f1 = "/home/testing/123.mp3"
+        d1 = "/home/testing"
+        d2 = "/home/testing/"
+        self.assertTrue( mmp.sub_path(d1, f1) )
+        self.assertTrue( mmp.sub_path(d2, f1) )
+
+    def test_parse_int(self):
+        self.assertEqual( mmp.parse_int("123"), "123" )
+        self.assertEqual( mmp.parse_int("123asf"), "123" )
+        self.assertEqual( mmp.parse_int("asdf"), None )
+
+    def test_truncate_to_length(self):
+        s1 = "testing with non string literal"
+        s2 = u"testing with unicode literal"
+        self.assertEqual( len(mmp.truncate_to_length(s1, 5)), 5)
+        self.assertEqual( len(mmp.truncate_to_length(s2, 8)), 8)
+
+
+    def test_owner_id(self):
+        start_path = "testing.mp3"
+        id_path    = "testing.mp3.identifier"
+        o_id   = 123
+        f = open(id_path, 'w')
+        f.write("123")
+        f.close()
+        possible_id = mmp.owner_id(start_path)
+        self.assertFalse( os.path.exists(id_path) )
+        self.assertEqual( possible_id, o_id )
+        self.assertEqual( -1, mmp.owner_id("something.random") )
+
+if __name__ == '__main__': unittest.main()