epgsql_pool_settings_tests.erl 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. -module(epgsql_pool_settings_tests).
  2. -include("epgsql_pool.hrl").
  3. -include_lib("eunit/include/eunit.hrl").
  4. get_set_test() ->
  5. epgsql_pool_settings:start_link(),
  6. ?assertEqual(10000, epgsql_pool_settings:get(connection_timeout)),
  7. ?assertEqual(10000, epgsql_pool_settings:get(query_timeout)),
  8. ?assertEqual(10000, epgsql_pool_settings:get(pooler_get_worker_timeout)),
  9. epgsql_pool_settings:set(connection_timeout, 5000),
  10. ?assertEqual(5000, epgsql_pool_settings:get(connection_timeout)),
  11. epgsql_pool_settings:set(max_reconnect_timeout, 4500),
  12. ?assertEqual(4500, epgsql_pool_settings:get(max_reconnect_timeout)),
  13. ?assertThrow({settings_not_found, some_key},
  14. epgsql_pool_settings:get(some_key)),
  15. epgsql_pool_settings:set(some_key, 42),
  16. ?assertEqual(42, epgsql_pool_settings:get(some_key)),
  17. epgsql_pool_settings ! stop,
  18. ok.
  19. connection_params_test() ->
  20. epgsql_pool_settings:start_link(),
  21. ?assertEqual({error, not_found}, epgsql_pool_settings:get_connection_params(some_pool)),
  22. Params1 = #epgsql_connection_params{host = "localhost", port = 5432,
  23. username="user", password="123",
  24. database="db"},
  25. epgsql_pool_settings:set_connection_params(pool_1, Params1),
  26. Params2 = #epgsql_connection_params{host = "some.host", port = 5432,
  27. username="someuser", password="123",
  28. database="somedb"},
  29. epgsql_pool_settings:set_connection_params(pool_2, Params2),
  30. ?assertEqual({ok, Params1}, epgsql_pool_settings:get_connection_params(pool_1)),
  31. ?assertEqual({ok, Params2}, epgsql_pool_settings:get_connection_params(pool_2)),
  32. epgsql_pool_settings ! stop,
  33. ok.