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)
})
