def setUp(self): self.pubnub = MagicMock(spec=PubNub, config=pnconf_pam, sdk_name=sdk_name, timestamp=MagicMock(return_value=123), uuid=None) self.pubnub.uuid = "UUID_AuditUnitTest" self.audit = Audit(self.pubnub)
def setUp(self): self.pubnub = MagicMock( spec=PubNub, config=pnconf_pam, sdk_name=sdk_name, timestamp=MagicMock(return_value=123), uuid=None ) self.pubnub.uuid = "UUID_AuditUnitTest" self.audit = Audit(self.pubnub)
class TestAudit(unittest.TestCase): def setUp(self): self.pubnub = MagicMock( spec=PubNub, config=pnconf_pam, sdk_name=sdk_name, timestamp=MagicMock(return_value=123), uuid=None ) self.pubnub.uuid = "UUID_AuditUnitTest" self.pubnub._telemetry_manager = TelemetryManager() self.audit = Audit(self.pubnub) def test_audit_channel(self): self.audit.channels('ch') self.assertEquals(self.audit.build_path(), Audit.AUDIT_PATH % pnconf_pam.subscribe_key) self.assertEqual(self.audit.build_params_callback()({}), { 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid, 'timestamp': '123', 'channel': 'ch', 'signature': utils.sign_sha256(pnconf_pam.secret_key, pnconf_pam.subscribe_key + "\n" + pnconf_pam.publish_key + "\n" + "audit\n" + utils.prepare_pam_arguments({ 'timestamp': 123, 'channel': 'ch', 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid })) }) def test_audit_channel_group(self): self.audit.channel_groups(['gr1', 'gr2']) self.assertEquals(self.audit.build_path(), Audit.AUDIT_PATH % pnconf_pam.subscribe_key) self.assertEqual(self.audit.build_params_callback()({}), { 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid, 'timestamp': '123', 'channel-group': 'gr1,gr2', 'signature': utils.sign_sha256(pnconf_pam.secret_key, pnconf_pam.subscribe_key + "\n" + pnconf_pam.publish_key + "\n" + "audit\n" + utils.prepare_pam_arguments({ 'timestamp': 123, 'channel-group': 'gr1,gr2', 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid })) })
class TestAudit(unittest.TestCase): def setUp(self): self.pubnub = MagicMock( spec=PubNub, config=pnconf_pam, sdk_name=sdk_name, timestamp=MagicMock(return_value=123), uuid=None ) self.pubnub.uuid = "UUID_AuditUnitTest" self.pubnub._telemetry_manager = TelemetryManager() self.audit = Audit(self.pubnub) def test_audit_channel(self): self.audit.channels('ch') self.assertEquals(self.audit.build_path(), Audit.AUDIT_PATH % pnconf_pam.subscribe_key) pam_args = utils.prepare_pam_arguments({ 'timestamp': 123, 'channel': 'ch', 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid }) sign_input = pnconf_pam.subscribe_key + "\n" + pnconf_pam.publish_key + "\n" + "audit\n" + pam_args self.assertEqual(self.audit.build_params_callback()({}), { 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid, 'timestamp': '123', 'channel': 'ch', 'signature': utils.sign_sha256(pnconf_pam.secret_key, sign_input) }) def test_audit_channel_group(self): self.audit.channel_groups(['gr1', 'gr2']) self.assertEquals(self.audit.build_path(), Audit.AUDIT_PATH % pnconf_pam.subscribe_key) pam_args = utils.prepare_pam_arguments({ 'timestamp': 123, 'channel-group': 'gr1,gr2', 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid }) sign_input = pnconf_pam.subscribe_key + "\n" + pnconf_pam.publish_key + "\n" + "audit\n" + pam_args self.assertEqual(self.audit.build_params_callback()({}), { 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid, 'timestamp': '123', 'channel-group': 'gr1,gr2', 'signature': utils.sign_sha256(pnconf_pam.secret_key, sign_input) })