示例#1
0
        public virtual void TestShowJob()
        {
            JobClientUnitTest.TestJobClient client = new JobClientUnitTest.TestJobClient(this
                                                                                         , new JobConf());
            long      startTime     = Runtime.CurrentTimeMillis();
            JobID     jobID         = new JobID(startTime.ToString(), 12345);
            JobStatus mockJobStatus = Org.Mockito.Mockito.Mock <JobStatus>();

            Org.Mockito.Mockito.When(mockJobStatus.GetJobID()).ThenReturn(jobID);
            Org.Mockito.Mockito.When(mockJobStatus.GetJobName()).ThenReturn(jobID.ToString());
            Org.Mockito.Mockito.When(mockJobStatus.GetState()).ThenReturn(JobStatus.State.Running
                                                                          );
            Org.Mockito.Mockito.When(mockJobStatus.GetStartTime()).ThenReturn(startTime);
            Org.Mockito.Mockito.When(mockJobStatus.GetUsername()).ThenReturn("mockuser");
            Org.Mockito.Mockito.When(mockJobStatus.GetQueue()).ThenReturn("mockqueue");
            Org.Mockito.Mockito.When(mockJobStatus.GetPriority()).ThenReturn(JobPriority.Normal
                                                                             );
            Org.Mockito.Mockito.When(mockJobStatus.GetNumUsedSlots()).ThenReturn(1);
            Org.Mockito.Mockito.When(mockJobStatus.GetNumReservedSlots()).ThenReturn(1);
            Org.Mockito.Mockito.When(mockJobStatus.GetUsedMem()).ThenReturn(1024);
            Org.Mockito.Mockito.When(mockJobStatus.GetReservedMem()).ThenReturn(512);
            Org.Mockito.Mockito.When(mockJobStatus.GetNeededMem()).ThenReturn(2048);
            Org.Mockito.Mockito.When(mockJobStatus.GetSchedulingInfo()).ThenReturn("NA");
            Job mockJob = Org.Mockito.Mockito.Mock <Job>();

            Org.Mockito.Mockito.When(mockJob.GetTaskReports(Matchers.IsA <TaskType>())).ThenReturn
                (new TaskReport[5]);
            Cluster mockCluster = Org.Mockito.Mockito.Mock <Cluster>();

            Org.Mockito.Mockito.When(mockCluster.GetJob(jobID)).ThenReturn(mockJob);
            client.SetCluster(mockCluster);
            ByteArrayOutputStream @out = new ByteArrayOutputStream();

            client.DisplayJobList(new JobStatus[] { mockJobStatus }, new PrintWriter(@out));
            string commandLineOutput = @out.ToString();

            System.Console.Out.WriteLine(commandLineOutput);
            NUnit.Framework.Assert.IsTrue(commandLineOutput.Contains("Total jobs:1"));
            Org.Mockito.Mockito.Verify(mockJobStatus, Org.Mockito.Mockito.AtLeastOnce()).GetJobID
                ();
            Org.Mockito.Mockito.Verify(mockJobStatus).GetState();
            Org.Mockito.Mockito.Verify(mockJobStatus).GetStartTime();
            Org.Mockito.Mockito.Verify(mockJobStatus).GetUsername();
            Org.Mockito.Mockito.Verify(mockJobStatus).GetQueue();
            Org.Mockito.Mockito.Verify(mockJobStatus).GetPriority();
            Org.Mockito.Mockito.Verify(mockJobStatus).GetNumUsedSlots();
            Org.Mockito.Mockito.Verify(mockJobStatus).GetNumReservedSlots();
            Org.Mockito.Mockito.Verify(mockJobStatus).GetUsedMem();
            Org.Mockito.Mockito.Verify(mockJobStatus).GetReservedMem();
            Org.Mockito.Mockito.Verify(mockJobStatus).GetNeededMem();
            Org.Mockito.Mockito.Verify(mockJobStatus).GetSchedulingInfo();
            // This call should not go to each AM.
            Org.Mockito.Mockito.Verify(mockCluster, Org.Mockito.Mockito.Never()).GetJob(jobID
                                                                                        );
            Org.Mockito.Mockito.Verify(mockJob, Org.Mockito.Mockito.Never()).GetTaskReports(Matchers.IsA
                                                                                            <TaskType>());
        }
示例#2
0
        public virtual void TestGetJobWithUnknownJob()
        {
            JobClientUnitTest.TestJobClient client = new JobClientUnitTest.TestJobClient(this
                                                                                         , new JobConf());
            Cluster mockCluster = Org.Mockito.Mockito.Mock <Cluster>();

            client.SetCluster(mockCluster);
            JobID id = new JobID("unknown", 0);

            Org.Mockito.Mockito.When(mockCluster.GetJob(id)).ThenReturn(null);
            NUnit.Framework.Assert.IsNull(client.GetJob(id));
        }
示例#3
0
        public virtual void TestReduceTaskReportsWithNullJob()
        {
            JobClientUnitTest.TestJobClient client = new JobClientUnitTest.TestJobClient(this
                                                                                         , new JobConf());
            Cluster mockCluster = Org.Mockito.Mockito.Mock <Cluster>();

            client.SetCluster(mockCluster);
            JobID id = new JobID("test", 0);

            Org.Mockito.Mockito.When(mockCluster.GetJob(id)).ThenReturn(null);
            TaskReport[] result = client.GetReduceTaskReports(id);
            NUnit.Framework.Assert.AreEqual(0, result.Length);
            Org.Mockito.Mockito.Verify(mockCluster).GetJob(id);
        }