added a test for username normalisation

blue-yonder · Sep 19, 2017 · 452299c · 452299c
1 parent 9d06e5f
commit 452299c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_unit/backends/postgres_cluster/test_postgres_cluster_driver.py b/tests/test_unit/backends/postgres_cluster/test_postgres_cluster_driver.py
@@ -10,4 +10,7 @@
 def test_get_normalized_username():
     username_with_host = 'testuser1234@testcluster1'
     stripped_username = pgcd.get_normalized_username(username_with_host)
+    assert stripped_username == "testuser1234"
+    username_without_host = 'testuser1234'
+    stripped_username = pgcd.get_normalized_username(username_with_host)
     assert stripped_username == "testuser1234"