Fix tests and reverse mappings

mongo_connector/doc_managers/mappings.py

@@ -124,8 +124,10 @@ def validate_mapping(mappings):
 
         for collection in dbmapping:
             mapping = dbmapping[collection]
+            reversed_pk_mapping = get_reversed_pk_mapping(mapping)
 
-            if mapping['pk'] not in mapping:
+            if (reversed_pk_mapping is None or reversed_pk_mapping not in mapping) and \
+                            mapping['pk'] not in mapping:
                 # look for a linked table
                 for linked_collection in dbmapping:
                     if linked_collection != collection:
@@ -207,3 +209,14 @@ def validate_mapping(mappings):
                                         dest
                                     )
                                 )
+
+
+def get_reversed_pk_mapping(mapping):
+    if 'pk' not in mapping:
+        return None
+
+    for field in mapping:
+        if 'dest' in mapping[field] and mapping[field]['dest'] == mapping['pk']:
+            return field
+
+    return None

mongo_connector/doc_managers/sql.py

@@ -84,7 +84,7 @@ def sql_add_foreign_keys(cursor, foreign_keys):
 
 
 def unique_list(input_list):
-    return list(set(input_list))
+    return sorted(list(set(input_list)))
 
 
 def sql_bulk_insert(cursor, mappings, namespace, documents):
@@ -198,9 +198,6 @@ def sql_bulk_insert(cursor, mappings, namespace, documents):
 
             LOG.error(u"Traceback:\n%s", traceback.format_exc())
 
-def unique_list(input_list):
-    return list(set(input_list))
-
 def _sql_bulk_insert(query, mappings, namespace, documents):
     if not documents:
         return

tests/test_postgresql_manager.py

@@ -1,14 +1,13 @@
 # -*- coding: utf-8 -*-
 
+from time import time
 from unittest import TestCase, main
+
 from mock import MagicMock, patch, mock_open, call
-from time import time
-import json
 
 from mongo_connector.doc_managers import postgresql_manager
 from .fixtures import *
 
-
 MAPPING_RAW = '''{
     "db": {
         "col": {