not enought tests

git-svn-id: http://freebase-python.googlecode.com/svn/trunk@10 5914aa95-5b3a-0410-a3b5-7b719e7fe9b2

1 files changed, 86 insertions, 0 deletions

diff --git a/test/test_all.py b/test/test_all.py
new file mode 100755
index 0000000..f23dc14
--- /dev/null
+++ b/test/test_all.py
@@ -0,0 +1,86 @@
+#!/usr/bin/python
+# ========================================================================
+# Copyright (c) 2007, Metaweb Technologies, Inc.
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions
+# are met:
+#     * Redistributions of source code must retain the above copyright
+#       notice, this list of conditions and the following disclaimer.
+#     * Redistributions in binary form must reproduce the above
+#       copyright notice, this list of conditions and the following
+#       disclaimer in the documentation and/or other materials provided
+#       with the distribution.
+# 
+# THIS SOFTWARE IS PROVIDED BY METAWEB TECHNOLOGIES ``AS IS'' AND ANY
+# EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL METAWEB TECHNOLOGIES BE
+# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
+# BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
+# WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
+# OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
+# IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+# ========================================================================
+
+
+import unittest
+import sys
+from freebase.api import HTTPMetawebSession
+
+
+USERNAME = 'username'
+PASSWORD = 'password'
+API_HOST = 'sandbox.freebase.com'
+TEST_QUERY = {'id': 'null', 'name': 'Sting'}
+
+class TestFreebase(unittest.TestCase):
+    
+    def test_login(self):
+        mss = HTTPMetawebSession(API_HOST, username=USERNAME,
+                                 password=PASSWORD)
+        mss.login()
+
+        # check mss._cookies?
+
+    def test_read(self):
+        query = {'type':'/music/artist','id':[{}],'name':'Sting', 'album':[{}]}
+
+        mss = HTTPMetawebSession(API_HOST)
+
+        result = mss.mqlread(query)
+        self.assertNotEqual(None, result)
+        self.assert_(result.has_key('id'))
+        self.assert_(result.has_key('type'))
+        self.assert_(result.has_key('name'))
+        self.assert_(result.has_key('album'))
+        self.assertEqual([].__class__, result['album'].__class__)
+        self.assert_( result['album'].__len__() > 0)
+        self.assertEqual( 'Sting', result['name'])
+        self.assertEqual( '/guid/9202a8c04000641f8000000000092a01', result['id'][0]['value'], str(result['id']))
+   
+    def test_write(self):
+        read_query = {'type':'/music/artist','name':'Yanni\'s Cousin Tom', 'id':{}}
+        mss = HTTPMetawebSession(API_HOST, username=USERNAME,
+                                 password=PASSWORD)
+        result = mss.mqlread(read_query)
+        self.assertEqual(None, result)
+
+        write_query = {'create':'unless_exists', 'type':'/music/artist','name':'Yanni'}
+
+        mss.login()
+        write_result = mss.mqlwrite(write_query)
+        self.assertNotEqual(None, write_result)
+        self.assert_(write_result.has_key('create'))
+        self.assert_(write_result.has_key('type'))
+        self.assert_(write_result.has_key('name'))
+        self.assertEqual('existed', write_result['create'])
+        self.assertEqual('Yanni', write_result['name'])
+        self.assertEqual('/music/artist', write_result['type'])
+   
+if __name__ == '__main__':
+  unittest.main()
+