Converted encoding to UTF-8.
[philipp/winterrodeln/wrpylib.git] / tests / test_sqlalchemy.py
index 76a9911862a20dd560665c0d231451fdb73ee3cf..f0f86649aff9bed8f2bc9a64aca167b343b4f815 100644 (file)
@@ -1,62 +1,56 @@
-#!/usr/bin/python2.7
-# -*- coding: iso-8859-15 -*-
-import types
+#!/usr/bin/python3.4
 from sqlalchemy import schema
 from sqlalchemy.engine import create_engine
 from sqlalchemy.sql import select
 from wrpylib import mwdb
+import unittest
 
 
 def connect():
     engine = create_engine('mysql://philipp@localhost:3306/philipp_winterrodeln_wiki?charset=utf8&use_unicode=1')
-    # engine = create_engine('mysql://philipp@localhost:3306/philipp_winterrodeln_wiki?charset=utf8&use_unicode=0')
     connection = engine.connect()
-    transaction = connection.begin()
+    connection.begin()
     return connection
 
 
-def test_datatype_page():
-    metadata = schema.MetaData()
-    tpage = mwdb.page_table(metadata)
-    page = connect().execute(select([tpage], tpage.c.page_id == 1321)).first()
-    assert type(page.page_title) == types.UnicodeType
-    assert type(page.page_restrictions) == types.StringType
-    assert type(page.page_touched) == types.UnicodeType
-
-
-def test_datatype_revision():
-    metadata = schema.MetaData()
-    trevision = mwdb.revision_table(metadata)
-    revision = connect().execute(select([trevision], trevision.c.rev_id == 7586)).first()
-    assert type(revision.rev_comment) == types.UnicodeType
-    assert type(revision.rev_user_text) == types.UnicodeType
-    assert type(revision.rev_timestamp) == types.UnicodeType
-
-
-def test_datatypes_text():
-    metadata = schema.MetaData()
-    ttext = mwdb.text_table(metadata)
-    text = connect().execute(select([ttext], ttext.c.old_id==7438)).first()
-    assert type(text.old_text) == types.UnicodeType
-    assert type(text.old_flags) == types.UnicodeType
-    assert text.old_flags == u'utf-8'
-
-
-def test_datatype_user():
-    metadata = schema.MetaData()
-    tuser = mwdb.user_table(metadata)
-    user = connect().execute(select([tuser], tuser.c.user_id == 1)).first()
-    assert type(user.user_name) == types.UnicodeType
-    assert type(user.user_real_name) == types.UnicodeType
-    assert type(user.user_email) == types.UnicodeType
-    assert user.user_name == u'Philipp'
-
-def test_datatype_categorylinks():
-    metadata = schema.MetaData()
-    tcategorylinks = mwdb.categorylinks_table(metadata)
-    categorylinks = connect().execute(select([tcategorylinks], tcategorylinks.c.cl_from == 609)).first()
-    assert type(categorylinks.cl_to) == types.UnicodeType
-    assert type(categorylinks.cl_sortkey) == types.UnicodeType
-    assert categorylinks.cl_sortkey == u'ALT BÄRNBAD'
-
+class TestSqlAlchemy(unittest.TestCase):
+    def test_datatype_page(self):
+        metadata = schema.MetaData()
+        tpage = mwdb.page_table(metadata)
+        page = connect().execute(select([tpage], tpage.c.page_id == 1321)).first()
+        assert type(page.page_title) == str
+        assert type(page.page_restrictions) == bytes
+        assert type(page.page_touched) == str
+
+    def test_datatype_revision(self):
+        metadata = schema.MetaData()
+        trevision = mwdb.revision_table(metadata)
+        revision = connect().execute(select([trevision], trevision.c.rev_id == 7586)).first()
+        assert type(revision.rev_comment) == str
+        assert type(revision.rev_user_text) == str
+        assert type(revision.rev_timestamp) == str
+
+    def test_datatypes_text(self):
+        metadata = schema.MetaData()
+        ttext = mwdb.text_table(metadata)
+        text = connect().execute(select([ttext], ttext.c.old_id==7438)).first()
+        assert type(text.old_text) == str
+        assert type(text.old_flags) == str
+        assert text.old_flags == 'utf-8'
+
+    def test_datatype_user(self):
+        metadata = schema.MetaData()
+        tuser = mwdb.user_table(metadata)
+        user = connect().execute(select([tuser], tuser.c.user_id == 1)).first()
+        assert type(user.user_name) == str
+        assert type(user.user_real_name) == str
+        assert type(user.user_email) == str
+        assert user.user_name == 'Philipp'
+
+    def test_datatype_categorylinks(self):
+        metadata = schema.MetaData()
+        tcategorylinks = mwdb.categorylinks_table(metadata)
+        categorylinks = connect().execute(select([tcategorylinks], tcategorylinks.c.cl_from == 609)).first()
+        assert type(categorylinks.cl_to) == str
+        assert type(categorylinks.cl_sortkey) == str
+        assert categorylinks.cl_sortkey == 'ALT BÄRNBAD'