Source code for oioioi.sioworkers.tests

from django.test import TestCase

from oioioi.sioworkers.jobs import run_sioworkers_job, run_sioworkers_jobs


class TestSioworkersBindings(TestCase):
    def test_sioworkers_bindings(self):
        env = run_sioworkers_job(dict(job_type='ping', ping='e1'))
        self.assertEqual(env.get('pong'), 'e1')
        envs = run_sioworkers_jobs(
            dict(
                key1=dict(job_type='ping', ping='e1'),
                key2=dict(job_type='ping', ping='e2'),
            )
        )
        self.assertEqual(envs['key1'].get('pong'), 'e1')
        self.assertEqual(envs['key2'].get('pong'), 'e2')
        self.assertEqual(len(envs), 2)