def test_rule_object_to_view_works(self): rule = doubles.make(self.mox, doubles.SecurityGroupRule, **self.to_view) self.mox.ReplayAll() with self.app.test_request_context(): res = fw_rules._fw_rule_object_to_view(rule) self.assertEquals(res, self.expected)
def test_create_rule_no_ports(self): fw_rules._get_security_group(self.sgid).AndReturn(self.sg) fw_rules.auth.client_set_for_tenant('PID', fallback_to_api=True, eperm_status=404) \ .AndReturn(self.tcs) self.tcs.compute.security_group_rules.create( parent_group_id=self.sgid, ip_protocol=u'tcp', from_port=-1, to_port=-1, cidr='10.0.0.0/8').AndReturn('Created rule') fw_rules._fw_rule_object_to_view('Created rule').AndReturn('REPLY') self.mox.ReplayAll() data = self.interact({ u'protocol': u'TCP', u'source': u'10.0.0.0/8' }) self.assertEquals(data, 'REPLY')