46 lines
1.5 KiB
Nix
46 lines
1.5 KiB
Nix
import ../make-test-python.nix ({...}: {
|
|
nodes = {
|
|
resourcemanager = {pkgs, ...}: {
|
|
services.hadoop.package = pkgs.hadoop_3_1;
|
|
services.hadoop.yarn.resourcemanager.enabled = true;
|
|
services.hadoop.yarnSite = {
|
|
"yarn.resourcemanager.scheduler.class" = "org.apache.hadoop.yarn.server.resourcemanager.scheduler.fifo.FifoScheduler";
|
|
};
|
|
networking.firewall.allowedTCPPorts = [
|
|
8088 # resourcemanager.webapp.address
|
|
8031 # resourcemanager.resource-tracker.address
|
|
];
|
|
};
|
|
nodemanager = {pkgs, ...}: {
|
|
services.hadoop.package = pkgs.hadoop_3_1;
|
|
services.hadoop.yarn.nodemanager.enabled = true;
|
|
services.hadoop.yarnSite = {
|
|
"yarn.resourcemanager.hostname" = "resourcemanager";
|
|
"yarn.nodemanager.log-dirs" = "/tmp/userlogs";
|
|
"yarn.nodemanager.address" = "0.0.0.0:8041";
|
|
};
|
|
networking.firewall.allowedTCPPorts = [
|
|
8042 # nodemanager.webapp.address
|
|
8041 # nodemanager.address
|
|
];
|
|
};
|
|
|
|
};
|
|
|
|
testScript = ''
|
|
start_all()
|
|
|
|
resourcemanager.wait_for_unit("yarn-resourcemanager")
|
|
resourcemanager.wait_for_unit("network.target")
|
|
resourcemanager.wait_for_open_port(8031)
|
|
resourcemanager.wait_for_open_port(8088)
|
|
|
|
nodemanager.wait_for_unit("yarn-nodemanager")
|
|
nodemanager.wait_for_unit("network.target")
|
|
nodemanager.wait_for_open_port(8042)
|
|
nodemanager.wait_for_open_port(8041)
|
|
|
|
resourcemanager.succeed("curl http://localhost:8088")
|
|
nodemanager.succeed("curl http://localhost:8042")
|
|
'';
|
|
})
|