neutron_test_tags.py

class TestReplaceNetwork(TestNetwork):

    # The network to set.
    _network = network_fakes.FakeNetwork.create_one_network(
        {'tags': ['green', 'red']})
    qos_policy = (network_fakes.FakeNetworkQosPolicy.
                  create_one_qos_policy(attrs={'id': _network.qos_policy_id}))

    def setUp(self):
        super(TestReplaceNetwork, self).setUp()

        self.network.update_network = mock.Mock(return_value=None)
        self.network.set_tags = mock.Mock(return_value=None)

        self.network.find_network = mock.Mock(return_value=self._network)
        self.network.find_qos_policy = mock.Mock(return_value=self.qos_policy)

        # Get the command object to test
        self.cmd = network.ReplaceNetwork(self.app, self.namespace)

    def _test_replace_tags(self):
        arglist = ['--tag', 'red', '--tag', 'blue']
        verifylist = [('tags', ['red', 'blue'])]
        expected_args = ['green']
        arglist.append(self._network.name)
        verifylist.append(
            ('network', self._network.name))

        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
        result = self.cmd.take_action(parsed_args)

        self.assertFalse(self.network.update_network.called)
        self.network.set_tags.assert_called_once_with(
            self._network,
            tests_utils.CompareBySet(expected_args))
        self.assertIsNone(result)