def test_create_command(self): """Test the sync_pgfunctions command. """ management.call_command('sync_pgfunctions', *[], **{}) # Check it was created cursor_wrapper = connection.cursor() cursor = cursor_wrapper.cursor self.assertEqual(_function_exists(cursor, 'user_type'), True)
def test_create_functions_from_models(self): """Create functions using the create_functions and passing the models module. """ create_result = create_functions(models) for status, _, _ in create_result: self.assertEqual(status, 'CREATED') # Now check it was created cursor_wrapper = connection.cursor() cursor = cursor_wrapper.cursor self.assertEqual(_function_exists(cursor, 'user_type'), True)