class TestWorkerConfig(unittest.TestCase): def setUp(self): self.worker_config = WorkerConfig() self.worker_config.r = fakeredis.FakeStrictRedis() def test_redis_set(self): key, value = "key_set", "5" self.worker_config.set_config(key, value) self.assertEqual(int(value), self.worker_config.get_config(key)) def test_redis_get_nonexistent_key(self): self.assertEqual(None, self.worker_config.get_config("invalid_key")) def test_redis_correct_casting(self): key, value = "key_casting", 5 self.worker_config.set_config(key, value) self.assertEqual(value, self.worker_config.get_config(key)) def test_redis_doesnt_cast_strings(self): key, value = "key_string_casting", b"value" self.worker_config.set_config(key, value) self.assertEqual(value, self.worker_config.get_config(key)) def test_timeout_cpu_and_memory_defaults(self): timeout_val_defualt = 600 cpu_share_default = 100 memory_limit_default = 1024 self.assertEqual(timeout_val_defualt, self.worker_config.timeout) self.assertEqual(cpu_share_default, self.worker_config.cpu_share) self.assertEqual(memory_limit_default, self.worker_config.memory_limit) def test_timeout_cpu_and_memory_properties(self): timeout, timeout_val = "timeout", "10" cpu_share, cpu_share_val = "cpu_share", "20" memory_limit, memory_limit_val = "memory_limit", "30" self.worker_config.set_config(timeout, timeout_val) self.worker_config.set_config(cpu_share, cpu_share_val) self.worker_config.set_config(memory_limit, memory_limit_val) self.assertEqual(int(timeout_val), self.worker_config.timeout) self.assertEqual(int(cpu_share_val), self.worker_config.cpu_share) self.assertEqual(int(memory_limit_val), self.worker_config.memory_limit)
class TestWorkerConfig(unittest.TestCase): def setUp(self): self.worker_config = WorkerConfig() self.worker_config.r = fakeredis.FakeStrictRedis() def test_redis_set(self): key, value = "key_set", "5" self.worker_config.set_config(key, value) self.assertEqual(int(value), self.worker_config.get_config(key)) def test_redis_get_nonexistent_key(self): self.assertEqual(None, self.worker_config.get_config("invalid_key")) def test_redis_correct_casting(self): key, value = "key_casting", 5 self.worker_config.set_config(key, value) self.assertEqual(value, self.worker_config.get_config(key)) def test_redis_doesnt_cast_strings(self): key, value = "key_string_casting", "value" self.worker_config.set_config(key, value) self.assertEqual(value, self.worker_config.get_config(key)) def test_timeout_cpu_and_memory_defaults(self): timeout_val_defualt = 30 cpu_share_default = 100 memory_limit_default = 1024 self.assertEqual(timeout_val_defualt, self.worker_config.timeout) self.assertEqual(cpu_share_default, self.worker_config.cpu_share) self.assertEqual(memory_limit_default, self.worker_config.memory_limit) def test_timeout_cpu_and_memory_properties(self): timeout, timeout_val = "timeout", "10" cpu_share, cpu_share_val = "cpu_share", "20" memory_limit, memory_limit_val = "memory_limit", "30" self.worker_config.set_config(timeout, timeout_val) self.worker_config.set_config(cpu_share, cpu_share_val) self.worker_config.set_config(memory_limit, memory_limit_val) self.assertEqual(int(timeout_val), self.worker_config.timeout) self.assertEqual(int(cpu_share_val), self.worker_config.cpu_share) self.assertEqual(int(memory_limit_val), self.worker_config.memory_limit)