require "cases/helper" require 'support/schema_dumping_helper' class PostgresqlNetworkTest < ActiveRecord::PostgreSQLTestCase include SchemaDumpingHelper class PostgresqlNetworkAddress < ActiveRecord::Base; end setup do @connection = ActiveRecord::Base.connection @connection.create_table('postgresql_network_addresses', force: true) do |t| t.inet 'inet_address', default: "192.168.1.1" t.cidr 'cidr_address', default: "192.168.1.0/24" t.macaddr 'mac_address', default: "ff:ff:ff:ff:ff:ff" end end teardown do @connection.drop_table 'postgresql_network_addresses', if_exists: true end def test_cidr_column column = PostgresqlNetworkAddress.columns_hash["cidr_address"] assert_equal :cidr, column.type assert_equal "cidr", column.sql_type assert_not column.array? type = PostgresqlNetworkAddress.type_for_attribute("cidr_address") assert_not type.binary? end def test_inet_column column = PostgresqlNetworkAddress.columns_hash["inet_address"] assert_equal :inet, column.type assert_equal "inet", column.sql_type assert_not column.array? type = PostgresqlNetworkAddress.type_for_attribute("inet_address") assert_not type.binary? end def test_macaddr_column column = PostgresqlNetworkAddress.columns_hash["mac_address"] assert_equal :macaddr, column.type assert_equal "macaddr", column.sql_type assert_not column.array? type = PostgresqlNetworkAddress.type_for_attribute("mac_address") assert_not type.binary? end def test_network_types PostgresqlNetworkAddress.create(cidr_address: '192.168.0.0/24', inet_address: '172.16.1.254/32', mac_address: '01:23:45:67:89:0a') address = PostgresqlNetworkAddress.first assert_equal IPAddr.new('192.168.0.0/24'), address.cidr_address assert_equal IPAddr.new('172.16.1.254'), address.inet_address assert_equal '01:23:45:67:89:0a', address.mac_address address.cidr_address = '10.1.2.3/32' address.inet_address = '10.0.0.0/8' address.mac_address = 'bc:de:f0:12:34:56' address.save! assert address.reload assert_equal IPAddr.new('10.1.2.3/32'), address.cidr_address assert_equal IPAddr.new('10.0.0.0/8'), address.inet_address assert_equal 'bc:de:f0:12:34:56', address.mac_address end def test_invalid_network_address invalid_address = PostgresqlNetworkAddress.new(cidr_address: 'invalid addr', inet_address: 'invalid addr') assert_nil invalid_address.cidr_address assert_nil invalid_address.inet_address assert_equal 'invalid addr', invalid_address.cidr_address_before_type_cast assert_equal 'invalid addr', invalid_address.inet_address_before_type_cast assert invalid_address.save invalid_address.reload assert_nil invalid_address.cidr_address assert_nil invalid_address.inet_address assert_nil invalid_address.cidr_address_before_type_cast assert_nil invalid_address.inet_address_before_type_cast end def test_schema_dump_with_shorthand output = dump_table_schema("postgresql_network_addresses") assert_match %r{t\.inet\s+"inet_address",\s+default: "192\.168\.1\.1"}, output assert_match %r{t\.cidr\s+"cidr_address",\s+default: "192\.168\.1\.0/24"}, output assert_match %r{t\.macaddr\s+"mac_address",\s+default: "ff:ff:ff:ff:ff:ff"}, output end end