#!/usr/bin/perl -w

use strict;

use warnings;

use diagnostics;

BEGIN {

use lib qw(/usr/local/pf/t);

use setup_test_config;

}

use pf::Switch;

use File::Basename qw(basename);

Log::Log4perl->init("log.conf");

my $logger = Log::Log4perl->get_logger( basename($0) );

Log::Log4perl::MDC->put( 'proc', basename($0) );

Log::Log4perl::MDC->put( 'tid', 0 );

use Test::More tests => 13;

use Test::MockModule;

use Test::MockObject::Extends;

use_ok('pf::floatingdevice');

use_ok('pf::floatingdevice::custom');

use pf::constants;

use pf::config;

use pf::SwitchFactory;

# test the object

my $fd = new pf::floatingdevice();

isa_ok($fd, 'pf::floatingdevice');

# subs

can_ok($fd, qw(

enablePortConfig

disablePortConfig

));

my $switch = pf::SwitchFactory->instantiate('10.0.0.1');

my $switch_port = '10001';

my $switch_locker;

my $result;

# Make the object mockable

$switch = Test::MockObject::Extends->new( $switch );

# TODO: first statement repeated to correctly init hash (should fix with less hackish perl)

$main::pf::config::ConfigFloatingDevices{'bb:bb:cc:dd:ee:ff'}{'trunkPort'} = 1;

$main::pf::config::ConfigFloatingDevices{'bb:bb:cc:dd:ee:ff'}{'trunkPort'} = 1;

$main::pf::config::ConfigFloatingDevices{'bb:bb:cc:dd:ee:ff'}{'taggedVlan'} = '1,2,3';

$main::pf::config::ConfigFloatingDevices{'bb:bb:cc:dd:ee:ff'}{'pvid'} = '10';

# Now testing enablePortConfig

# testing all failure cases

$switch->mock('disablePortSecurityByIfIndex', sub { return (0); });

$result = $fd->enablePortConfig('bb:bb:cc:dd:ee:ff', $switch, $switch_port, $switch_locker);

is($result, $FALSE, "enablePortConfig: failed to disable port-security on port");

$switch->unmock('disablePortSecurityByIfindex');

$switch->mock('disablePortSecurityByIfIndex', sub { return (1); });

$switch->mock('enablePortConfigAsTrunk', sub { return (0); });

$result = $fd->enablePortConfig('bb:bb:cc:dd:ee:ff', $switch, $switch_port, $switch_locker);

is($result, $FALSE, "enablePortConfig: failed to configure trunk");

$switch->unmock('enablePortConfigAsTrunk');

$switch->mock('enablePortConfigAsTrunk', sub { return (1); });

$switch->mock('setVlan', sub { return (0); });

$result = $fd->enablePortConfig('bb:bb:cc:dd:ee:ff', $switch, $switch_port, $switch_locker);

is($result, $FALSE, "enablePortConfig: failed to set PVID on port");

$switch->unmock('setVlan');

$switch->mock('setVlan', sub { return (1); });

$switch->mock('setIfLinkUpDownTrapEnable', sub { return (0); });

$result = $fd->enablePortConfig('bb:bb:cc:dd:ee:ff', $switch, $switch_port, $switch_locker);

is($result, $FALSE, "enablePortConfig: failed to enable LinkDown traps on port");

$switch->unmock('setIfLinkUpDownTrapEnable');

# testing success

$switch->mock('setIfLinkUpDownTrapEnable', sub { return (1); });

$result = $fd->enablePortConfig('bb:bb:cc:dd:ee:ff', $switch, $switch_port, $switch_locker);

is($result, $TRUE, "enablePortConfig: testing success");

$switch->unmock('setIfLinkUpDownTrapEnable');

# Now testing disablePortConfig

$switch->mock('setIfLinkUpDownTrapEnable', sub { return (0); });

$result = $fd->disablePortConfig('bb:bb:cc:dd:ee:ff', $switch, $switch_port, $switch_locker);

is($result, $FALSE, "disablePortConfig: failed to disable LinkDown traps on port");

$switch->unmock('setIfLinkUpDownTrapEnable');

$switch->mock('setIfLinkUpDownTrapEnable', sub { return (1); });

$switch->mock('isTrunkPort', sub { return (1); });

$switch->mock('disablePortConfigAsTrunk', sub { return (0); });

$result = $fd->disablePortConfig('bb:bb:cc:dd:ee:ff', $switch, $switch_port, $switch_locker);

is($result, $FALSE, "disablePortConfig: failed to deconfigure trunk");

$switch->unmock('disablePortConfigAsTrunk');

$switch->mock('enablePortSecurityByIfIndex', sub { return (0); });

$result = $fd->disablePortConfig('bb:bb:cc:dd:ee:ff', $switch, $switch_port, $switch_locker);

is($result, $FALSE, "disablePortConfig: failed to disable port-security on port");

$switch->unmock('enablePortSecurityByIfIndex');

# testing success

$switch->mock('enablePortSecurityByIfIndex', sub { return (1); });

$result = $fd->disablePortConfig('bb:bb:cc:dd:ee:ff', $switch, $switch_port, $switch_locker);

is($result, $TRUE, "disablePortConfig: testing success");

=head1 AUTHOR

Inverse inc. <[email protected]>

=head1 COPYRIGHT

Copyright (C) 2005-2021 Inverse inc.

=head1 LICENSE

This program is free software; you can redistribute it and/or

modify it under the terms of the GNU General Public License

as published by the Free Software Foundation; either version 2

of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful,

but WITHOUT ANY WARRANTY; without even the implied warranty of

MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

GNU General Public License for more details.

You should have received a copy of the GNU General Public License

along with this program; if not, write to the Free Software

Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,

USA.

=cut