You are an expert in writing code in many different languages. Your goal is to perform code completion for the following code, keeping in mind the rest of the code and the file meta data.
    Metadata details:
    filename: partition_ack_manager_test.go
    path of file: ./repos/cadence/common/messaging/kafka
    
    the code of the file until where you have to start completion: // The MIT License (MIT)

// Copyright (c) 2017-2020 Uber Technologies Inc.

// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribu
    
func TestAddMessage(t *testing.T) {
	metricsClient := metrics.NewClient(tally.NoopScope, metrics.History)
	logger := testlogger.New(t) // Mocked
	pam := newPartitionAckManager(metricsClient, logger)
	partitionID := int32(1)
	messageID := int64(100)
	// Test adding a message
	pam.AddMessage(partitionID, messageID)

	// Verify the message is added to the ack manager
	pam.RLock() // Read lock since we are only reading data
	_, ok := pam.ackMgrs[partitionID]
	pam.RUnlock()

	assert.True(t, ok, "AckManager for partition %v was not created", partitionID)
}