def test_version_prerelease(self):
b = server_ingester.ServerBinary("./server", version="0.5.0a0")
self.assertTrue(b.at_least_version("0.4.0"))
self.assertTrue(b.at_least_version("0.5.0a0"))
self.assertFalse(b.at_least_version("0.5.0"))
def test(self):
# Create a fake server binary so that the `os.path.exists` check
# passes.
fake_binary_path = os.path.join(self.get_temp_dir(), "server")
with open(fake_binary_path, "wb"):
pass
binary_info = server_ingester.ServerBinary(fake_binary_path,
version=None)
tmpdir = tempfile.TemporaryDirectory()
self.enter_context(
mock.patch.object(tempfile,
"TemporaryDirectory",
return_value=tmpdir))
port_file = os.path.join(tmpdir.name, "port")
error_file = os.path.join(tmpdir.name, "startup_error")
real_popen = subprocess.Popen
# Stub out `subprocess.Popen` to write the port file.
def fake_popen(subprocess_args, *args, **kwargs):
def target():
time.sleep(0.2) # wait one cycle
with open(port_file, "w") as outfile:
outfile.write("23456\n")
result = mock.create_autospec(real_popen, instance=True)
result.stdin = mock.Mock()
result.poll = lambda: None
result.pid = 789
threading.Thread(target=target).start()
return result
tilde_logdir = "~/tmp/logs"
expanded_logdir = os.path.expanduser(tilde_logdir)
self.assertNotEqual(tilde_logdir, expanded_logdir)
with mock.patch.object(subprocess, "Popen", wraps=fake_popen) as popen:
with mock.patch.object(grpc, "secure_channel",
autospec=True) as sc:
ingester = server_ingester.SubprocessServerDataIngester(
server_binary=binary_info,
logdir=tilde_logdir,
reload_interval=5,
channel_creds_type=grpc_util.ChannelCredsType.LOCAL,
samples_per_plugin={
"scalars": 500,
"images": 0,
},
extra_flags=["--extra-flags", "--for-fun"],
)
ingester.start()
self.assertIsInstance(ingester.data_provider,
grpc_provider.GrpcDataProvider)
expected_args = [
fake_binary_path,
"--logdir=%s" % expanded_logdir,
"--reload=5",
"--samples-per-plugin=scalars=500,images=all",
"--port=0",
"--port-file=%s" % port_file,
"--die-after-stdin",
"--error-file=%s" % error_file,
"--verbose", # logging is enabled in tests
"--extra-flags",
"--for-fun",
]
popen.assert_called_once_with(expected_args, stdin=subprocess.PIPE)
sc.assert_called_once_with("localhost:23456",
mock.ANY,
options=mock.ANY)
def test_version_none(self):
b = server_ingester.ServerBinary("./server", version=None)
self.assertTrue(b.at_least_version("0.1.0"))
self.assertTrue(b.at_least_version("999.999.999"))