def test_process_df_match_badpartition(self): sizer = SizerSSH(self.args, self.cluster) match_obj = sizer.size_re.match("5002\t/path/to/data/testTopic1-2\n") sizer.process_df_match(match_obj, 1)
def test_process_df_match_badtopic(self): sizer = SizerSSH(self.args, self.cluster) match_obj = sizer.size_re.match("5002\t/path/to/data/badTopic1-1\n") sizer.process_df_match(match_obj, 1)
def test_process_df_match(self): sizer = SizerSSH(self.args, self.cluster) match_obj = sizer.size_re.match("5002\t/path/to/data/testTopic1-1\n") sizer.process_df_match(match_obj, 1) assert self.cluster.topics['testTopic1'].partitions[1].size == 5002